cleaner tests i thought

This commit is contained in:
breel
2020-08-01 01:09:35 -06:00
parent baf0aaa287
commit 71b1de33ec
7 changed files with 141 additions and 141 deletions

View File

@@ -4,7 +4,6 @@ import (
"context"
"errors"
"fmt"
"local/dndex/config"
"local/dndex/storage/entity"
"os"
"regexp"
@@ -20,9 +19,8 @@ type BoltDB struct {
db *bolt.DB
}
func NewBoltDB() *BoltDB {
config := config.New()
db, err := bolt.Open(config.DBURI, os.ModePerm, &bolt.Options{
func NewBoltDB(path string) *BoltDB {
db, err := bolt.Open(path, os.ModePerm, &bolt.Options{
Timeout: time.Second * 3,
})
if err != nil {

View File

@@ -364,8 +364,7 @@ func tempBoltDB(t *testing.T) (*BoltDB, func()) {
t.Fatal(err)
}
f.Close()
os.Args = []string{"a", "-dburi", f.Name()}
bdb := NewBoltDB()
bdb := NewBoltDB(f.Name())
fillBoltDB(t, bdb)
return bdb, func() {
bdb.db.Close()

View File

@@ -15,12 +15,15 @@ type Driver interface {
Delete(context.Context, string, interface{}) error
}
func New() Driver {
func New(path ...string) Driver {
if len(path) == 0 {
path = []string{config.New().DBURI}
}
switch strings.ToLower(config.New().DriverType) {
case "mongo":
return NewMongo()
return NewMongo(path[0])
case "boltdb":
return NewBoltDB()
return NewBoltDB(path[0])
}
panic("unknown driver type " + strings.ToLower(config.New().DriverType))
}

View File

@@ -17,8 +17,8 @@ type Mongo struct {
db string
}
func NewMongo() Mongo {
opts := options.Client().ApplyURI(config.New().DBURI)
func NewMongo(path string) Mongo {
opts := options.Client().ApplyURI(path)
c, err := mongo.NewClient(opts)
if err != nil {
panic(err)