GlcoinCore/squeaknode
1
0
Fork 0
mirror of https://github.com/yzernik/squeaknode.git synced 2026-08-17 13:07:32 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

Mock out db in unit test (#1485)

Browse source 
* Simplify fixtures in squeak node unit test

* Mock out get connection string which creates data dir
This commit is contained in:
Jonathan Zernik 2021-10-03 19:49:22 -07:00 • committed by GitHub
parent be8ebe7f42
commit 24e64fc30b
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 39 additions and 36 deletions
Download patch file Download diff file Expand all files Collapse all files

75
tests/node/test_squeak_node.py
View file

@ -26,47 +26,50 @@ from squeaknode.config.config import SqueaknodeConfig
from squeaknode.node.squeak_node import SqueakNode
# TODO: move to conftest.
@pytest.fixture
def config():
squeaknode_config = SqueaknodeConfig(
dict_config={
'node': {
'network': 'mainnet'
},
},
@pytest.fixture()
def mock_lightning_client():
yield mock.patch('squeaknode.node.squeak_node.LNDLightningClient', autospec=True)
def get_config(config_dict):
config = SqueaknodeConfig(
dict_config=config_dict,
)
squeaknode_config.read()
return squeaknode_config
config.read()
return config
@pytest.fixture
def config_webadmin_enabled():
squeaknode_config = SqueaknodeConfig(
dict_config={
'node': {
'network': 'mainnet'
},
'webadmin': {
'enabled': 'true'
},
},
)
squeaknode_config.read()
return squeaknode_config
mainnet_config = {
'node': {
'network': 'mainnet'
},
}
@mock.patch('squeaknode.node.squeak_node.LNDLightningClient', autospec=True)
def test_start_stop(mock_lightning_client, config):
squeak_node = SqueakNode(config)
squeak_node.start_running()
squeak_node.stop_running()
@mock.patch('squeaknode.node.squeak_node.LNDLightningClient', autospec=True)
def test_start_stop_webadmin_enabled(mock_lightning_client, config_webadmin_enabled):
squeak_node = SqueakNode(config_webadmin_enabled)
webadmin_enabled_config = {
'node': {
'network': 'mainnet'
},
'webadmin': {
'enabled': 'true'
},
}
@pytest.fixture(scope="module", params=[mainnet_config, webadmin_enabled_config])
def squeak_node(request):
print('--------request.param---------')
print(request.param)
with mock.patch('squeaknode.node.squeak_node.LNDLightningClient', autospec=True), \
mock.patch('squeaknode.node.squeak_node.get_connection_string', autospec=True), \
mock.patch('squeaknode.node.squeak_node.get_engine', autospec=True), \
mock.patch('squeaknode.node.squeak_node.SqueakDb', autospec=True):
config = get_config(request.param)
yield SqueakNode(config)
def test_start_stop(squeak_node):
print('--------squeak_node.config---------')
print(squeak_node.config)
squeak_node.start_running()
squeak_node.stop_running()