fix update ql output

rockset · Nov 6, 2023 · 19c5e09 · 19c5e09
1 parent b0425da
commit 19c5e09
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/cmd/query_lambda.go b/cmd/query_lambda.go
@@ -336,7 +336,7 @@ func newUpdateQueryLambdaCmd() *cobra.Command {
 			if err != nil {
 				return err
 			}
-			_, _ = fmt.Fprintf(cmd.OutOrStdout(), "updated query lambda %s.%s:%s",
+			_, _ = fmt.Fprintf(cmd.OutOrStdout(), "updated query lambda %s.%s:%s\n",
 				ql.GetWorkspace(), ql.GetName(), ql.GetVersion())
 
 			if err = waitUntilQLActive(rs, cmd, ws, args[0], ql.GetVersion()); err != nil {

diff --git a/cmd/query_lambda_test.go b/cmd/query_lambda_test.go
@@ -28,9 +28,9 @@ func (s *QueryLambdaSuite) Test_0_Create() {
 	c := cmd.NewRootCmd("test")
 	out := test.WrapAndExecute(s.T(), c, "create", "ql", "--sql", "testdata/query_lambda.sql", "--wait", s.name)
 
-	re := regexp.MustCompile(`created query lambda commons\.\w+:(\w+)`)
+	re := regexp.MustCompile(`created query lambda commons\.\w+:(\w+)\s`)
 	m := re.FindStringSubmatch(out.String())
-	s.NotNil(m)
+	s.Require().NotNil(m, "no regexp match")
 	s.version = m[1]
 }
 
@@ -59,9 +59,9 @@ func (s *QueryLambdaSuite) Test_4_Update() {
 	c := cmd.NewRootCmd("test")
 	out := test.WrapAndExecute(s.T(), c, "update", "ql", "--sql", "testdata/query_lambda_updated.sql", "--wait", s.name)
 
-	re := regexp.MustCompile(`updated query lambda commons\.\w+:(\w+)`)
+	re := regexp.MustCompile(`updated query lambda commons\.\w+:(\w+)\s`)
 	m := re.FindStringSubmatch(out.String())
-	s.NotNil(m)
+	s.Require().NotNil(m, "no regexp match")
 	s.version = m[1]
 }