def test_simple_config(): with temporary_file() as fp: fp.write(MESOS_CONFIG) fp.flush() proxy_config1 = AuroraConfig.load(fp.name) proxy_config2 = AuroraConfig.load(fp.name, name="hello_world") assert proxy_config1.job() assert proxy_config1._job == proxy_config2._job assert proxy_config1._job == REIFIED_CONFIG assert proxy_config1.name() == "hello_world" assert proxy_config1.role() == "john_doe" assert proxy_config1.cluster() == "smf1-test" assert proxy_config1.ports() == set()
def test_simple_config(): with temporary_file() as fp: fp.write(MESOS_CONFIG) fp.flush() proxy_config1 = AuroraConfig.load(fp.name) proxy_config2 = AuroraConfig.load(fp.name, name="hello_world") assert proxy_config1.job() assert proxy_config1._job == proxy_config2._job assert proxy_config1._job == REIFIED_CONFIG assert proxy_config1.name() == 'hello_world' assert proxy_config1.role() == 'john_doe' assert proxy_config1.cluster() == 'smf1-test' assert proxy_config1.ports() == set()
def test_docker_binding_throws(self, mock_resolve): mock_resolve.side_effect = Exception('mock resolve failure') binding_helper.unregister_all() BindingHelper.register(DockerBindingHelper()) with temporary_file() as fp: fp.write(DOCKER_BINDING_CONFIG) fp.flush() with CLUSTERS.patch(TEST_CLUSTERS): cfg = AuroraConfig.load(fp.name) with pytest.raises(Exception): binding_helper.apply_all(cfg) assert mock_resolve.mock_calls == [call(TEST_CLUSTER, 'some/name', 'some.tag')]
def test_docker_binding(self, mock_resolve): image_reference = 'registry.example.com/some/repo@some:digest' mock_resolve.return_value = image_reference binding_helper.unregister_all() BindingHelper.register(DockerBindingHelper()) with temporary_file() as fp: fp.write(DOCKER_BINDING_CONFIG) fp.flush() with CLUSTERS.patch(TEST_CLUSTERS): cfg = AuroraConfig.load(fp.name) binding_helper.apply_all(cfg) assert cfg.job().taskConfig.container.docker.image == image_reference assert mock_resolve.mock_calls == [call(TEST_CLUSTER, 'some/name', 'some.tag')]
def test_docker_binding_throws(self, mock_resolve): mock_resolve.side_effect = Exception('mock resolve failure') binding_helper.unregister_all() BindingHelper.register(DockerBindingHelper()) with temporary_file() as fp: fp.write(DOCKER_BINDING_CONFIG) fp.flush() with CLUSTERS.patch(TEST_CLUSTERS): cfg = AuroraConfig.load(fp.name) with pytest.raises(Exception): binding_helper.apply_all(cfg) assert mock_resolve.mock_calls == [ call(TEST_CLUSTER, 'some/name', 'some.tag') ]
def test_docker_binding(self, mock_resolve): image_reference = 'registry.example.com/some/repo@some:digest' mock_resolve.return_value = image_reference binding_helper.unregister_all() BindingHelper.register(DockerBindingHelper()) with temporary_file() as fp: fp.write(DOCKER_BINDING_CONFIG) fp.flush() with CLUSTERS.patch(TEST_CLUSTERS): cfg = AuroraConfig.load(fp.name) binding_helper.apply_all(cfg) assert cfg.job( ).taskConfig.container.docker.image == image_reference assert mock_resolve.mock_calls == [ call(TEST_CLUSTER, 'some/name', 'some.tag') ]
def test_empty_config(): with pytest.raises(AuroraConfig.InvalidConfig): with temporary_file() as fp: fp.write(UNDERSPECIFIED_MESOS_CONFIG) fp.flush() AuroraConfig.load(fp.name)
def write_and_load_config(role): with temporary_file() as fp: fp.write(GENERIC_CONFIG) fp.flush() return AuroraConfig.load(fp.name, name='hello_world', select_role=role)