test: fix TestQueryCreateFromEditor and TestComponent* (#1594)

* test: fix lql tests

* test: revert spaces

* fix: test cdk components

Author: PengyuanZhao

cli/cmd/lql.go

@@ -298,7 +298,8 @@ func inputQueryFromURL(url string) (query string, err error) {
 
 func inputQueryFromEditor(action string) (query string, err error) {
 	language := "LQL"
-	if action == "create" && !queryCmdState.EmptyTemplate && cli.LwApi.V2.Query.RegoQueryEnabled() {
+	regoQueryEnabled := os.Getenv("LW_CLI_INTEGRATION_MODE") != "" || cli.LwApi.V2.Query.RegoQueryEnabled()
+	if action == "create" && !queryCmdState.EmptyTemplate && regoQueryEnabled {
 		languageSelect := &survey.Select{
 			Message: "Choose query language: ",
 			Options: []string{
@@ -400,7 +401,7 @@ Verify that the JSON is formatted properly and adheres to the following schema:
 	}
 	// smells like plain text
 	return errors.New(`invalid query
-	
+
 It looks like you attempted to submit a query in YAML format.
 Verify that the text adheres to the following schema:
 

lwcomponent/catalog.go

@@ -23,6 +23,9 @@ const (
 )
 
 func CatalogV1Enabled(client *api.Client) bool {
+	if os.Getenv("LW_CLI_INTEGRATION_MODE") != "" {
+		return true
+	}
 	response, err := client.V2.FeatureFlags.GetFeatureFlagsMatchingPrefix(featureFlag)
 	if err != nil {
 		return false