def test_build_initial_docker_image_error(tmpdir, output):
docker_client = flexmock()
docker_client.should_receive("build").with_args(fileobj=io.BytesIO, rm=True).and_return([json.dumps(output)])
flexmock(exec_).should_receive("_initial_dockerfile")
with pytest.raises(exec_.DockerError):
exec_._build_initial_docker_image(docker_client, {})
def test_build_initial_docker_image(tmpdir):
docker_client = flexmock()
docker_client.should_receive("build").with_args(fileobj=io.BytesIO, rm=True).and_return(
[json.dumps({"stream": "Successfully built 0123456789ab"}), json.dumps({"status": "ok"})]
)
flexmock(exec_).should_receive("_initial_dockerfile")
exec_._build_initial_docker_image(docker_client, {})
