Function TestGetQuerySpan

backend/plugin/parser/mongodb/query_span_test.go:63–184

(t *testing.T)

}

func TestGetQuerySpan(t *testing.T) {
	tests := []struct {
		description string
		statement   string
		wantType    base.QueryType
	}{
		{
			description: "find is Select",
			statement:   `db.users.find({})`,
			wantType:    base.Select,
		},
		{
			description: "findOne is Select",
			statement:   `db.users.findOne({name: "John"})`,
			wantType:    base.Select,
		},
		{
			description: "insertOne is DML",
			statement:   `db.users.insertOne({name: "John"})`,
			wantType:    base.DML,
		},
		{
			description: "insertMany is DML",
			statement:   `db.users.insertMany([{name: "A"}, {name: "B"}])`,
			wantType:    base.DML,
		},
		{
			description: "updateOne is DML",
			statement:   `db.users.updateOne({name: "John"}, {$set: {age: 30}})`,
			wantType:    base.DML,
		},
		{
			description: "deleteOne is DML",
			statement:   `db.users.deleteOne({name: "John"})`,
			wantType:    base.DML,
		},
		{
			description: "createCollection is DDL",
			statement:   `db.createCollection("test")`,
			wantType:    base.DDL,
		},
		{
			description: "dropDatabase is DDL",
			statement:   `db.dropDatabase()`,
			wantType:    base.DDL,
		},
		{
			description: "createIndex is DDL",
			statement:   `db.users.createIndex({name: 1})`,
			wantType:    base.DDL,
		},
		{
			description: "drop collection is DDL",
			statement:   `db.users.drop()`,
			wantType:    base.DDL,
		},
		{
			description: "find with explain suffix is Explain",