forked from LBRYCommunity/lbry-sdk
135 lines
5 KiB
Python
135 lines
5 KiB
Python
import contextlib
|
|
from io import StringIO
|
|
from unittest import TestCase
|
|
|
|
import docopt
|
|
from torba.testcase import AsyncioTestCase
|
|
|
|
from lbrynet.extras.cli import normalize_value, main
|
|
from lbrynet.extras.system_info import get_platform
|
|
from lbrynet.extras.daemon.Daemon import Daemon
|
|
|
|
|
|
class CLITest(AsyncioTestCase):
|
|
|
|
@staticmethod
|
|
def shell(argv):
|
|
actual_output = StringIO()
|
|
with contextlib.redirect_stdout(actual_output):
|
|
with contextlib.redirect_stderr(actual_output):
|
|
try:
|
|
main(argv)
|
|
except SystemExit as e:
|
|
print(e.args[0])
|
|
return actual_output.getvalue().strip()
|
|
|
|
def test_guess_type(self):
|
|
self.assertEqual('0.3.8', normalize_value('0.3.8'))
|
|
self.assertEqual('0.3', normalize_value('0.3'))
|
|
self.assertEqual(3, normalize_value('3'))
|
|
self.assertEqual(3, normalize_value(3))
|
|
|
|
self.assertEqual(
|
|
'VdNmakxFORPSyfCprAD/eDDPk5TY9QYtSA==',
|
|
normalize_value('VdNmakxFORPSyfCprAD/eDDPk5TY9QYtSA==')
|
|
)
|
|
|
|
self.assertTrue(normalize_value('TRUE'))
|
|
self.assertTrue(normalize_value('true'))
|
|
self.assertTrue(normalize_value('TrUe'))
|
|
self.assertFalse(normalize_value('FALSE'))
|
|
self.assertFalse(normalize_value('false'))
|
|
self.assertFalse(normalize_value('FaLsE'))
|
|
self.assertTrue(normalize_value(True))
|
|
|
|
self.assertEqual('3', normalize_value('3', key="uri"))
|
|
self.assertEqual('0.3', normalize_value('0.3', key="uri"))
|
|
self.assertEqual('True', normalize_value('True', key="uri"))
|
|
self.assertEqual('False', normalize_value('False', key="uri"))
|
|
|
|
self.assertEqual('3', normalize_value('3', key="file_name"))
|
|
self.assertEqual('3', normalize_value('3', key="name"))
|
|
self.assertEqual('3', normalize_value('3', key="download_directory"))
|
|
self.assertEqual('3', normalize_value('3', key="channel_name"))
|
|
|
|
self.assertEqual(3, normalize_value('3', key="some_other_thing"))
|
|
|
|
def test_help(self):
|
|
self.assertIn(
|
|
'Usage: lbrynet [-v] [--api HOST:PORT]', self.shell(['--help'])
|
|
)
|
|
# start is special command, with separate help handling
|
|
self.assertIn(
|
|
'--share-usage-data', self.shell(['start', '--help'])
|
|
)
|
|
# publish is ungrouped command, returns usage only implicitly
|
|
self.assertIn(
|
|
'publish (<name> | --name=<name>)', self.shell(['publish'])
|
|
)
|
|
# publish is ungrouped command, with explicit --help
|
|
self.assertIn(
|
|
'Make a new name claim and publish', self.shell(['publish', '--help'])
|
|
)
|
|
# account is a group, returns help implicitly
|
|
self.assertIn(
|
|
'Return the balance of an account',
|
|
self.shell(['account'])
|
|
)
|
|
# account is a group, with explicit --help
|
|
self.assertIn(
|
|
'Return the balance of an account',
|
|
self.shell(['account', '--help'])
|
|
)
|
|
# account add is a grouped command, returns usage implicitly
|
|
self.assertIn(
|
|
'account_add (<account_name> | --account_name=<account_name>)',
|
|
self.shell(['account', 'add'])
|
|
)
|
|
# account add is a grouped command, with explicit --help
|
|
self.assertIn(
|
|
'Add a previously created account from a seed,', self.shell(['account', 'add', '--help'])
|
|
)
|
|
# help for invalid command, with explicit --help
|
|
self.assertIn(
|
|
"invalid choice: 'publish1'", self.shell(['publish1', '--help'])
|
|
)
|
|
# help for invalid command, implicit
|
|
self.assertIn(
|
|
"invalid choice: 'publish1'", self.shell(['publish1'])
|
|
)
|
|
|
|
def test_version_command(self):
|
|
self.assertEqual(
|
|
"lbrynet {lbrynet_version}".format(**get_platform()), self.shell(['--version'])
|
|
)
|
|
|
|
def test_valid_command_daemon_not_started(self):
|
|
self.assertEqual(
|
|
"Could not connect to daemon. Are you sure it's running?",
|
|
self.shell(["publish", '--name=asd', '--bid=99'])
|
|
)
|
|
|
|
def test_deprecated_command_daemon_not_started(self):
|
|
actual_output = StringIO()
|
|
with contextlib.redirect_stdout(actual_output):
|
|
main(["wallet", "balance"])
|
|
self.assertEqual(
|
|
actual_output.getvalue().strip(),
|
|
"wallet_balance is deprecated, using account_balance.\n"
|
|
"Could not connect to daemon. Are you sure it's running?"
|
|
)
|
|
|
|
|
|
class DaemonDocsTests(TestCase):
|
|
|
|
def test_can_parse_api_method_docs(self):
|
|
failures = []
|
|
for name, fn in Daemon.callable_methods.items():
|
|
try:
|
|
docopt.docopt(fn.__doc__, ())
|
|
except docopt.DocoptLanguageError as err:
|
|
failures.append(f"invalid docstring for {name}, {err.message}")
|
|
except docopt.DocoptExit:
|
|
pass
|
|
if failures:
|
|
self.fail("\n" + "\n".join(failures))
|