async def test_success(job_with_adapter):
complete_job_mock = AsyncMock()
job_with_adapter.zeebe_adapter.complete_job = complete_job_mock
await job_with_adapter.set_success_status()
complete_job_mock.assert_called_with(job_key=job_with_adapter.key,
variables=job_with_adapter.variables)
async def test_failure(job_with_adapter):
fail_job_mock = AsyncMock()
job_with_adapter.zeebe_adapter.fail_job = fail_job_mock
message = str(uuid4())
await job_with_adapter.set_failure_status(message)
fail_job_mock.assert_called_with(job_key=job_with_adapter.key,
retries=job_with_adapter.retries - 1,
message=message)
async def test_error(job_with_adapter):
throw_error_mock = AsyncMock()
job_with_adapter.zeebe_adapter.throw_error = throw_error_mock
message = str(uuid4())
await job_with_adapter.set_error_status(message)
throw_error_mock.assert_called_with(job_key=job_with_adapter.key,
message=message,
error_code="")
