def test_get_container_id_name_not_found(): mock_docker_client = mock.MagicMock(spec_set=docker.Client) fake_containers = [ {'Names': ['/paasta_local_run_1'], 'Id': '11111'}, {'Names': ['/paasta_local_run_2'], 'Id': '22222'}, ] mock_docker_client.containers = mock.MagicMock( spec_set=docker.Client, return_value=fake_containers, ) container_name = 'paasta_local_run_DOES_NOT_EXIST' with raises(LostContainerException): get_container_id(mock_docker_client, container_name)
def test_get_container_id(): mock_docker_client = mock.MagicMock(spec_set=docker.Client) fake_containers = [ {'Names': ['/paasta_local_run_1'], 'Id': '11111'}, {'Names': ['/paasta_local_run_2'], 'Id': '22222'}, ] mock_docker_client.containers = mock.MagicMock( spec_set=docker.Client, return_value=fake_containers, ) container_name = 'paasta_local_run_2' expected = '22222' actual = get_container_id(mock_docker_client, container_name) assert actual == expected