def main():
parser = create_parser()
args = parser.parse_args()
conf = get_config(args)
_adjust_log_level(args, conf)
conf_valid, conf_errors = validate_config(conf)
if not conf_valid:
for e in conf_errors:
log.critical(e)
exit(1)
configure_logging(conf)
def_args = [args, conf]
def_kwargs = {}
known_commands = {
'cleanup': {'f': sbws.core.cleanup.main,
'a': def_args, 'kw': def_kwargs},
'scanner': {'f': sbws.core.scanner.main,
'a': def_args, 'kw': def_kwargs},
'generate': {'f': sbws.core.generate.main,
'a': def_args, 'kw': def_kwargs},
'init': {'f': sbws.core.init.main,
'a': def_args, 'kw': def_kwargs},
'stats': {'f': sbws.core.stats.main,
'a': def_args, 'kw': def_kwargs},
}
try:
if args.command not in known_commands:
parser.print_help()
else:
comm = known_commands[args.command]
exit(comm['f'](*comm['a'], **comm['kw']))
except KeyboardInterrupt:
print('')
def test_stats_fresh_result(tmpdir, capsys, caplog):
'''
An initialized .sbws directory with a fresh error result should have some
boring stats and exit cleanly
'''
caplog.set_level(logging.DEBUG)
init_directory(tmpdir)
add_single_fresh_result(tmpdir)
p = create_parser()
args = p.parse_args(
'-d {} --log-level DEBUG stats --error-types'.format(tmpdir).split())
conf = get_config(args)
sbws.core.stats.main(args, conf)
captured = capsys.readouterr()
lines = captured.out.strip().split('\n')
needed_output_lines = [
'1 relays have recent results',
'Mean 0.00 successful measurements per relay',
'0 success results and 1 error results',
]
for needed_line in needed_output_lines:
assert needed_line in lines
lines = [l.getMessage() for l in caplog.records]
needed_log_lines = [
'Keeping 1/1 read lines from {}/{}/{}.txt'.format(
tmpdir, 'datadir',
datetime.utcfromtimestamp(time.time()).date()),
'Keeping 1/1 results after removing old ones',
]
for needed_line in needed_log_lines:
assert needed_line in lines
def test_stats_stale_result(tmpdir, caplog):
'''
An initialized .sbws directory with no fresh results should say so and
exit cleanly
'''
init_directory(tmpdir)
add_single_stale_result(tmpdir)
p = create_parser()
args = p.parse_args('-d {} --log-level DEBUG stats'.format(tmpdir).split())
conf = get_config(args)
sbws.core.stats.main(args, conf)
assert 'No fresh results' == caplog.records[-1].getMessage()
def test_stats_uninitted(tmpdir, caplog):
'''
An un-initialized .sbws directory should fail hard and exit immediately
'''
p = create_parser()
args = p.parse_args('-d {} --log-level DEBUG stats'.format(tmpdir).split())
conf = get_config(args)
try:
sbws.core.stats.main(args, conf)
except SystemExit as e:
assert e.code == 1
else:
assert None, 'Should have failed'
assert 'Sbws isn\'t initialized. Try sbws init' == \
caplog.records[-1].getMessage()
def test_stats_initted(tmpdir, caplog):
'''
An initialized but rather empty .sbws directory should fail about missing
~/.sbws/datadir
'''
init_directory(tmpdir)
p = create_parser()
args = p.parse_args('-d {} --log-level DEBUG stats'.format(tmpdir).split())
conf = get_config(args)
try:
sbws.core.stats.main(args, conf)
except SystemExit as e:
assert e.code == 1
else:
assert None, 'Should have failed'
assert '{}/datadir does not exist'.format(tmpdir) == \
caplog.records[-1].getMessage()
def test_stats_fresh_results(time_mock, tmpdir, capsys, caplog):
'''
An initialized .sbws directory with a fresh error and fresh success should
have some exciting stats and exit cleanly
'''
caplog.set_level(logging.DEBUG)
init_directory(tmpdir)
start = 1524769441
time_mock.side_effect = monotonic_time(start=start)
add_two_fresh_results(tmpdir, start - 1)
p = create_parser()
args = p.parse_args(
'-d {} --log-level DEBUG stats --error-types'.format(tmpdir).split())
conf = get_config(args)
sbws.core.stats.main(args, conf)
needed_output_lines = [
'1 relays have recent results',
'1 success results and 1 error results',
'Mean 1.00 successful measurements per relay',
'1/2 (50.00%) results were error-misc',
]
captured = capsys.readouterr()
lines = captured.out.strip().split('\n')
for needed_line in needed_output_lines:
assert needed_line in lines
lines = [l.getMessage() for l in caplog.records]
needed_log_lines = [
'Keeping 2/2 read lines from {}/{}/{}.txt'.format(
tmpdir, 'datadir',
datetime.utcfromtimestamp(time.time()).date()),
'Keeping 2/2 results after removing old ones',
'Found a _ResultType.Error for the first time',
'Found a _ResultType.Success for the first time',
]
for needed_line in needed_log_lines:
assert needed_line in lines
def init_directory(dname):
p = create_parser()
args = p.parse_args('-d {} --log-level DEBUG init'.format(dname).split())
conf = get_config(args)
sbws.core.init.main(args, conf)
def parser():
return create_parser()
