mirror of
https://github.com/yzernik/squeaknode.git
synced 2026-08-15 12:50:47 +02:00
138ef7d984
* Add squeak_core module * Remove unused import * Delete squeakblockverifier * Use squeak_core to make squeak * Remove blockchain client from controller * Move create offer method to squeak core * Add some type hints * Move get buy offer method to squeak_core * Delete commented old get buy offer block * Add pay_offer to squeak_core * Add get received payments generator function to squak core * Add get_offer method to squeak_core * Remove lightning_client property from peer_task * Remove old comments from payment verifier class * Use bytes internally for payment_hash * Represent secret_key as bytes internally
160 lines
3.5 KiB
Python
160 lines
3.5 KiB
Python
import mock
|
|
import pytest
|
|
|
|
from squeaknode.config.config import SqueaknodeConfig
|
|
from squeaknode.core.lightning_address import LightningAddressHostPort
|
|
from squeaknode.core.squeak_controller import SqueakController
|
|
from squeaknode.core.squeak_core import SqueakCore
|
|
from squeaknode.core.squeak_peer import SqueakPeer
|
|
from squeaknode.db.squeak_db import SqueakDb
|
|
from squeaknode.node.squeak_store import SqueakStore
|
|
from squeaknode.node.squeak_whitelist import SqueakWhitelist
|
|
|
|
|
|
@pytest.fixture
|
|
def config():
|
|
squeaknode_config = SqueaknodeConfig()
|
|
squeaknode_config.read()
|
|
return squeaknode_config
|
|
|
|
|
|
@pytest.fixture
|
|
def regtest_config():
|
|
squeaknode_config = SqueaknodeConfig(
|
|
dict_config={'core': {'network': 'regtest'}}
|
|
)
|
|
squeaknode_config.read()
|
|
return squeaknode_config
|
|
|
|
|
|
@pytest.fixture
|
|
def squeak_db():
|
|
# return SqueakDb(None, None, None)
|
|
return mock.Mock(spec=SqueakDb)
|
|
|
|
|
|
@pytest.fixture
|
|
def squeak_core():
|
|
return mock.Mock(spec=SqueakCore)
|
|
|
|
|
|
@pytest.fixture
|
|
def lightning_host_port():
|
|
return LightningAddressHostPort(host="my_lightning_host", port=8765)
|
|
|
|
|
|
@pytest.fixture
|
|
def price_msat():
|
|
return 777
|
|
|
|
|
|
@pytest.fixture
|
|
def max_squeaks_per_address_per_hour():
|
|
return 5000
|
|
|
|
|
|
@pytest.fixture
|
|
def squeak_whitelist():
|
|
return mock.Mock(spec=SqueakWhitelist)
|
|
|
|
|
|
@pytest.fixture
|
|
def squeak_store():
|
|
return mock.Mock(spec=SqueakStore)
|
|
|
|
|
|
@pytest.fixture
|
|
def squeak_controller(
|
|
squeak_db,
|
|
squeak_core,
|
|
squeak_store,
|
|
squeak_whitelist,
|
|
config,
|
|
):
|
|
return SqueakController(
|
|
squeak_db,
|
|
squeak_core,
|
|
squeak_store,
|
|
squeak_whitelist,
|
|
config,
|
|
)
|
|
|
|
|
|
@pytest.fixture
|
|
def regtest_squeak_controller(
|
|
squeak_db,
|
|
squeak_core,
|
|
squeak_store,
|
|
squeak_whitelist,
|
|
regtest_config,
|
|
):
|
|
return SqueakController(
|
|
squeak_db,
|
|
squeak_core,
|
|
squeak_store,
|
|
squeak_whitelist,
|
|
regtest_config,
|
|
)
|
|
|
|
|
|
def test_nothing():
|
|
assert True
|
|
|
|
|
|
def test_get_buy_offer(squeak_controller):
|
|
assert squeak_controller.get_buy_offer is not None
|
|
|
|
|
|
def test_get_network_default(squeak_controller):
|
|
assert squeak_controller.get_network() == "testnet"
|
|
|
|
|
|
def test_get_network_regtest(regtest_squeak_controller):
|
|
assert regtest_squeak_controller.get_network() == "regtest"
|
|
|
|
|
|
# def test_get_network_regtest(config, squeak_controller):
|
|
# # with mock.patch.object(Config, 'squeaknode_network', new_callable=mock.PropertyMock) as mock_config:
|
|
# # mock_config.return_value = 'regtest'
|
|
# config.squeaknode_network = "regtest"
|
|
# print(config.squeaknode_network)
|
|
|
|
# assert squeak_controller.get_network() == "regtest"
|
|
|
|
|
|
def test_create_peer(squeak_db, squeak_controller):
|
|
squeak_controller.create_peer(
|
|
"fake_peer_name",
|
|
"fake_host",
|
|
5678,
|
|
)
|
|
|
|
squeak_db.insert_peer.assert_called_with(
|
|
SqueakPeer(
|
|
peer_id=None,
|
|
peer_name="fake_peer_name",
|
|
host="fake_host",
|
|
port=5678,
|
|
uploading=False,
|
|
downloading=False,
|
|
)
|
|
)
|
|
|
|
|
|
def test_create_peer_default_port(config, squeak_db, squeak_controller):
|
|
squeak_controller.create_peer(
|
|
"fake_peer_name",
|
|
"fake_host",
|
|
0,
|
|
)
|
|
|
|
squeak_db.insert_peer.assert_called_with(
|
|
SqueakPeer(
|
|
peer_id=None,
|
|
peer_name="fake_peer_name",
|
|
host="fake_host",
|
|
port=config.core.default_peer_rpc_port,
|
|
uploading=False,
|
|
downloading=False,
|
|
)
|
|
)
|