Add Some Unit Test

This commit is contained in:
hantmac
2019-08-18 12:09:54 +08:00
parent 8c96056f4e
commit 15e7fa662e
3 changed files with 36 additions and 0 deletions

View File

@@ -0,0 +1,16 @@
package config
import (
. "github.com/smartystreets/goconvey/convey"
"testing"
)
func TestInitConfig(t *testing.T) {
Convey("Test InitConfig func", t, func() {
x := InitConfig("")
Convey("The value should be nil", func() {
So(x, ShouldEqual, nil)
})
})
}

View File

@@ -0,0 +1,19 @@
package database
import (
. "github.com/smartystreets/goconvey/convey"
"github.com/spf13/viper"
"testing"
)
func TestGetDb(t *testing.T) {
Convey("Test GetDb", t, func() {
s, db := GetDb()
Convey("The value should be Session.Copy", func() {
So(s, ShouldEqual, Session.Copy())
})
Convey("The value should be reference of database", func() {
So(db, ShouldEqual, s.DB(viper.GetString("mongo.db")))
})
})
}

View File

@@ -11,5 +11,6 @@ require (
github.com/gomodule/redigo v2.0.0+incompatible
github.com/pkg/errors v0.8.1
github.com/satori/go.uuid v1.2.0
github.com/smartystreets/goconvey v0.0.0-20190731233626-505e41936337
github.com/spf13/viper v1.4.0
)