added unit tests

jjcfrancisco · Mar 26, 2024 · 06e92a3 · 06e92a3
1 parent 4c9d1b1
commit 06e92a3
Show file tree

Hide file tree

Showing 6 changed files with 49 additions and 9 deletions.
diff --git a/cli.go b/cli.go
@@ -23,14 +23,6 @@ func init() {
     }()
 }
 
-func cleanup() {
-	filename := os.Getenv("FILENAME")
-	if err := os.Remove(filename); err != nil {
-		fmt.Println(err)
-		os.Exit(1)
-	}
-}
-
 func main() {
 	app := &cli.App{
 		Name:  "Yeo",

diff --git a/db.go b/db.go
@@ -104,3 +104,4 @@ func prepareDb(database string) error {
 	return nil
 
 }
+
diff --git a/go.mod b/go.mod
@@ -4,13 +4,17 @@ go 1.22.0
 
 require (
 	github.com/gernest/wow v0.1.0
+	github.com/stretchr/testify v1.9.0
 	github.com/urfave/cli/v2 v2.27.1
 )
 
 require (
 	github.com/cpuguy83/go-md2man/v2 v2.0.2 // indirect
+	github.com/davecgh/go-spew v1.1.1 // indirect
+	github.com/pmezard/go-difflib v1.0.0 // indirect
 	github.com/russross/blackfriday/v2 v2.1.0 // indirect
 	github.com/xrash/smetrics v0.0.0-20201216005158-039620a65673 // indirect
 	golang.org/x/crypto v0.0.0-20190103213133-ff983b9c42bc // indirect
 	golang.org/x/sys v0.0.0-20190116161447-11f53e031339 // indirect
+	gopkg.in/yaml.v3 v3.0.1 // indirect
 )
diff --git a/go.sum b/go.sum
@@ -1,9 +1,15 @@
 github.com/cpuguy83/go-md2man/v2 v2.0.2 h1:p1EgwI/C7NhT0JmVkwCD2ZBK8j4aeHQX2pMHHBfMQ6w=
 github.com/cpuguy83/go-md2man/v2 v2.0.2/go.mod h1:tgQtvFlXSQOSOSIRvRPT7W67SCa46tRHOmNcaadrF8o=
+github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c=
+github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
 github.com/gernest/wow v0.1.0 h1:g9xdwCwP0+xgVYlA2sopI0gZHqXe7HjI/7/LykG4fks=
 github.com/gernest/wow v0.1.0/go.mod h1:dEPabJRi5BneI1Nev1VWo0ZlcTWibHWp43qxKms4elY=
+github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM=
+github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
 github.com/russross/blackfriday/v2 v2.1.0 h1:JIOH55/0cWyOuilr9/qlrm0BSXldqnqwMsf35Ld67mk=
 github.com/russross/blackfriday/v2 v2.1.0/go.mod h1:+Rmxgy9KzJVeS9/2gXHxylqXiyQDYRxCVz55jmeOWTM=
+github.com/stretchr/testify v1.9.0 h1:HtqpIVDClZ4nwg75+f6Lvsy/wHu+3BoSGCbBAcpTsTg=
+github.com/stretchr/testify v1.9.0/go.mod h1:r2ic/lqez/lEtzL7wO/rwa5dbSLXVDPFyf8C91i36aY=
 github.com/urfave/cli/v2 v2.27.1 h1:8xSQ6szndafKVRmfyeUMxkNUJQMjL1F2zmsZ+qHpfho=
 github.com/urfave/cli/v2 v2.27.1/go.mod h1:8qnjx1vcq5s2/wpsqoZFndg2CE5tNFyrTvS6SinrnYQ=
 github.com/xrash/smetrics v0.0.0-20201216005158-039620a65673 h1:bAn7/zixMGCfxrRTfdpNzjtPYqr8smhKouy9mxVdGPU=
@@ -12,3 +18,7 @@ golang.org/x/crypto v0.0.0-20190103213133-ff983b9c42bc h1:F5tKCVGp+MUAHhKp5MZtGq
 golang.org/x/crypto v0.0.0-20190103213133-ff983b9c42bc/go.mod h1:6SG95UA2DQfeDnfUPMdvaQW0Q7yPrPDi9nlGo2tz2b4=
 golang.org/x/sys v0.0.0-20190116161447-11f53e031339 h1:g/Jesu8+QLnA0CPzF3E1pURg0Byr7i6jLoX5sqjcAh0=
 golang.org/x/sys v0.0.0-20190116161447-11f53e031339/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=
+gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405 h1:yhCVgyC4o1eVCa2tZl7eS0r+SDo693bJlVdllGtEeKM=
+gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
+gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA=
+gopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
diff --git a/utils.go b/utils.go
@@ -17,7 +17,7 @@ func openConfigs(dbName string) (*database, error) {
 
 	raw, err := os.ReadFile("./databases.json")
 	if err != nil {
-		raw, err = os.ReadFile(homeDir+"/databases.json")
+		raw, err = os.ReadFile(homeDir + "/databases.json")
 		if err != nil {
 			fmt.Println("databases.json file is not present")
 			os.Exit(1)
@@ -63,3 +63,11 @@ func validateTargetDb(targetDb string) error {
 
 	return nil
 }
+
+func cleanup() {
+	filename := os.Getenv("FILENAME")
+	if err := os.Remove(filename); err != nil {
+		fmt.Println(err)
+		os.Exit(1)
+	}
+}
diff --git a/utils_test.go b/utils_test.go
@@ -0,0 +1,25 @@
+package main
+
+import (
+	"github.com/stretchr/testify/assert"
+	"os"
+	"testing"
+)
+
+func TestValidateFilename(t *testing.T) {
+	err := validateFilename("test.dump")
+	assert.NoError(t, err)
+}
+
+func TestCleanup(t *testing.T) {
+	file, err := os.Create("./test.dump")
+	if err != nil {
+		panic(err)
+	}
+	defer file.Close()
+	os.Setenv("FILENAME", "test.dump")
+	cleanup()
+	_, err = os.Stat("test.dump")
+	// File exists and cleanup has not done its job
+	assert.Error(t, err)
+}
Original file line number	Diff line number	Diff line change
Expand Up		@@ -104,3 +104,4 @@ func prepareDb(database string) error {
		return nil

		}