update e2e

e2e-tests/test_sessionjob_operations.sh

@@ -67,6 +67,16 @@ assert_available_slots 1 $CLUSTER_ID
 
 echo "Successfully run the sessionjob savepoint upgrade test"
 
+# Testing last-state mode upgrade
+# Update the FlinkSessionJob and trigger the last-state upgrade
+kubectl patch sessionjob ${SESSION_JOB_NAME} --type merge --patch '{"spec":{"job": {"parallelism": 2, "upgradeMode": last-state } } }'
+
+# Check the job was restarted with the new parallelism
+wait_for_status $SESSION_JOB_IDENTIFIER '.status.jobStatus.state' RUNNING ${TIMEOUT} || exit 1
+assert_available_slots 0 $CLUSTER_ID
+
+echo "Successfully run the sessionjob savepoint upgrade test"
+
 # Test Operator restart
 echo "Delete session job " + $SESSION_JOB_NAME
 kubectl delete flinksessionjob $SESSION_JOB_NAME