diff --git a/persistent/storage.go b/persistent/storage.go index 1491d2b7..b76b58a8 100644 --- a/persistent/storage.go +++ b/persistent/storage.go @@ -1,8 +1,6 @@ package persistent import ( - "errors" - "github.com/TykTechnologies/storage/persistent/internal/driver/mongo" "github.com/TykTechnologies/storage/persistent/internal/driver/mgo" @@ -26,9 +24,8 @@ func NewPersistentStorage(opts *ClientOpts) (types.PersistentStorage, error) { switch opts.Type { case OfficialMongo: return mongo.NewMongoDriver(&clientOpts) - case Mgo: - return mgo.NewMgoDriver(&clientOpts) default: - return nil, errors.New("invalid driver") + //case MGO, or any unvalid + return mgo.NewMgoDriver(&clientOpts) } } diff --git a/persistent/storage_test.go b/persistent/storage_test.go index c4edd385..e6d94b24 100644 --- a/persistent/storage_test.go +++ b/persistent/storage_test.go @@ -17,11 +17,7 @@ func TestNewPersistentStorage(t *testing.T) { Type: tc, }) - if tc == "unvalid" { - assert.Error(t, err) - } else { - assert.Nil(t, err) - } + assert.Nil(t, err) }) } }