예제 #1
0
    def test_generate_logging_config_no_file(self, open_mock, spec):
        generated_config = {"foo": "bar"}
        config_generator = Mock(return_value=generated_config)

        logging_config = bg_utils.generate_logging_config_file(
            spec, config_generator, [])
        config_generator.assert_called_with("INFO", None)
        assert logging_config == generated_config
        assert open_mock.called is False
예제 #2
0
    def test_generate_logging_config_no_mock(self, tmpdir, spec):
        config_path = tmpdir.join("logging-config.json")
        generated_config = {"foo": "bar"}
        config_generator = Mock(return_value=generated_config)

        logging_config = bg_utils.generate_logging_config_file(
            spec, config_generator,
            ["--log-config", str(config_path)])

        assert logging_config == generated_config
예제 #3
0
    def test_generate_logging_config(self, open_mock, spec):
        fake_file = Mock()
        fake_file.__exit__ = Mock()
        fake_file.__enter__ = Mock(return_value=fake_file)
        open_mock.return_value = fake_file
        generated_config = {"foo": "bar"}
        config_generator = Mock(return_value=generated_config)

        logging_config = bg_utils.generate_logging_config_file(
            spec, config_generator,
            ["--log-config-file", "/path/to/log/config"])
        assert logging_config == generated_config
        assert open_mock.called is True
예제 #4
0
def generate_logging_config():
    spec = YapconfSpec(SPECIFICATION, env_prefix="BG_")
    bg_utils.generate_logging_config_file(spec, get_default_logging_config,
                                          sys.argv[1:])