forked from LBRYCommunity/lbry-sdk
65 lines
2.7 KiB
Python
65 lines
2.7 KiB
Python
import contextlib
|
|
import asyncio
|
|
import logging
|
|
from io import StringIO
|
|
from torba.testcase import AsyncioTestCase
|
|
|
|
from lbry.conf import Config
|
|
from lbry.extras import cli
|
|
from lbry.extras.daemon.Components import (
|
|
DATABASE_COMPONENT, BLOB_COMPONENT, HEADERS_COMPONENT, WALLET_COMPONENT, DHT_COMPONENT,
|
|
HASH_ANNOUNCER_COMPONENT, STREAM_MANAGER_COMPONENT, PEER_PROTOCOL_SERVER_COMPONENT,
|
|
UPNP_COMPONENT, EXCHANGE_RATE_MANAGER_COMPONENT
|
|
)
|
|
from lbry.extras.daemon.Daemon import Daemon
|
|
|
|
|
|
class CLIIntegrationTest(AsyncioTestCase):
|
|
|
|
async def asyncSetUp(self):
|
|
conf = Config()
|
|
conf.data_dir = '/tmp'
|
|
conf.share_usage_data = False
|
|
conf.api = 'localhost:5299'
|
|
conf.components_to_skip = (
|
|
DATABASE_COMPONENT, BLOB_COMPONENT, HEADERS_COMPONENT, WALLET_COMPONENT, DHT_COMPONENT,
|
|
HASH_ANNOUNCER_COMPONENT, STREAM_MANAGER_COMPONENT, PEER_PROTOCOL_SERVER_COMPONENT,
|
|
UPNP_COMPONENT, EXCHANGE_RATE_MANAGER_COMPONENT
|
|
)
|
|
Daemon.component_attributes = {}
|
|
self.daemon = Daemon(conf)
|
|
await self.daemon.start()
|
|
|
|
async def asyncTearDown(self):
|
|
await self.daemon.stop(shutdown_runner=False)
|
|
|
|
def test_cli_status_command_with_auth(self):
|
|
actual_output = StringIO()
|
|
with contextlib.redirect_stdout(actual_output):
|
|
cli.main(["--api", "localhost:5299", "status"])
|
|
actual_output = actual_output.getvalue()
|
|
self.assertIn("connection_status", actual_output)
|
|
|
|
def test_setup_logging(self):
|
|
def setup(argv):
|
|
parser = cli.get_argument_parser()
|
|
args, command_args = parser.parse_known_args(argv)
|
|
loop = asyncio.get_event_loop()
|
|
conf = Config.create_from_arguments(args)
|
|
cli.setup_logging(args, conf, loop)
|
|
|
|
setup(["start"])
|
|
self.assertTrue(logging.getLogger("lbry").isEnabledFor(logging.INFO))
|
|
self.assertFalse(logging.getLogger("lbry").isEnabledFor(logging.DEBUG))
|
|
|
|
setup(["start", "--verbose"])
|
|
self.assertTrue(logging.getLogger("lbry").isEnabledFor(logging.DEBUG))
|
|
self.assertTrue(logging.getLogger("lbry").isEnabledFor(logging.INFO))
|
|
self.assertFalse(logging.getLogger("torba").isEnabledFor(logging.DEBUG))
|
|
|
|
setup(["start", "--verbose", "lbry.extras", "lbry.wallet", "torba.client"])
|
|
self.assertTrue(logging.getLogger("lbry.extras").isEnabledFor(logging.DEBUG))
|
|
self.assertTrue(logging.getLogger("lbry.wallet").isEnabledFor(logging.DEBUG))
|
|
self.assertTrue(logging.getLogger("torba.client").isEnabledFor(logging.DEBUG))
|
|
self.assertFalse(logging.getLogger("lbry").isEnabledFor(logging.DEBUG))
|
|
self.assertFalse(logging.getLogger("torba").isEnabledFor(logging.DEBUG))
|