コード例 #1
0
    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)
コード例 #2
0
    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'])
コード例 #3
0
    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'])