mirror of
https://github.com/crawlab-team/crawlab.git
synced 2026-01-23 17:31:11 +01:00
29 lines
645 B
Go
29 lines
645 B
Go
package database
|
|
|
|
import (
|
|
"crawlab/config"
|
|
. "github.com/smartystreets/goconvey/convey"
|
|
"github.com/spf13/viper"
|
|
"testing"
|
|
)
|
|
|
|
func TestGetDb(t *testing.T) {
|
|
Convey("Test GetDb", t, func() {
|
|
if err := config.InitConfig(""); err != nil {
|
|
t.Fatal("Init config failed")
|
|
}
|
|
t.Log("初始化配置成功")
|
|
err := InitMongo()
|
|
if err != nil {
|
|
t.Fatal("Init mongodb failed")
|
|
}
|
|
s, db := GetDb()
|
|
Convey("The value should be Session.Copy", func() {
|
|
So(s, ShouldResemble, Session.Copy())
|
|
})
|
|
Convey("The value should be reference of database", func() {
|
|
So(db, ShouldResemble, s.DB(viper.GetString("mongo.db")))
|
|
})
|
|
})
|
|
}
|