test: add more UTs

Signed-off-by: Ramkumar Chinchani <rchincha@cisco.com>

pkg/lib/dir_test.go

@@ -0,0 +1,74 @@
+package lib_test
+
+import (
+	"os"
+	"path"
+	"testing"
+
+	. "github.com/smartystreets/goconvey/convey"
+	"stackerbuild.io/stacker/pkg/lib"
+)
+
+func TestDir(t *testing.T) {
+	Convey("IsSymLink", t, func() {
+		src, err := os.CreateTemp("", "src")
+		So(err, ShouldBeNil)
+		So(src, ShouldNotBeNil)
+		defer os.Remove(src.Name())
+
+		_, err = src.Write([]byte("hello world!"))
+		So(err, ShouldBeNil)
+
+		ok := lib.IsSymlink(src.Name())
+		So(ok, ShouldBeFalse)
+	})
+
+	Convey("DirCopy", t, func() {
+		src, err := os.CreateTemp("", "src")
+		So(err, ShouldBeNil)
+		So(src, ShouldNotBeNil)
+		defer os.Remove(src.Name())
+
+		_, err = src.Write([]byte("hello world!"))
+		So(err, ShouldBeNil)
+
+		dest, err := os.MkdirTemp("", "dest")
+		So(err, ShouldBeNil)
+		So(dest, ShouldNotBeNil)
+		defer os.Remove(dest)
+
+		err = lib.DirCopy(path.Dir(src.Name()), dest)
+		So(err, ShouldBeNil)
+	})
+
+	Convey("CopyThing", t, func() {
+		src, err := os.CreateTemp("", "src")
+		So(err, ShouldBeNil)
+		So(src, ShouldNotBeNil)
+		defer os.Remove(src.Name())
+
+		_, err = src.Write([]byte("hello world!"))
+		So(err, ShouldBeNil)
+
+		dest, err := os.CreateTemp("", "dest")
+		So(err, ShouldBeNil)
+		So(dest, ShouldNotBeNil)
+		defer os.Remove(dest.Name())
+
+		err = lib.CopyThing(src.Name(), dest.Name())
+		So(err, ShouldBeNil)
+	})
+
+	Convey("Chmod", t, func() {
+		src, err := os.CreateTemp("", "src")
+		So(err, ShouldBeNil)
+		So(src, ShouldNotBeNil)
+		defer os.Remove(src.Name())
+
+		_, err = src.Write([]byte("hello world!"))
+		So(err, ShouldBeNil)
+
+		err = lib.Chmod("644", src.Name())
+		So(err, ShouldBeNil)
+	})
+}

pkg/lib/file_test.go

@@ -0,0 +1,47 @@
+package lib_test
+
+import (
+	"io/fs"
+	"os"
+	"testing"
+
+	. "github.com/smartystreets/goconvey/convey"
+	"stackerbuild.io/stacker/pkg/lib"
+)
+
+func TestFile(t *testing.T) {
+	Convey("FileCopy", t, func() {
+		src, err := os.CreateTemp("", "src")
+		So(err, ShouldBeNil)
+		So(src, ShouldNotBeNil)
+		defer os.Remove(src.Name())
+
+		_, err = src.Write([]byte("hello world!"))
+		So(err, ShouldBeNil)
+
+		Convey("With defaults", func() {
+			dest, err := os.CreateTemp("", "dest")
+			So(err, ShouldBeNil)
+			defer os.Remove(dest.Name())
+
+			err = lib.FileCopy(dest.Name(), src.Name(), nil, -1, -1)
+			So(err, ShouldBeNil)
+		})
+
+		Convey("With non-default mode", func() {
+			dest, err := os.CreateTemp("", "dest")
+			So(err, ShouldBeNil)
+			defer os.Remove(dest.Name())
+
+			mode := fs.FileMode(0644)
+			err = lib.FileCopy(dest.Name(), src.Name(), &mode, -1, -1)
+			So(err, ShouldBeNil)
+		})
+	})
+
+	Convey("FindFiles", t, func() {
+		files, err := lib.FindFiles("/tmp", ".*")
+		So(err, ShouldBeNil)
+		So(files, ShouldNotBeEmpty)
+	})
+}

pkg/log/log_test.go

@@ -17,6 +17,12 @@ func TestLog(t *testing.T) {
 		So(func() { log.Debugf("debug msg") }, ShouldNotPanic)
 		So(func() { log.Infof("info msg") }, ShouldNotPanic)
 		So(func() { log.Errorf("error msg") }, ShouldNotPanic)
+
+		So(func() { log.FilterNonStackerLogs(handler, 1) }, ShouldNotPanic)
+
+		So(func() { log.Debugf("debug msg") }, ShouldNotPanic)
+		So(func() { log.Infof("info msg") }, ShouldNotPanic)
+		So(func() { log.Errorf("error msg") }, ShouldNotPanic)
 	})
 
 	Convey("Without timestamps", t, func() {
@@ -26,5 +32,11 @@ func TestLog(t *testing.T) {
 		So(func() { log.Debugf("debug msg") }, ShouldNotPanic)
 		So(func() { log.Infof("info msg") }, ShouldNotPanic)
 		So(func() { log.Errorf("error msg") }, ShouldNotPanic)
+
+		So(func() { log.FilterNonStackerLogs(handler, 1) }, ShouldNotPanic)
+
+		So(func() { log.Debugf("debug msg") }, ShouldNotPanic)
+		So(func() { log.Infof("info msg") }, ShouldNotPanic)
+		So(func() { log.Errorf("error msg") }, ShouldNotPanic)
 	})
 }