def main(): """ Setup consumer """ config = loader.load_config() if config.version: show_version() start_consumer(config)
def main(): """ Setup consumer """ config = loader.load_config() if config.benchmark: print("Starting in benchmark mode. Stand by while creating sample messages.") create_sample_messages(config) start_consumer(config)
def main(): """ Setup consumer """ config = loader.load_config() if config.version: show_version() if config.benchmark: print("Starting in benchmark mode. Stand by while creating sample messages.") create_sample_messages(config) start_consumer(config)
def test_cli_overwrite(self): # Fake commandline arguments # Argparse returns a namespace, not a dictionary fake_args = argparse.Namespace() fake_args.influxdb_use_ssl = False argparse.ArgumentParser.parse_args = MagicMock(return_value=fake_args) # Fake default config default_config.DEFAULT_CONFIG = MagicMock(return_value={'influxdb_use_ssl': True}) config = loader.load_config() # Check if the default setting got overwritten self.assertEqual(config.influxdb_use_ssl, False)