Function TestParseMongoShell

backend/plugin/parser/mongodb/mongodb_test.go:12–96

(t *testing.T)

)

func TestParseMongoShell(t *testing.T) {
	testCases := []struct {
		description    string
		statement      string
		wantStatements int
		wantText       string
	}{
		{
			description:    "single find statement",
			statement:      `db.collection.find({})`,
			wantStatements: 1,
			wantText:       `db.collection.find({})`,
		},
		{
			description:    "single find with filter",
			statement:      `db.users.find({ name: "John" })`,
			wantStatements: 1,
			wantText:       `db.users.find({ name: "John" })`,
		},
		{
			description:    "multiple statements",
			statement:      "db.users.find({});\ndb.products.insertOne({ name: \"test\" })",
			wantStatements: 2,
		},
		{
			description:    "show databases command",
			statement:      "show dbs",
			wantStatements: 1,
		},
		{
			description:    "show collections command",
			statement:      "show collections",
			wantStatements: 1,
		},
		{
			description:    "aggregation pipeline",
			statement:      `db.orders.aggregate([{ $match: { status: "A" } }, { $group: { _id: "$cust_id", total: { $sum: "$amount" } } }])`,
			wantStatements: 1,
		},
		{
			description:    "bracket notation for collection",
			statement:      `db["my-collection"].find({})`,
			wantStatements: 1,
		},
		{
			description:    "method chaining",
			statement:      `db.users.find({}).sort({ name: 1 }).limit(10)`,
			wantStatements: 1,
		},
		{
			description:    "empty input",
			statement:      "",
			wantStatements: 0,
		},
		{
			description:    "whitespace only",
			statement:      "   \n\t  ",
			wantStatements: 0,