backup/warcprox
1
0
Fork 0
mirror of https://github.com/internetarchive/warcprox.git synced 2025-01-18 13:22:09 +01:00
Code Issues Projects Releases Activity

clean up small items, typos, change command name, clean up tests in re: exit code testing.

Browse Source
This commit is contained in:
James Kafader 2017-10-03 16:13:42 -07:00
parent df7c0b8e32
commit e1b4153712
3 changed files with 8 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
doublethink/cli.py
View File

@ -1,6 +1,6 @@
#!/usr/bin/env python
'''
doublethink/orm.py - rethinkdb ORM Command Line Interface
doublethink/cli.py - doublethink Command Line Tools
Copyright (C) 2017 Internet Archive
@ -30,7 +30,7 @@ def purge_stale_services(argv=None):
argv = argv or sys.argv
arg_parser = argparse.ArgumentParser(
prog=os.path.basename(argv[0]),
description='purge-stale-services: utility to periodically purge stale entries from the "services" table.')
description='doublethink-purge-stale-services: utility to periodically purge stale entries from the "services" table.')
arg_parser.add_argument("-d", "--rethinkdb-db", required=True,
dest="database",
@ -55,4 +55,4 @@ def purge_stale_services(argv=None):
rethinker = doublethink.Rethinker(servers=args.servers, db=args.database)
registry = doublethink.services.ServiceRegistry(rethinker)
registry.purge_stale_services()
sys.exit(0)
return 0

2
setup.py
View File

@ -20,7 +20,7 @@ setuptools.setup(
'README.rst', mode='r', encoding='utf-8').read(),
entry_points={
'console_scripts': [
'purge-stale-services=doublethink.cli:purge_stale_services',
'doublethink-purge-stale-services=doublethink.cli:purge_stale_services',
]
},
)

7
tests/test_cli.py
View File

@ -50,14 +50,15 @@ def rr():
def test_cli(capsys, rr):
entrypoint = pkg_resources.get_entry_map(
'doublethink')['console_scripts']['purge-stale-services']
'doublethink')['console_scripts']['doublethink-purge-stale-services']
callable = entrypoint.resolve()
with pytest.raises(SystemExit) as exit:
callable(['purge-stale-services'])
callable(['doublethink-purge-stale-services'])
print(dir(exit))
assert exit.value.code != 0
out, err = capsys.readouterr()
with pytest.raises(SystemExit) as exit:
callable(['purge-stale-services', '-d', 'test'])
# this wrap with sys.exit matches what occurs in the generated command
sys.exit(callable(['doublethink-purge-stale-services', '-d', 'test']))
assert exit.value.code == 0
out, err = capsys.readouterr()