def test_user_provided_store_and_stdout(self):
storage_location = 'mydatastore'
sys.argv = [__file__, '-d' + storage_location, '--stdout']
configurator = Configurator()
app_config = configurator.parse_arguments()
nt.assert_is_none(app_config.bucket)
nt.assert_equal(app_config.data_store, storage_location)
nt.assert_true(app_config.write_stdout)
def test_user_provided_bucket(self):
bucket_name = 'mybucket'
web_store = '/var/www/results/Recordings/'
sys.argv = [__file__, '-b' + bucket_name]
configurator = Configurator()
app_config = configurator.parse_arguments()
nt.assert_equal(app_config.bucket, bucket_name)
nt.assert_equal(app_config.data_store, web_store)
nt.assert_false(app_config.write_stdout)
def test_user_provided_bucket_and_store(self):
bucket_name = 'mybucket'
storage_location = 'mydatastore'
sys.argv = [__file__, '-b' + bucket_name, '-d' + storage_location]
configurator = Configurator()
app_config = configurator.parse_arguments()
nt.assert_equal(app_config.bucket, bucket_name)
nt.assert_equal(app_config.data_store, storage_location)
nt.assert_false(app_config.write_stdout)
def main(vlans_filename, reqs_filename, output_filename, verbosity):
""" Main execution point for this application. """
# SETUP
devices = Devices()
devices.all = Configurator().setup_devices(vlans_filename, verbosity)
# INPUT
reqs_lst = FileOperators.get_reqs_lst(reqs_filename)
# PROCESS
_process(devices, reqs_lst)
if verbosity:
print(devices.__str__())
# OUTPUT
_create_formatted_output(devices, output_filename, verbosity)
def test_get_recordings_and_identify_kiwi(self):
from configuration import Configurator
from ornithokrites import ParallelOrnithokrites
results_test = './tests/results_test.out'
results_expected = './tests/results_expected.csv'
app_config = Configurator().parse_arguments()
ornithokrites = ParallelOrnithokrites(app_config)
handler = logging.FileHandler(results_test, 'w')
formatter = logging.Formatter('%(message)s')
handler.setFormatter(formatter)
log = logging.getLogger('log.html')
log.addHandler(handler)
ornithokrites.run()
results_expected_dict = load_expected_results(results_expected)
results_text_dict = load_results(results_test)
nt.assert_dict_contains_subset(results_expected_dict, results_text_dict)
