test: add get-state to mocked cuda-checkpoint tool

Signed-off-by: Radostin Stoyanov <[email protected]>
checkpoint-restore · Nov 12, 2024 · 7d9b7f0 · 7d9b7f0
1 parent dcdd00c
commit 7d9b7f0
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/test/cuda-checkpoint/cuda-checkpoint.c b/test/cuda-checkpoint/cuda-checkpoint.c
@@ -11,6 +11,7 @@ int main(int argc, char *argv[])
 		int option_index = 0;
 		static struct option long_options[] = {
 			{ "pid", required_argument, 0, 'p' },
+			{ "get-state", required_argument, 0, 's' },
 			{ "get-restore-tid", no_argument, 0, 'g' },
 			{ "action", required_argument, 0, 'a' },
 			{ "timeout", required_argument, 0, 't' },
@@ -30,6 +31,7 @@ int main(int argc, char *argv[])
 		case 'g':
 		case 'a':
 		case 't':
+		case 's':
 			break;
 		case 'h':
 			printf("--action - execute an action");