def test__abort_update(stateless_job, in_place):
stateless_job.create()
stateless_job.wait_for_state(goal_state="RUNNING")
update = StatelessUpdate(
stateless_job,
updated_job_file=UPDATE_STATELESS_JOB_UPDATE_AND_ADD_INSTANCES_SPEC,
batch_size=1,
)
update.create(in_place=in_place)
update.wait_for_state(goal_state="ROLLING_FORWARD")
update.abort()
update.wait_for_state(goal_state="ABORTED")
def test__abort_update_bad_version(stateless_job, in_place):
stateless_job.create()
stateless_job.wait_for_state(goal_state="RUNNING")
update = StatelessUpdate(
stateless_job,
updated_job_file=UPDATE_STATELESS_JOB_UPDATE_AND_ADD_INSTANCES_SPEC,
batch_size=1,
)
update.create(in_place=in_place)
try:
update.abort(entity_version="1-2-3")
except grpc.RpcError as e:
assert e.code() == grpc.StatusCode.ABORTED
assert INVALID_ENTITY_VERSION_ERR_MESSAGE in e.details()
return
raise Exception("entity version mismatch error not received")