From dd6b580b43b8a678a22190a42b76d971b4352bfe Mon Sep 17 00:00:00 2001
From: Radostin Stoyanov <rstoyanov@fedoraproject.org>
Date: Tue, 12 Nov 2024 15:14:51 +0000
Subject: [PATCH] test: add get-state to mocked cuda-checkpoint tool

Signed-off-by: Radostin Stoyanov <rstoyanov@fedoraproject.org>
---
 test/cuda-checkpoint/cuda-checkpoint.c | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/test/cuda-checkpoint/cuda-checkpoint.c b/test/cuda-checkpoint/cuda-checkpoint.c
index f35a4b41df..3b7ce8b9ff 100644
--- 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", no_argument, 0, 's' },
 			{ "get-restore-tid", no_argument, 0, 'g' },
 			{ "action", required_argument, 0, 'a' },
 			{ "timeout", required_argument, 0, 't' },
@@ -31,6 +32,9 @@ int main(int argc, char *argv[])
 		case 'a':
 		case 't':
 			break;
+		case 's':
+			printf("running\n");
+			break;
 		case 'h':
 			printf("--action - execute an action");
 			break;