def test_mark_as_done1(client):
"""
Tests marking a container as done renames the container
"""
with client.application.app_context():
containers = [
ContainerFixture.new(
container_id="fastlane-job-123",
name="fastlane-job-123",
stdout="stdout",
stderr="stderr",
)
]
_, pool_mock, _ = PoolFixture.new_defaults(r"test.+",
max_running=1,
containers=containers)
task, job, execution = JobExecutionFixture.new_defaults(
container_id="fastlane-job-123")
executor = Executor(client.application, pool_mock)
executor.mark_as_done(task, job, execution)
new_name = f"defunct-fastlane-job-123"
containers[0].rename.assert_called_with(new_name)
def test_mark_as_done2(client):
"""
Tests marking a container as done raises HostUnavailableError
when docker host is not available
"""
with client.application.app_context():
containers = [
ContainerFixture.new(
container_id="fastlane-job-123",
name="fastlane-job-123",
stdout="stdout",
stderr="stderr",
)
]
_, pool_mock, _ = PoolFixture.new_defaults(r"test.+",
max_running=1,
containers=containers)
task, job, execution = JobExecutionFixture.new_defaults(
container_id="fastlane-job-123")
executor = Executor(client.application, pool_mock)
containers[0].rename.side_effect = requests.exceptions.ConnectionError(
"failed")
message = "Connection to host host:1234 failed with error: failed"
with expect.error_to_happen(HostUnavailableError, message=message):
executor.mark_as_done(task, job, execution)
def test_circuit6(client):
"""
Tests that when marking a container as done with a
docker host that's not accessible, the circuit is open and
a HostUnavailableError is raised
"""
with client.application.app_context():
client.application.config["DOCKER_CIRCUIT_BREAKER_MAX_FAILS"] = 1
task, job, execution = JobExecutionFixture.new_defaults(
docker_host="localhost",
docker_port=4567,
container_id=str(uuid4()))
pool = DockerPool(([None, ["localhost:4567"], 2], ))
executor = Executor(client.application, pool)
expect(executor.get_circuit("localhost:4567").current_state).to_equal(
"closed")
with expect.error_to_happen(HostUnavailableError):
executor.mark_as_done(task, job, execution)
expect(executor.get_circuit("localhost:4567").current_state).to_equal(
"open")