def test_should_return_metrics_file(self): mock_loader = Mock(ReceiverConfigLoader) mock_loader.get_metrics_directory.return_value = '/tmp/metrics' actual_metrics_file = ReceiverConfigLoader.get_metrics_file( mock_loader) self.assertEqual('/tmp/metrics/yrc.metrics', actual_metrics_file)
def test_should_return_targets(self): mock_loader = Mock(ReceiverConfigLoader) mock_parser = Mock(YadtConfigParser) mock_parser.get_option_as_set.return_value = set(['abc', 'def']) mock_loader._parser = mock_parser actual_targets = ReceiverConfigLoader.get_targets(mock_loader) self.assertEqual(set(['abc', 'def']), actual_targets) self.assertEqual( call(SECTION_RECEIVER, 'targets', DEFAULT_TARGETS), mock_parser.get_option_as_set.call_args)
def test_should_return_broadcaster_port_option(self): mock_loader = Mock(ReceiverConfigLoader) mock_parser = Mock(YadtConfigParser) mock_parser.get_option_as_int.return_value = 42 mock_loader._parser = mock_parser actual_app_status_port = ReceiverConfigLoader.get_app_status_port( mock_loader) self.assertEqual(42, actual_app_status_port) self.assertEqual( call(SECTION_RECEIVER, 'app_status_port', DEFAULT_APP_STATUS_PORT), mock_parser.get_option_as_int.call_args)
def test_should_return_broadcaster_port_option(self): mock_loader = Mock(ReceiverConfigLoader) mock_parser = Mock(YadtConfigParser) mock_parser.get_option_as_int.return_value = 8081 mock_loader._parser = mock_parser actual_broadcaster_port = ReceiverConfigLoader.get_broadcaster_port( mock_loader) self.assertEqual(8081, actual_broadcaster_port) self.assertEqual( call(SECTION_BROADCASTER, 'port', DEFAULT_BROADCASTER_PORT), mock_parser.get_option_as_int.call_args)
def test_should_return_broadcaster_host_option(self): mock_loader = Mock(ReceiverConfigLoader) mock_parser = Mock(YadtConfigParser) mock_parser.get_option.return_value = 'broadcaster.host' mock_loader._parser = mock_parser actual_broadcaster_host = ReceiverConfigLoader.get_broadcaster_host( mock_loader) self.assertEqual('broadcaster.host', actual_broadcaster_host) self.assertEqual( call(SECTION_BROADCASTER, 'host', DEFAULT_BROADCASTER_HOST), mock_parser.get_option.call_args)
def test_should_return_metrics_directory(self): mock_loader = Mock(ReceiverConfigLoader) mock_parser = Mock(YadtConfigParser) mock_parser.get_option.return_value = '/tmp/metrics' mock_loader._parser = mock_parser actual_metrics_directory = ReceiverConfigLoader.get_metrics_directory( mock_loader) self.assertEqual('/tmp/metrics', actual_metrics_directory) self.assertEqual( call(SECTION_RECEIVER, 'metrics_directory', None), mock_parser.get_option.call_args)
def test_should_delegate_read_configuration_file(self): mock_loader = Mock(ReceiverConfigLoader) mock_parser = Mock(YadtConfigParser) mock_parser.read_configuration_file.return_value = {'spam': 'eggs'} mock_loader._parser = mock_parser actual_configuration = ReceiverConfigLoader.read_configuration_file( mock_loader, 'filename.cfg') self.assertEqual({'spam': 'eggs'}, actual_configuration) self.assertEqual( call('filename.cfg'), mock_parser.read_configuration_file.call_args)
def test_should_return_targets_directory(self): mock_loader = Mock(ReceiverConfigLoader) mock_parser = Mock(YadtConfigParser) mock_parser.get_option.return_value = '/directory/with/targets' mock_loader._parser = mock_parser actual_targets_directory = ReceiverConfigLoader.get_targets_directory( mock_loader) self.assertEqual('/directory/with/targets', actual_targets_directory) self.assertEqual( call(SECTION_RECEIVER, 'targets_directory', DEFAULT_TARGETS_DIRECTORY), mock_parser.get_option.call_args)
def test_should_return_script_to_execute(self): mock_loader = Mock(ReceiverConfigLoader) mock_parser = Mock(YadtConfigParser) mock_parser.get_option.return_value = '/script/to/execute' mock_loader._parser = mock_parser actual_script_to_execute = ReceiverConfigLoader.get_script_to_execute( mock_loader) self.assertEqual('/script/to/execute', actual_script_to_execute) self.assertEqual( call(SECTION_RECEIVER, 'script_to_execute', DEFAULT_SCRIPT_TO_EXECUTE), mock_parser.get_option.call_args)
def test_should_return_python_command(self): mock_loader = Mock(ReceiverConfigLoader) mock_parser = Mock(YadtConfigParser) mock_parser.get_option.return_value = '/spam/eggs/python' mock_loader._parser = mock_parser actual_python_command = ReceiverConfigLoader.get_python_command( mock_loader) self.assertEqual('/spam/eggs/python', actual_python_command) self.assertEqual( call(SECTION_RECEIVER, 'python_command', DEFAULT_PYTHON_COMMAND), mock_parser.get_option.call_args)
def test_should_return_log_filename(self): mock_loader = Mock(ReceiverConfigLoader) mock_parser = Mock(YadtConfigParser) mock_parser.get_option.return_value = 'filename.log' mock_loader._parser = mock_parser actual_log_filename = ReceiverConfigLoader.get_log_filename( mock_loader) self.assertEqual('filename.log', actual_log_filename) self.assertEqual( call(SECTION_RECEIVER, 'log_filename', DEFAULT_LOG_FILENAME), mock_parser.get_option.call_args)
def test_should_return_hostname_from_receiver_section(self, mock_socket): mock_socket.gethostname.return_value = 'localhost' mock_loader = Mock(ReceiverConfigLoader) mock_parser = Mock(YadtConfigParser) mock_parser.get_option.return_value = 'actual.hostname' mock_loader._parser = mock_parser actual_hostname = ReceiverConfigLoader.get_hostname(mock_loader) self.assertEqual('actual.hostname', actual_hostname) self.assertEqual(call(), mock_socket.gethostname) self.assertEqual( call(SECTION_RECEIVER, 'hostname', 'localhost'), mock_parser.get_option.call_args)
def test_should_create_instance_of_SafeConfigParser(self): parser = ReceiverConfigLoader() name_of_type = parser._parser.__class__.__name__ self.assertEqual('YadtConfigParser', name_of_type)