def test__pause_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.pause(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")
def test__pause_resume_initialized_update(stateless_job, in_place):
stateless_job.create()
stateless_job.wait_for_state(goal_state="RUNNING")
old_pod_infos = stateless_job.query_pods()
old_instance_zero_spec = stateless_job.get_pod(0).get_pod_spec()
update = StatelessUpdate(stateless_job,
batch_size=1,
updated_job_file=UPDATE_STATELESS_JOB_SPEC)
update.create(in_place=in_place)
# immediately pause the update, so the update may still be INITIALIZED
update.pause()
update.wait_for_state(goal_state="PAUSED")
update.resume()
update.wait_for_state(goal_state="SUCCEEDED")
new_pod_infos = stateless_job.query_pods()
new_instance_zero_spec = stateless_job.get_pod(0).get_pod_spec()
assert_pod_id_changed(old_pod_infos, new_pod_infos)
assert_pod_spec_changed(old_instance_zero_spec, new_instance_zero_spec)
def test__pause_resume__update(stateless_job, in_place):
stateless_job.create()
stateless_job.wait_for_state(goal_state="RUNNING")
old_pod_infos = stateless_job.query_pods()
old_instance_zero_spec = stateless_job.get_pod(0).get_pod_spec()
update = StatelessUpdate(stateless_job,
batch_size=1,
updated_job_file=UPDATE_STATELESS_JOB_SPEC)
update.create(in_place=in_place)
# sleep for 1 sec so update can begin to roll forward
time.sleep(1)
update.pause()
update.wait_for_state(goal_state="PAUSED")
update.resume()
update.wait_for_state(goal_state="SUCCEEDED")
new_pod_infos = stateless_job.query_pods()
new_instance_zero_spec = stateless_job.get_pod(0).get_pod_spec()
assert_pod_id_changed(old_pod_infos, new_pod_infos)
assert_pod_spec_changed(old_instance_zero_spec, new_instance_zero_spec)