Initial import

main_test.go

@@ -0,0 +1,134 @@
+package mongo
+
+import (
+	"os"
+	"time"
+	"reflect"
+	"testing"
+  "context"
+  "fmt"
+	"encoding/json"
+
+	"go.mongodb.org/mongo-driver/v2/mongo/options"
+	"go.mongodb.org/mongo-driver/v2/bson"
+
+  "github.com/testcontainers/testcontainers-go"
+  "github.com/testcontainers/testcontainers-go/modules/mongodb"
+)
+
+type Sample struct {
+	Id   			string `json:"id" bson:"_id,omitempty"`
+	Name 		  string `json:"name" bson:"name"`
+	Age       int32 `json:"age"  bson:"age"`
+	CreatedAt time.Time `json:"created_at" bson:"created_at"`
+}
+
+func TestMain(m *testing.M) {
+  if os.Getenv("RUN_INTEGRATION") == "" {
+    fmt.Println("Skipping package tests: RUN_INTEGRATION is missing")
+    os.Exit(0) // Exit with success, but no tests ran
+  }
+
+  ctx := context.Background()
+
+  user := "admin"
+  pass := "1"
+
+  // 1. Setup: Start the MongoDB container
+  mongoContainer, err := mongodb.Run(ctx, "mongo:8",
+    testcontainers.WithEnv(map[string]string{
+      "MONGO_INITDB_ROOT_USERNAME": user,
+      "MONGO_INITDB_ROOT_PASSWORD": pass,
+    }),)
+  if err != nil {
+    panic("failed to start container")
+  }
+
+  // 2. Get the connection string dynamically
+  endpoint, _ := mongoContainer.ConnectionString(ctx)
+
+  Start(&MongoStartProps{
+    MongoUri: endpoint,
+    MongoUser: user,
+    MongoPass: pass,
+  })
+
+  // 3. Run tests
+  LoadTestSample()
+
+  code := m.Run()
+
+  // 4. Teardown: Clean up resources
+  Stop()
+
+  _ = testcontainers.TerminateContainer(mongoContainer)
+
+  os.Exit(code)
+}
+
+func TestApplySchema(t *testing.T) {
+	// prepare schema sample
+	schemaStr := `
+	{
+		"bsonType": "object",
+		"properties": {
+			"name": {
+				"bsonType": "string"
+			},
+			"active": {
+				"bsonType": "boolean"
+			}
+		},
+		"required": ["name"],
+		"additionalProperties": true
+	}
+	`
+
+	var result bson.M
+	err := json.Unmarshal([]byte(schemaStr), &result)
+	if err != nil {
+		t.Fatalf("%v", err)
+	}
+
+	cd := &CollectionDefinition{
+		Schema: result,
+	}
+
+	opts := options.CreateCollection()
+
+	ApplySchema(cd, opts)
+
+	options := &options.CreateCollectionOptions{}
+
+	for _, o := range opts.List() {
+		o(options)
+	}
+
+	expected := bson.M{"$jsonSchema":bson.M{"additionalProperties":true, "bsonType":"object", "properties":bson.D{bson.E{Key:"active", Value:bson.D{bson.E{Key:"bsonType", Value:"boolean"}}}, bson.E{Key:"name", Value:bson.D{bson.E{Key:"bsonType", Value:"string"}}}}, "required":bson.A{"name"}}}
+
+	aj, _ := json.Marshal(options.Validator)
+	ej, _ := json.Marshal(expected)
+
+	var am, em any
+	json.Unmarshal(aj, &am)
+	json.Unmarshal(ej, &em)
+
+	if !reflect.DeepEqual(am, em) {
+		t.Fatalf("Validator should have been set")
+	}
+}
+
+func TestGetCollection_Default(t *testing.T) {
+	c := GetMongoClient()
+
+	col := c.GetCollection("mydb", "mycol")
+
+	if col == nil {
+		t.Fatal("should have return collection")
+	}
+
+	if col.Name() != "mycol" {
+		t.Fatal("Wrong name")
+	}
+
+}