def test_should_load_configuration_from_file(self, mock_loader_class, _): mock_loader = Mock(ReceiverConfigLoader) mock_loader_class.return_value = mock_loader load('abc') self.assertEqual( call('abc'), mock_loader.read_configuration_file.call_args)
def test_should_get_broadcaster_properties_from_parser(self, mock_loader_class, _): mock_loader = Mock(ReceiverConfigLoader) mock_loader.get_broadcaster_host.return_value = 'broadcaster host' mock_loader.get_broadcaster_port.return_value = 12345 mock_loader_class.return_value = mock_loader actual_configuration = load('abc') self.assertEqual(call(), mock_loader.get_broadcaster_host.call_args) self.assertEqual( 'broadcaster host', actual_configuration['broadcaster_host']) self.assertEqual(call(), mock_loader.get_broadcaster_port.call_args) self.assertEqual(12345, actual_configuration['broadcaster_port'])
def test_should_get_receiver_properties_from_parser(self, mock_loader_class): mock_loader = Mock(ReceiverConfigLoader) mock_loader.get_hostname.return_value = 'this is a name' mock_loader.get_log_filename.return_value = '/var/log/somewhere/rec.log' mock_loader.get_python_command.return_value = '/usr/bin/python' mock_loader.get_script_to_execute.return_value = '/usr/bin/yadtshell' mock_loader.get_log_filename.return_value = '/var/log/somewhere/rec.log' mock_loader.get_targets.return_value = set(['dev123']) mock_loader.get_targets_directory.return_value = '/etc/yadtshell/targets' mock_loader.get_metrics_directory.return_value = '/tmp/metrics' mock_loader.get_metrics_file.return_value = '/tmp/metrics/yrc.metrics' mock_loader_class.return_value = mock_loader actual_configuration = load('abc') self.assertEqual(call(), mock_loader.get_hostname.call_args) self.assertEqual('this is a name', actual_configuration['hostname']) self.assertEqual(call(), mock_loader.get_log_filename.call_args) self.assertEqual( '/var/log/somewhere/rec.log', actual_configuration['log_filename']) self.assertEqual(call(), mock_loader.get_python_command.call_args) self.assertEqual( '/usr/bin/python', actual_configuration['python_command']) self.assertEqual(call(), mock_loader.get_script_to_execute.call_args) self.assertEqual( '/usr/bin/yadtshell', actual_configuration['script_to_execute']) self.assertEqual(call(), mock_loader.get_targets.call_args) self.assertEqual(set(['dev123']), actual_configuration['targets']) self.assertEqual(call(), mock_loader.get_targets_directory.call_args) self.assertEqual('/etc/yadtshell/targets', actual_configuration['targets_directory']) self.assertEqual(call(), mock_loader.get_metrics_directory.call_args) self.assertEqual('/tmp/metrics', actual_configuration['metrics_directory']) self.assertEqual(call(), mock_loader.get_metrics_file.call_args) self.assertEqual('/tmp/metrics/yrc.metrics', actual_configuration['metrics_file'])
def test(self): configuration_filename = join( self.temporary_directory, 'configuration.cfg') with open(configuration_filename, 'w') as configuration_file: configuration_file.write("""[receiver] log_filename = /spam/eggs/yadtreceiver.log targets = spam, eggs targets_directory = /spam/eggs/targets script_to_execute = /spam/eggs/yadtshell python_command = /spam/eggs/python hostname = spameggs app_status_port = 42 [broadcaster] host = broadcaster.domain.tld port = 8081 """) actual_configuration = load(configuration_filename) self.assertEqual( '/spam/eggs/yadtreceiver.log', actual_configuration['log_filename']) self.assertEqual( set(['spam', 'eggs']), actual_configuration['targets']) self.assertEqual( 42, actual_configuration['app_status_port']) self.assertEqual( '/spam/eggs/targets', actual_configuration['targets_directory']) self.assertEqual( '/spam/eggs/yadtshell', actual_configuration['script_to_execute']) self.assertEqual( '/spam/eggs/python', actual_configuration['python_command']) self.assertEqual('spameggs', actual_configuration['hostname']) self.assertEqual('broadcaster.domain.tld', actual_configuration['broadcaster_host']) self.assertEqual(8081, actual_configuration['broadcaster_port'])
def test(self): configuration_filename = join(self.temporary_directory, 'configuration.cfg') with open(configuration_filename, 'w') as configuration_file: configuration_file.write("""[receiver] log_filename = /spam/eggs/yadtreceiver.log targets = spam, eggs targets_directory = /spam/eggs/targets script_to_execute = /spam/eggs/yadtshell python_command = /spam/eggs/python hostname = spameggs app_status_port = 42 [broadcaster] host = broadcaster.domain.tld port = 8081 """) actual_configuration = load(configuration_filename) self.assertEqual('/spam/eggs/yadtreceiver.log', actual_configuration['log_filename']) self.assertEqual(set(['spam', 'eggs']), actual_configuration['targets']) self.assertEqual(42, actual_configuration['app_status_port']) self.assertEqual('/spam/eggs/targets', actual_configuration['targets_directory']) self.assertEqual('/spam/eggs/yadtshell', actual_configuration['script_to_execute']) self.assertEqual('/spam/eggs/python', actual_configuration['python_command']) self.assertEqual('spameggs', actual_configuration['hostname']) self.assertEqual('broadcaster.domain.tld', actual_configuration['broadcaster_host']) self.assertEqual(8081, actual_configuration['broadcaster_port'])