Function TestGetStatementRanges

backend/plugin/parser/mongodb/mongodb_test.go:104–142

(t *testing.T)

}

func TestGetStatementRanges(t *testing.T) {
	testCases := []struct {
		description string
		statement   string
		wantRanges  int
	}{
		{
			description: "single statement",
			statement:   `db.collection.find({})`,
			wantRanges:  1,
		},
		{
			description: "multiple statements on separate lines",
			statement:   "db.users.find({});\ndb.products.find({})",
			wantRanges:  2,
		},
		{
			description: "multiline statement",
			statement: `db.users.aggregate([
  { $match: { status: "A" } },
  { $group: { _id: "$cust_id" } }
])`,
			wantRanges: 1,
		},
		{
			description: "empty input",
			statement:   "",
			wantRanges:  0,
		},
	}

	for _, tc := range testCases {
		t.Run(tc.description, func(t *testing.T) {
			ranges, err := GetStatementRanges(context.Background(), base.StatementRangeContext{}, tc.statement)
			require.NoError(t, err)
			require.Len(t, ranges, tc.wantRanges)
		})
	}
}

func TestGetStatementRangesWithHindiCharacters(t *testing.T) {
	// Test with Hindi characters - ANTLR returns character (rune) offsets, not byte offsets