From 2c774fb207df41d3514080a940def1ea28170ca6 Mon Sep 17 00:00:00 2001
From: Sumedh Alok Sharma <sumsharma@microsoft.com>
Date: Fri, 6 Sep 2024 13:04:05 +0530
Subject: [PATCH] ci: Add tests for CopyFile api.

This commit introduces test cases for testing
CopyFile API using kata-agent-ctl with improved command
semantics and handling.
- copy a file to /run/kata-containers
- copy symlink to /run/kata-containers
- copy directory to /run/kata-containers
- copy file to /tmp
- copy large file to /run/kata-containers

Signed-off-by: Sumedh Alok Sharma <sumsharma@microsoft.com>
---
 .../kata-agent-apis/api-tests/dummy.bats      | 20 ------
 .../api-tests/test_copy_file.bats             | 67 +++++++++++++++++++
 2 files changed, 67 insertions(+), 20 deletions(-)
 delete mode 100755 tests/functional/kata-agent-apis/api-tests/dummy.bats
 create mode 100755 tests/functional/kata-agent-apis/api-tests/test_copy_file.bats

diff --git a/tests/functional/kata-agent-apis/api-tests/dummy.bats b/tests/functional/kata-agent-apis/api-tests/dummy.bats
deleted file mode 100755
index 3659415247..0000000000
--- a/tests/functional/kata-agent-apis/api-tests/dummy.bats
+++ /dev/null
@@ -1,20 +0,0 @@
-#!/usr/bin/env bats
-
-# Copyright (c) 2024 Microsoft Corporation
-#
-# SPDX-License-Identifier: Apache-2.0
-
-load "${BATS_TEST_DIRNAME}/../../../common.bash"
-load "${BATS_TEST_DIRNAME}/../setup_common.sh"
-
-setup_file() {
-    info "setup"
-}
-
-@test "Dummy test" {
-    info "placeholder"
-}
-
-teardown_file() {
-    info "teardown"
-}
diff --git a/tests/functional/kata-agent-apis/api-tests/test_copy_file.bats b/tests/functional/kata-agent-apis/api-tests/test_copy_file.bats
new file mode 100755
index 0000000000..d9875683a2
--- /dev/null
+++ b/tests/functional/kata-agent-apis/api-tests/test_copy_file.bats
@@ -0,0 +1,67 @@
+#!/usr/bin/env bats
+
+# Copyright (c) 2024 Microsoft Corporation
+#
+# SPDX-License-Identifier: Apache-2.0
+
+load "${BATS_TEST_DIRNAME}/../../../common.bash"
+load "${BATS_TEST_DIRNAME}/../setup_common.sh"
+
+setup_file() {
+    info "setup"
+}
+
+@test "Test CopyFile API: Copy a file to /run/kata-containers" {
+    info "Copy file to /run/kata-containers"
+    src_file=$(mktemp)
+    local cmds=()
+    cmds+=("-c 'CopyFile json://{\"src\": \"$src_file\", \"dest\":\"/run/kata-containers/foo\"}'")
+    run_agent_ctl "${cmds[@]}"
+    rm $src_file
+}
+
+@test "Test CopyFile API: Copy a symlink to /run/kata-containers" {
+    info "Copy symlink to /run/kata-containers"
+    src_file=$(mktemp)
+    link_file="/tmp/link"
+    ln -s $src_file $link_file
+    local cmds=()
+    cmds+=("-c 'CopyFile json://{\"src\": \"$link_file\", \"dest\":\"/run/kata-containers/link\"}'")
+    run_agent_ctl "${cmds[@]}"
+    unlink $link_file
+    rm $src_file
+}
+
+@test "Test CopyFile API: Copy a directory to /run/kata-containers" {
+    info "Copy directory to /run/kata-containers"
+    src_dir=$(mktemp -d)
+    local cmds=()
+    cmds+=("-c 'CopyFile json://{\"src\": \"$src_dir\", \"dest\":\"/run/kata-containers/dir\"}'")
+    run_agent_ctl "${cmds[@]}"
+    rmdir $src_dir
+}
+
+@test "Test CopyFile API: Copy a file to an unallowed destination" {
+    info "Copy file to /tmp"
+    src_file=$(mktemp)
+    local cmds=()
+    cmds+=("-c 'CopyFile json://{\"src\": \"$src_file\", \"dest\":\"/tmp/foo\"}'")
+    run run_agent_ctl "${cmds[@]}"
+    [ "$status" -ne 0 ]
+    rm $src_file
+}
+
+@test "Test CopyFile API: Copy a large file to /run/kata-containers" {
+    info "Copy large file to /run/kata-containers"
+    src_file="/tmp/large_file_2M.txt"
+    truncate -s 2M $src_file
+    local cmds=()
+    cmds+=("-c 'CopyFile json://{\"src\": \"$src_file\", \"dest\":\"/run/kata-containers/large_file_2M.txt\"}'")
+    run_agent_ctl "${cmds[@]}"
+    rm $src_file
+}
+
+teardown_file() {
+    info "teardown"
+    sudo rm -r /run/kata-containers/ || echo "Failed to clean /run/kata-containers"
+}