forked from LBRYCommunity/lbry-sdk
59 lines
2.2 KiB
Python
59 lines
2.2 KiB
Python
import contextlib
|
|
from twisted.trial import unittest
|
|
from io import StringIO
|
|
from twisted.internet import defer
|
|
|
|
from lbrynet import conf
|
|
from lbrynet import cli
|
|
from lbrynet.daemon.Components import DATABASE_COMPONENT, BLOB_COMPONENT, HEADERS_COMPONENT, WALLET_COMPONENT, \
|
|
DHT_COMPONENT, HASH_ANNOUNCER_COMPONENT, STREAM_IDENTIFIER_COMPONENT, FILE_MANAGER_COMPONENT, \
|
|
PEER_PROTOCOL_SERVER_COMPONENT, REFLECTOR_COMPONENT, UPNP_COMPONENT, EXCHANGE_RATE_MANAGER_COMPONENT, \
|
|
RATE_LIMITER_COMPONENT, PAYMENT_RATE_COMPONENT
|
|
from lbrynet.daemon.Daemon import Daemon
|
|
|
|
|
|
class CLIIntegrationTest(unittest.TestCase):
|
|
USE_AUTH = False
|
|
|
|
@defer.inlineCallbacks
|
|
def setUp(self):
|
|
skip = [
|
|
DATABASE_COMPONENT, BLOB_COMPONENT, HEADERS_COMPONENT, WALLET_COMPONENT,
|
|
DHT_COMPONENT, HASH_ANNOUNCER_COMPONENT, STREAM_IDENTIFIER_COMPONENT, FILE_MANAGER_COMPONENT,
|
|
PEER_PROTOCOL_SERVER_COMPONENT, REFLECTOR_COMPONENT, UPNP_COMPONENT, EXCHANGE_RATE_MANAGER_COMPONENT,
|
|
RATE_LIMITER_COMPONENT, PAYMENT_RATE_COMPONENT
|
|
]
|
|
conf.initialize_settings(load_conf_file=False)
|
|
conf.settings['use_auth_http'] = self.USE_AUTH
|
|
conf.settings["components_to_skip"] = skip
|
|
conf.settings.initialize_post_conf_load()
|
|
Daemon.component_attributes = {}
|
|
self.daemon = Daemon()
|
|
yield self.daemon.start_listening()
|
|
|
|
def tearDown(self):
|
|
return self.daemon._shutdown()
|
|
|
|
|
|
class AuthenticatedCLITest(CLIIntegrationTest):
|
|
USE_AUTH = True
|
|
|
|
def test_cli_status_command_with_auth(self):
|
|
self.assertTrue(self.daemon._use_authentication)
|
|
actual_output = StringIO()
|
|
with contextlib.redirect_stdout(actual_output):
|
|
cli.main(["status"])
|
|
actual_output = actual_output.getvalue()
|
|
self.assertIn("connection_status", actual_output)
|
|
|
|
|
|
class UnauthenticatedCLITest(CLIIntegrationTest):
|
|
USE_AUTH = False
|
|
|
|
def test_cli_status_command_with_auth(self):
|
|
self.assertFalse(self.daemon._use_authentication)
|
|
actual_output = StringIO()
|
|
with contextlib.redirect_stdout(actual_output):
|
|
cli.main(["status"])
|
|
actual_output = actual_output.getvalue()
|
|
self.assertIn("connection_status", actual_output)
|