Ephemeral database for testing in the testing package

v2 v2.0.1
Nikita Tokarchuk 4 years ago
parent ab4227eaef
commit 8267abe5be
  1. 16
      mongox-testing/database/ephemeral.go
  2. 4
      mongox/base/jsonbased/id_test.go
  3. 4
      mongox/base/oidbased/id_test.go
  4. 4
      mongox/base/stringbased/id_test.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
type TempDB struct {
// EphemeralDatabase is a temporary database connection that will be destroyed after close
type EphemeralDatabase struct {
mongox.Database
}
// NewTempDB creates new mongo connection
func NewTempDB(URI string) (tempdb *TempDB, err error) {
// NewEphemeral creates new mongo connection
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() {
_ = tdb.Client().Database(tdb.Name()).Drop(tdb.Context())
func (e *EphemeralDatabase) Close() error {
return e.Client().Database(e.Name()).Drop(e.Context())
}

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

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

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

Loading…
Cancel
Save