trantor/lib/database/database_test.go

package database

import (
	"testing"

	mgo "gopkg.in/mgo.v2"
)

const (
	test_coll = "test_trantor"
	test_host = "127.0.0.1"
)

func TestInit(t *testing.T) {
	db := Init(test_host, test_coll)
	defer db.Close()
}

func del(db DB) {
	db.Close()
	session, _ := mgo.Dial(test_host)
	defer session.Close()
	session.DB(test_coll).DropDatabase()
}