WIP

Author: qingyang-hu

mongo/test/search_index_view_test.go

@@ -8,26 +8,57 @@ import (
 	"github.com/stretchr/testify/require"
 	"github.com/testcontainers/testcontainers-go"
 	"github.com/testcontainers/testcontainers-go/modules/mongodb/atlaslocal"
+	"go.mongodb.org/mongo-driver/v2/bson"
+	"go.mongodb.org/mongo-driver/v2/mongo"
+	"go.mongodb.org/mongo-driver/v2/mongo/options"
 )
 
 func TestNewSingleResultFromDocument(t *testing.T) {
 	ctx := context.Background()
 
-	atlaslocalContainer, err := atlaslocal.Run(ctx, "mongodb/mongodb-atlas-local:latest") // testcontainers.WithExposedPorts("27017/tcp"),
-	// testcontainers.WithWaitStrategy(
-	// 	wait.ForListeningPort("27017/tcp"),
-	// 	wait.ForLog("Ready to accept connections"),
-	// ),
+	atlaslocalContainer, err := atlaslocal.Run(ctx, "mongodb/mongodb-atlas-local:latest")
 
 	defer func() {
 		err := testcontainers.TerminateContainer(atlaslocalContainer)
 		assert.NoErrorf(t, err, "failed to terminate container: %v", err)
 	}()
 	require.NoErrorf(t, err, "failed to start container: %v", err)
 
-	// connString, err := atlaslocalContainer.ConnectionString(ctx)
-	// require.NoErrorf(t, err, "failed to get connection string: %v", err)
+	connString, err := atlaslocalContainer.ConnectionString(ctx)
+	require.NoErrorf(t, err, "failed to get connection string: %v", err)
 
-	// _, err = mongo.Connect(options.Client().ApplyURI(connString))
-	// require.NoErrorf(t, err, "failed to connect to MongoDB: %v", err)
+	client, err := mongo.Connect(options.Client().ApplyURI(connString))
+	require.NoErrorf(t, err, "failed to connect to MongoDB: %v", err)
+	defer func() {
+		err := client.Disconnect(ctx)
+		assert.NoErrorf(t, err, "failed to disconnect: %v", err)
+	}()
+
+	collection := client.Database("testdb").Collection("testCollection")
+
+	documents := []interface{}{
+		bson.M{
+			"string_field": "test1 test1",
+		},
+	}
+
+	collection.InsertMany(ctx, documents)
+
+	idx := bson.D{
+		// Define custom analyzer for case-insensitive keyword matching
+		{"mappings", bson.D{
+			{"dynamic", true},
+		}},
+	}
+
+	// Create search index with explicit field mappings
+	// Index fields with multiple types (like ES multi-fields):
+	// - type: "string" for text/wildcard search
+	// - type: "token" for exact matching with equals operator
+	searchIndexModel := mongo.SearchIndexModel{
+		Definition: idx,
+	}
+
+	_, err = collection.SearchIndexes().CreateOne(ctx, searchIndexModel)
+	require.NoErrorf(t, err, "failed to create search index: %v", err)
 }