Ephemeral database for testing in the testing package

mongox-testing/database/ephemeral.go

@@ -1,4 +1,4 @@
-package tempdb
+package database
 
 import (
 	"context"
@@ -12,24 +12,24 @@ import (
 	"github.com/mainnika/mongox-go-driver/v2/mongox/database"
 )
 
-// TempDB is a temporary database connection that will be destroyed after close
+// EphemeralDatabase is a temporary database connection that will be destroyed after close
-type TempDB struct {
+type EphemeralDatabase struct {
 	mongox.Database
 }
 
-// NewTempDB creates new mongo connection
+// NewEphemeral creates new mongo connection
-func NewTempDB(URI string) (tempdb *TempDB, err error) {
+func NewEphemeral(URI string) (db *EphemeralDatabase, 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)}
+	db = &EphemeralDatabase{Database: database.NewDatabase(client, name)}
 
 	return
 }
 
 // Close the connection and drop database
-func (tdb *TempDB) Close() {
+func (e *EphemeralDatabase) Close() error {
-	_ = tdb.Client().Database(tdb.Name()).Drop(tdb.Context())
+	return e.Client().Database(e.Name()).Drop(e.Context())
 }

mongox/base/jsonbased/id_test.go

@@ -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

@@ -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

@@ -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)
 	}