From 6001312acc90b219ded8aec31cb33eb0cf5db44f Mon Sep 17 00:00:00 2001
From: kerwin612 <kerwin612@qq.com>
Date: Thu, 13 Jul 2023 10:48:23 +0800
Subject: [PATCH] Add test cases

---
 main_test.go | 19 +++++++++++++++++++
 1 file changed, 19 insertions(+)
 create mode 100644 main_test.go

diff --git a/main_test.go b/main_test.go
new file mode 100644
index 0000000..97528dc
--- /dev/null
+++ b/main_test.go
@@ -0,0 +1,19 @@
+package main
+
+import "testing"
+
+func Do_getFiles(t *testing.T, parentDir, keys string, no int) {
+    files, err := getFiles(parentDir, keys)
+    if err != nil {
+        t.Fatalf("getFiles failed in %s by %s, %v", parentDir, keys, err)
+    }
+    if len(files) != no {
+        t.Fatalf("getFiles in %s by %s, expected %d, but got %d", parentDir, keys, no, len(files))
+    }
+}
+
+func Test_getFiles(t *testing.T) {
+    Do_getFiles(t, ".", "*.mod", 1)
+    Do_getFiles(t, ".", "*.mod *.sum", 2)
+    Do_getFiles(t, ".", "*.mod\n*.sum", 2)
+}