mainnika/mongox-go-driver
1
0
Fork 0
mirror of https://github.com/mainnika/mongox-go-driver.git synced 2026-05-22 15:53:36 +00:00
Code Issues Packages Projects Releases Wiki Activity

Ephemeral database for testing in the testing package

Browse Source
This commit is contained in:
Nikita Tokarchuk
2020-03-10 00:01:36 +01:00
parent ab4227eaef
commit 8267abe5be
4 changed files with 14 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files
mongox/base/jsonbased/id_test.go
+2 -2
View File
@@ -7,7 +7,7 @@ import (
"github.com/stretchr/testify/assert"
"go.mongodb.org/mongo-driver/bson/primitive"
"github.com/mainnika/mongox-go-driver/v2/mongox/tempdb"
"github.com/mainnika/mongox-go-driver/v2/mongox-testing/database"
)
func Test_GetID(t *testing.T) {
@@ -41,7 +41,7 @@ func Test_SaveLoad(t *testing.T) {
Primary `bson:",inline" json:",inline" collection:"1"`
}
db, err := tempdb.NewTempDB("mongodb://localhost")
db, err := database.NewEphemeral("mongodb://localhost")
if err != nil {
t.Fatal(err)
}
mongox/base/oidbased/id_test.go
+2 -2
View File
@@ -7,7 +7,7 @@ import (
"github.com/stretchr/testify/assert"
"go.mongodb.org/mongo-driver/bson/primitive"
"github.com/mainnika/mongox-go-driver/v2/mongox/tempdb"
"github.com/mainnika/mongox-go-driver/v2/mongox-testing/database"
)
func Test_GetID(t *testing.T) {
@@ -41,7 +41,7 @@ func Test_SaveLoad(t *testing.T) {
Primary `bson:",inline" json:",inline" collection:"1"`
}
db, err := tempdb.NewTempDB("mongodb://localhost")
db, err := database.NewEphemeral("mongodb://localhost")
if err != nil {
t.Fatal(err)
}
mongox/base/stringbased/id_test.go
+2 -2
View File
@@ -6,7 +6,7 @@ import (
"github.com/stretchr/testify/assert"
"github.com/mainnika/mongox-go-driver/v2/mongox/tempdb"
"github.com/mainnika/mongox-go-driver/v2/mongox-testing/database"
)
func Test_GetID(t *testing.T) {
@@ -40,7 +40,7 @@ func Test_SaveLoad(t *testing.T) {
Primary `bson:",inline" json:",inline" collection:"1"`
}
db, err := tempdb.NewTempDB("mongodb://localhost")
db, err := database.NewEphemeral("mongodb://localhost")
if err != nil {
t.Fatal(err)
}
mongox/tempdb/tempdb.go
-35
View File
@@ -1,35 +0,0 @@
package tempdb
import (
"context"
"math/rand"
"strconv"
"go.mongodb.org/mongo-driver/mongo"
"go.mongodb.org/mongo-driver/mongo/options"
"github.com/mainnika/mongox-go-driver/v2/mongox"
"github.com/mainnika/mongox-go-driver/v2/mongox/database"
)
// TempDB is a temporary database connection that will be destroyed after close
type TempDB struct {
mongox.Database
}
// NewTempDB creates new mongo connection
func NewTempDB(URI string) (tempdb *TempDB, err error) {
name := strconv.Itoa(rand.Int())
opts := options.Client().ApplyURI(URI)
client, err := mongo.Connect(context.Background(), opts)
tempdb = &TempDB{Database: database.NewDatabase(client, name)}
return
}
// Close the connection and drop database
func (tdb *TempDB) Close() {
_ = tdb.Client().Database(tdb.Name()).Drop(tdb.Context())
}