mirror of
https://github.com/yzernik/squeaknode.git
synced 2026-08-15 12:50:47 +02:00
508d3ab539
* Stop using save_squeak method of squeak_store * Stop using get_squeak method of squeak_store * Stop using delete_squeak method of squeak store * Stop using unlock_squeak in squeak_store * Remove squeak_store usage from peer task * Remove some usages of squeak_db from peer_task * Remove all squeak_db usage from peer_task * Remove squeak_store from main module * Delete squeak store class
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_rate_limiter import SqueakRateLimiter
|
|
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_rate_limiter():
|
|
return mock.Mock(spec=SqueakRateLimiter)
|
|
|
|
|
|
@pytest.fixture
|
|
def squeak_controller(
|
|
squeak_db,
|
|
squeak_core,
|
|
squeak_whitelist,
|
|
squeak_rate_limiter,
|
|
config,
|
|
):
|
|
return SqueakController(
|
|
squeak_db,
|
|
squeak_core,
|
|
squeak_whitelist,
|
|
squeak_rate_limiter,
|
|
config,
|
|
)
|
|
|
|
|
|
@pytest.fixture
|
|
def regtest_squeak_controller(
|
|
squeak_db,
|
|
squeak_core,
|
|
squeak_whitelist,
|
|
squeak_rate_limiter,
|
|
regtest_config,
|
|
):
|
|
return SqueakController(
|
|
squeak_db,
|
|
squeak_core,
|
|
squeak_whitelist,
|
|
squeak_rate_limiter,
|
|
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,
|
|
)
|
|
)
|