feat(RAIN-74824): support queryLanguage when creating query

lacework · Nov 6, 2023 · 22edb19 · 22edb19
1 parent f44bf6f
commit 22edb19
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 8 deletions.
diff --git a/api/lql.go b/api/lql.go
@@ -29,8 +29,9 @@ import (
 )
 
 type NewQuery struct {
-	QueryID   string `json:"queryId" yaml:"queryId"`
-	QueryText string `json:"queryText" yaml:"queryText"`
+	QueryID       string `json:"queryId" yaml:"queryId"`
+	QueryLanguage string `json:"queryLanguage" yaml:"queryLanguage"`
+	QueryText     string `json:"queryText" yaml:"queryText"`
 }
 
 func ParseNewQuery(s string) (NewQuery, error) {

diff --git a/api/lql_test.go b/api/lql_test.go
@@ -33,18 +33,22 @@ import (
 
 var (
 	queryID      = "my_lql"
+	lqlLanguage  = "lql"
 	newQueryText = `my_lql { source { CloudTrailRawEvents } return { INSERT_ID } }`
 	newQuery     = api.NewQuery{
-		QueryID:   queryID,
-		QueryText: newQueryText,
+		QueryID:       queryID,
+		QueryLanguage: lqlLanguage,
+		QueryText:     newQueryText,
 	}
 	newQueryJSON = fmt.Sprintf(`{
 	"queryId": "%s",
+"queryLanguage": "%s",
 	"queryText": "%s"
-}`, queryID, newQueryText)
+}`, queryID, lqlLanguage, newQueryText)
 	newQueryYAML = fmt.Sprintf(`---
 queryId: %s
-queryText: %s`, newQuery.QueryID, newQuery.QueryText)
+queryLanguage: %s
+queryText: %s`, newQuery.QueryID, newQuery.QueryLanguage, newQuery.QueryText)
 	lqlErrorReponse = `{ "message": "This is an error message" }`
 )
 

diff --git a/cli/cmd/lql_internal_test.go b/cli/cmd/lql_internal_test.go
@@ -31,8 +31,9 @@ import (
 
 var (
 	newQuery = api.NewQuery{
-		QueryID:   "my_lql",
-		QueryText: `my_lql { source { CloudTrailRawEvents } return { INSERT_ID } }`,
+		QueryID:       "my_lql",
+		QueryLanguage: "lql",
+		QueryText:     `my_lql { source { CloudTrailRawEvents } return { INSERT_ID } }`,
 	}
 )