Added unit tests for go/cmd/rulesctl/ package

go/cmd/rulesctl/cmd/add_test.go

@@ -0,0 +1,78 @@
+/*
+Copyright 2024 The Vitess Authors.
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+    http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+*/
+
+package cmd
+
+import (
+	"io"
+	"os"
+	"testing"
+
+	"github.com/spf13/cobra"
+	"github.com/stretchr/testify/require"
+)
+
+func TestAdd(t *testing.T) {
+	cmd := Add()
+	require.NotNil(t, cmd)
+	require.Equal(t, "add-rule", cmd.Name())
+	configFile = "./testdata/rules.json"
+
+	tests := []struct {
+		name           string
+		args           []string
+		expectedOutput string
+	}{
+		{
+			name:           "Action fail",
+			args:           []string{"--dry-run=true", "--name=Rule", `--description="New rules that will be added to the file"`, "--action=fail", "--plan=Select"},
+			expectedOutput: "[\n  {\n    \"Description\": \"Some value\",\n    \"Name\": \"Name\",\n    \"Action\": \"FAIL\"\n  },\n  {\n    \"Description\": \"\\\"New rules that will be added to the file\\\"\",\n    \"Name\": \"Rule\",\n    \"Plans\": [\n      \"Select\"\n    ],\n    \"Action\": \"FAIL\"\n  }\n]\n",
+		},
+		{
+			name:           "Action fail_retry",
+			args:           []string{"--dry-run=true", "--name=Rule", `--description="New rules that will be added to the file"`, "--action=fail_retry", "--plan=Select"},
+			expectedOutput: "[\n  {\n    \"Description\": \"Some value\",\n    \"Name\": \"Name\",\n    \"Action\": \"FAIL\"\n  },\n  {\n    \"Description\": \"\\\"New rules that will be added to the file\\\"\",\n    \"Name\": \"Rule\",\n    \"Plans\": [\n      \"Select\",\n      \"Select\"\n    ],\n    \"Action\": \"FAIL_RETRY\"\n  }\n]\n",
+		},
+		{
+			name:           "Action continue with query",
+			args:           []string{"--dry-run=true", "--name=Rule", `--description="New rules that will be added to the file"`, "--action=continue", "--plan=Select", "--query=secret", "--leading-comment=None", "--trailing-comment=Yoho", "--table=Temp"},
+			expectedOutput: "[\n  {\n    \"Description\": \"Some value\",\n    \"Name\": \"Name\",\n    \"Action\": \"FAIL\"\n  },\n  {\n    \"Description\": \"\\\"New rules that will be added to the file\\\"\",\n    \"Name\": \"Rule\",\n    \"Query\": \"secret\",\n    \"LeadingComment\": \"None\",\n    \"TrailingComment\": \"Yoho\",\n    \"Plans\": [\n      \"Select\",\n      \"Select\",\n      \"Select\"\n    ],\n    \"TableNames\": [\n      \"Temp\"\n    ]\n  }\n]\n",
+		},
+	}
+	for _, tt := range tests {
+		tt := tt
+		t.Run(tt.name, func(t *testing.T) {
+			if tt.args != nil {
+				cmd.SetArgs(tt.args)
+				err := cmd.Execute()
+				require.NoError(t, err)
+			}
+
+			// Redirect stdout to a buffer
+			rescueStdout := os.Stdout
+			r, w, _ := os.Pipe()
+			os.Stdout = w
+
+			cmd.Run(&cobra.Command{}, []string{})
+
+			w.Close()
+			got, _ := io.ReadAll(r)
+			os.Stdout = rescueStdout
+
+			require.Contains(t, string(got), tt.expectedOutput)
+		})
+	}
+}

go/cmd/rulesctl/cmd/explain_test.go

@@ -0,0 +1,68 @@
+/*
+Copyright 2024 The Vitess Authors.
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+    http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+*/
+
+package cmd
+
+import (
+	"io"
+	"os"
+	"testing"
+
+	"github.com/spf13/cobra"
+	"github.com/stretchr/testify/require"
+)
+
+func TestExplainWithQueryPlanArguement(t *testing.T) {
+	explainCmd := Explain()
+
+	require.NotNil(t, explainCmd)
+	require.Equal(t, "explain", explainCmd.Name())
+
+	// Redirect stdout to a buffer
+	rescueStdout := os.Stdout
+	r, w, _ := os.Pipe()
+	os.Stdout = w
+
+	explainCmd.Run(&cobra.Command{}, []string{"query-plans"})
+
+	w.Close()
+	got, _ := io.ReadAll(r)
+	os.Stdout = rescueStdout
+
+	expected := "Query Plans!"
+	require.Contains(t, string(got), expected)
+}
+
+func TestExplainWithRandomArguement(t *testing.T) {
+	explainCmd := Explain()
+
+	require.NotNil(t, explainCmd)
+	require.Equal(t, "explain", explainCmd.Name())
+
+	// Redirect stdout to a buffer
+	rescueStdout := os.Stdout
+	r, w, _ := os.Pipe()
+	os.Stdout = w
+
+	explainCmd.Run(&cobra.Command{}, []string{"random"})
+
+	w.Close()
+	got, _ := io.ReadAll(r)
+	os.Stdout = rescueStdout
+
+	expected := "I don't know anything about"
+	require.Contains(t, string(got), expected)
+}

go/cmd/rulesctl/cmd/list_test.go

@@ -0,0 +1,87 @@
+/*
+Copyright 2024 The Vitess Authors.
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+    http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+*/
+
+package cmd
+
+import (
+	"io"
+	"os"
+	"testing"
+
+	"github.com/spf13/cobra"
+	"github.com/stretchr/testify/require"
+)
+
+func TestList(t *testing.T) {
+	cmd := List()
+	require.NotNil(t, cmd)
+	require.Equal(t, "list", cmd.Name())
+	configFile = "./testdata/rules.json"
+
+	tests := []struct {
+		name           string
+		args           []string
+		expectedOutput string
+	}{
+		{
+			name:           "No args",
+			expectedOutput: "[\n  {\n    \"Description\": \"Some value\",\n    \"Name\": \"Name\",\n    \"Action\": \"FAIL\"\n  }\n]\n",
+		},
+		{
+			name:           "Name only",
+			args:           []string{"--names-only=true"},
+			expectedOutput: "[\n  \"Name\"\n]\n",
+		},
+		{
+			name:           "Name flag set",
+			args:           []string{"--name=Name"},
+			expectedOutput: "\"Name\"\n",
+		},
+		{
+			name:           "Random name in name flag",
+			args:           []string{"--name=Random"},
+			expectedOutput: "\"\"\n",
+		},
+		{
+			name:           "Random name in name flag and names-only false",
+			args:           []string{"--name=Random", "--names-only=false"},
+			expectedOutput: "null\n",
+		},
+	}
+	for _, tt := range tests {
+		tt := tt
+		t.Run(tt.name, func(t *testing.T) {
+			if tt.args != nil {
+				cmd.SetArgs(tt.args)
+				err := cmd.Execute()
+				require.NoError(t, err)
+			}
+
+			// Redirect stdout to a buffer
+			rescueStdout := os.Stdout
+			r, w, _ := os.Pipe()
+			os.Stdout = w
+
+			cmd.Run(&cobra.Command{}, []string{})
+
+			w.Close()
+			got, _ := io.ReadAll(r)
+			os.Stdout = rescueStdout
+
+			require.Contains(t, string(got), tt.expectedOutput)
+		})
+	}
+}

go/cmd/rulesctl/cmd/main_test.go

@@ -0,0 +1,51 @@
+/*
+Copyright 2024 The Vitess Authors.
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+    http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+*/
+
+package cmd
+
+import (
+	"io"
+	"os"
+	"testing"
+
+	"github.com/stretchr/testify/require"
+)
+
+func TestMain(t *testing.T) {
+	rootCmd := Main()
+	require.NotNil(t, rootCmd)
+	require.Equal(t, "rulesctl", rootCmd.Name())
+
+	// Redirect stdout to a buffer
+	rescueStdout := os.Stdout
+	t.Cleanup(func() {
+		os.Stdout = rescueStdout
+	})
+	r, w, _ := os.Pipe()
+	os.Stdout = w
+
+	args := os.Args
+	t.Cleanup(func() { os.Args = args })
+	os.Args = []string{"rulesctl", "-f=testdata/rules.json", "list"}
+	err := rootCmd.Execute()
+	require.NoError(t, err)
+
+	w.Close()
+	got, _ := io.ReadAll(r)
+
+	expected := "[\n  {\n    \"Description\": \"Some value\",\n    \"Name\": \"Name\",\n    \"Action\": \"FAIL\"\n  }\n]\n"
+	require.Equal(t, string(got), expected)
+}

go/cmd/rulesctl/cmd/remove_test.go

@@ -0,0 +1,99 @@
+/*
+Copyright 2024 The Vitess Authors.
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+    http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+*/
+
+package cmd
+
+import (
+	"io"
+	"os"
+	"testing"
+
+	"github.com/spf13/cobra"
+	"github.com/stretchr/testify/require"
+)
+
+func TestRemoveOld(t *testing.T) {
+	removeCmd := Remove()
+
+	require.NotNil(t, removeCmd)
+	require.Equal(t, "remove-rule", removeCmd.Name())
+
+	// Redirect stdout to a buffer
+	rescueStdout := os.Stdout
+	r, w, _ := os.Pipe()
+	os.Stdout = w
+
+	configFile = "../common/testdata/rules.json"
+	removeCmd.Run(&cobra.Command{}, []string{""})
+
+	w.Close()
+	got, _ := io.ReadAll(r)
+	os.Stdout = rescueStdout
+
+	expected := "No rule found:"
+	require.Contains(t, string(got), expected)
+}
+
+func TestRemove(t *testing.T) {
+	cmd := Remove()
+	require.NotNil(t, cmd)
+	require.Equal(t, "remove-rule", cmd.Name())
+	configFile = "./testdata/rules.json"
+
+	tests := []struct {
+		name           string
+		args           []string
+		expectedOutput string
+	}{
+		{
+			name:           "No args",
+			expectedOutput: "No rule found: ''",
+		},
+		{
+			name:           "Dry run and name both set",
+			args:           []string{"--dry-run=true", "--name=Name"},
+			expectedOutput: "[]\n",
+		},
+		{
+			name:           "Dry run not set name set",
+			args:           []string{"--dry-run=false", "--name=Name"},
+			expectedOutput: "No rule found: 'Name'",
+		},
+	}
+	for _, tt := range tests {
+		tt := tt
+		t.Run(tt.name, func(t *testing.T) {
+			if tt.args != nil {
+				cmd.SetArgs(tt.args)
+				err := cmd.Execute()
+				require.NoError(t, err)
+			}
+
+			// Redirect stdout to a buffer
+			rescueStdout := os.Stdout
+			r, w, _ := os.Pipe()
+			os.Stdout = w
+
+			cmd.Run(&cobra.Command{}, []string{})
+
+			w.Close()
+			got, _ := io.ReadAll(r)
+			os.Stdout = rescueStdout
+
+			require.Contains(t, string(got), tt.expectedOutput)
+		})
+	}
+}

go/cmd/rulesctl/cmd/testdata/rules.json

@@ -0,0 +1,6 @@
+[
+    {
+        "Description": "Some value",
+        "Name": "Name"
+    }
+]