From 5c24ded1b34d126f7e5199ebde222311f855cff0 Mon Sep 17 00:00:00 2001
From: Mengyi Wang <mengyi.wang@canonical.com>
Date: Fri, 17 Jun 2022 15:43:23 +0200
Subject: [PATCH] Add unit test for copyDir and copyFile

---
 utils_test.go | 25 +++++++++++++++++++++++++
 1 file changed, 25 insertions(+)

diff --git a/utils_test.go b/utils_test.go
index 5c393f8..eefce66 100644
--- a/utils_test.go
+++ b/utils_test.go
@@ -147,3 +147,28 @@ func getConfigValue(t *testing.T, key string) string {
 	require.NoError(t, err, "Error getting config value via snapctl.")
 	return strings.TrimSpace(string(out))
 }
+
+func TestCopyFile(t *testing.T) {
+	tmpdir := t.TempDir()
+	tmpfile, _ := os.CreateTemp(tmpdir, "tmpSrcFile")
+	srcPath := tmpfile.Name()
+
+	tmpdir = t.TempDir()
+	tmpfile, _ = os.CreateTemp(tmpdir, "tmpDstFile")
+	dstPath := tmpfile.Name()
+
+	require.NoError(t, CopyFile(srcPath, dstPath), "Error copying file.")
+}
+
+func TestCopyDir(t *testing.T) {
+	tmpdir := t.TempDir()
+	tmpSrcDir, _ := os.MkdirTemp(tmpdir, "tmpSrcDir")
+	os.CreateTemp(tmpSrcDir, "tmpSrcFile1")
+	os.CreateTemp(tmpSrcDir, "tmpSrcFile2")
+	os.CreateTemp(tmpSrcDir, "tmpSrcFile3")
+
+	tmpdir = t.TempDir()
+	tmpDstDir, _ := os.MkdirTemp(tmpdir, "tmpDstDir")
+
+	require.NoError(t, CopyDir(tmpSrcDir, tmpDstDir), "Error copying directory.")
+}