From 6011f85268d52d662bf9c499a499b03a6382b3b1 Mon Sep 17 00:00:00 2001
From: Kouame Behouba Manasse <behouba@gmail.com>
Date: Sun, 21 Jul 2024 21:44:16 -0400
Subject: [PATCH] test: add test cases for `--search` and `--context` flags in
 `memparse`

Signed-off-by: Kouame Behouba Manasse <behouba@gmail.com>
---
 test/checkpointctl.bats | 39 +++++++++++++++++++++++++++++++++++++++
 1 file changed, 39 insertions(+)

diff --git a/test/checkpointctl.bats b/test/checkpointctl.bats
index 480472e1..f48d8153 100644
--- a/test/checkpointctl.bats
+++ b/test/checkpointctl.bats
@@ -577,6 +577,45 @@ function teardown() {
 	[[ ${lines[3]} == *"....H...H.../..H"* ]]
 }
 
+@test "Run checkpointctl memparse with --search=PATH and --context=-1" {
+	cp data/config.dump \
+		data/spec.dump "$TEST_TMP_DIR1"
+	mkdir "$TEST_TMP_DIR1"/checkpoint
+	cp test-imgs/pstree.img \
+		test-imgs/pagemap-*.img \
+		test-imgs/pages-*.img "$TEST_TMP_DIR1"/checkpoint
+	( cd "$TEST_TMP_DIR1" && tar cf "$TEST_TMP_DIR2"/test.tar . )
+	checkpointctl memparse --search=PATH --context=-1 "$TEST_TMP_DIR2"/test.tar --pid=1
+	[ "$status" -eq 1 ]
+	[[ ${lines[0]} == *"context size cannot be negative"* ]]
+}
+
+@test "Run checkpointctl memparse with --search=NON_EXISTING_PATTERN" {
+	cp data/config.dump \
+		data/spec.dump "$TEST_TMP_DIR1"
+	mkdir "$TEST_TMP_DIR1"/checkpoint
+	cp test-imgs/pstree.img \
+		test-imgs/pagemap-*.img \
+		test-imgs/pages-*.img "$TEST_TMP_DIR1"/checkpoint
+	( cd "$TEST_TMP_DIR1" && tar cf "$TEST_TMP_DIR2"/test.tar . )
+	checkpointctl memparse --search=NON_EXISTING_PATTERN "$TEST_TMP_DIR2"/test.tar --pid=1
+	[ "$status" -eq 0 ]
+	[[ ${lines[0]} == *"No matches"* ]]
+}
+
+@test "Run checkpointctl memparse with --search=PATH and --context=10 flags" {
+	cp data/config.dump \
+		data/spec.dump "$TEST_TMP_DIR1"
+	mkdir "$TEST_TMP_DIR1"/checkpoint
+	cp test-imgs/pstree.img \
+		test-imgs/pagemap-*.img \
+		test-imgs/pages-*.img "$TEST_TMP_DIR1"/checkpoint
+	( cd "$TEST_TMP_DIR1" && tar cf "$TEST_TMP_DIR2"/test.tar . )
+	checkpointctl memparse --search=PATH --context=10 "$TEST_TMP_DIR2"/test.tar --pid=1
+	[ "$status" -eq 0 ]
+	[[ ${lines[3]} == *"PATH"* ]]
+}
+
 @test "Run checkpointctl memparse with tar file and invalid PID" {
 	cp data/config.dump \
 		data/spec.dump "$TEST_TMP_DIR1"