add unit tests

Author: LinuxSuRen

cmd/jsonschema_test.go

@@ -6,11 +6,12 @@ import (
 	"testing"
 
 	"github.com/linuxsuren/api-testing/cmd"
+	fakeruntime "github.com/linuxsuren/go-fake-runtime"
 	"github.com/stretchr/testify/assert"
 )
 
 func TestJSONSchemaCmd(t *testing.T) {
-	c := cmd.NewRootCmd()
+	c := cmd.NewRootCmd(fakeruntime.FakeExecer{ExpectOS: "linux"})
 
 	buf := new(bytes.Buffer)
 	c.SetOut(buf)

cmd/root.go

@@ -4,11 +4,12 @@ import (
 	"os"
 
 	"github.com/linuxsuren/api-testing/pkg/version"
+	fakeruntime "github.com/linuxsuren/go-fake-runtime"
 	"github.com/spf13/cobra"
 )
 
 // NewRootCmd creates the root command
-func NewRootCmd() (c *cobra.Command) {
+func NewRootCmd(execer fakeruntime.Execer) (c *cobra.Command) {
 	c = &cobra.Command{
 		Use:   "atest",
 		Short: "API testing tool",
@@ -18,6 +19,6 @@ func NewRootCmd() (c *cobra.Command) {
 	c.AddCommand(createInitCommand(),
 		createRunCommand(), createSampleCmd(),
 		createServerCmd(), createJSONSchemaCmd(),
-		createServiceCommand())
+		createServiceCommand(execer))
 	return
 }

cmd/run_test.go

@@ -8,6 +8,7 @@ import (
 
 	"github.com/h2non/gock"
 	"github.com/linuxsuren/api-testing/pkg/limit"
+	fakeruntime "github.com/linuxsuren/go-fake-runtime"
 	"github.com/spf13/cobra"
 	"github.com/stretchr/testify/assert"
 )
@@ -103,7 +104,7 @@ func TestRunCommand(t *testing.T) {
 }
 
 func TestRootCmd(t *testing.T) {
-	c := NewRootCmd()
+	c := NewRootCmd(fakeruntime.FakeExecer{ExpectOS: "linux"})
 	assert.NotNil(t, c)
 	assert.Equal(t, "atest", c.Use)
 }

cmd/sample_test.go

@@ -6,11 +6,12 @@ import (
 
 	"github.com/linuxsuren/api-testing/cmd"
 	"github.com/linuxsuren/api-testing/sample"
+	fakeruntime "github.com/linuxsuren/go-fake-runtime"
 	"github.com/stretchr/testify/assert"
 )
 
 func TestSampleCmd(t *testing.T) {
-	c := cmd.NewRootCmd()
+	c := cmd.NewRootCmd(fakeruntime.FakeExecer{ExpectOS: "linux"})
 
 	buf := new(bytes.Buffer)
 	c.SetOut(buf)

cmd/server_test.go

@@ -5,6 +5,7 @@ import (
 	"strings"
 	"testing"
 
+	fakeruntime "github.com/linuxsuren/go-fake-runtime"
 	"github.com/stretchr/testify/assert"
 )
 
@@ -30,7 +31,7 @@ func TestPrintProto(t *testing.T) {
 	for _, tt := range tests {
 		t.Run(tt.name, func(t *testing.T) {
 			buf := new(bytes.Buffer)
-			root := NewRootCmd()
+			root := NewRootCmd(fakeruntime.FakeExecer{ExpectOS: "linux"})
 			root.SetOut(buf)
 			root.SetArgs(tt.args)
 			err := root.Execute()

cmd/service.go

@@ -3,13 +3,15 @@ package cmd
 import (
 	"fmt"
 	"os"
-	"runtime"
 
+	fakeruntime "github.com/linuxsuren/go-fake-runtime"
 	"github.com/spf13/cobra"
 )
 
-func createServiceCommand() (c *cobra.Command) {
-	opt := &serviceOption{}
+func createServiceCommand(execer fakeruntime.Execer) (c *cobra.Command) {
+	opt := &serviceOption{
+		Execer: execer,
+	}
 	c = &cobra.Command{
 		Use:     "service",
 		Aliases: []string{"s"},
@@ -18,28 +20,47 @@ func createServiceCommand() (c *cobra.Command) {
 		RunE:    opt.runE,
 	}
 	flags := c.Flags()
-	flags.StringVarP(&opt.action, "action", "a", "", "The action of service, support actions: install")
+	flags.StringVarP(&opt.action, "action", "a", "", "The action of service, support actions: install, start, stop, restart, status")
+	flags.StringVarP(&opt.scriptPath, "script-path", "", "/lib/systemd/system/atest.service", "The service script file path")
 	return
 }
 
 type serviceOption struct {
-	action string
+	action     string
+	scriptPath string
+	fakeruntime.Execer
 }
 
 func (o *serviceOption) preRunE(c *cobra.Command, args []string) (err error) {
-	if runtime.GOOS != "linux" {
+	if o.Execer.OS() != "linux" {
 		err = fmt.Errorf("only support on Linux")
 	}
+	if o.action == "" && len(args) > 0 {
+		o.action = args[0]
+	}
 	return
 }
 
 func (o *serviceOption) runE(c *cobra.Command, args []string) (err error) {
+	var output string
 	switch o.action {
 	case "install", "i":
-		err = os.WriteFile("/lib/systemd/system/atest.service", []byte(script), os.ModeAppend)
+		err = os.WriteFile(o.scriptPath, []byte(script), os.ModeAppend)
+	case "start":
+		output, err = o.Execer.RunCommandAndReturn("systemctl", "", "start", "atest")
+	case "stop":
+		output, err = o.Execer.RunCommandAndReturn("systemctl", "", "stop", "atest")
+	case "restart":
+		output, err = o.Execer.RunCommandAndReturn("systemctl", "", "restart", "atest")
+	case "status":
+		output, err = o.Execer.RunCommandAndReturn("systemctl", "", "status", "atest")
 	default:
 		err = fmt.Errorf("not support action: '%s'", o.action)
 	}
+
+	if output != "" {
+		c.Println(output)
+	}
 	return
 }
 

cmd/service_test.go

@@ -1,14 +1,69 @@
 package cmd
 
 import (
+	"bytes"
+	"os"
 	"testing"
 
+	fakeruntime "github.com/linuxsuren/go-fake-runtime"
 	"github.com/stretchr/testify/assert"
 )
 
 func TestService(t *testing.T) {
-	root := NewRootCmd()
-	root.SetArgs([]string{"service", "--action", "fake"})
+	root := NewRootCmd(fakeruntime.FakeExecer{ExpectOS: "linux"})
+	root.SetArgs([]string{"service", "fake"})
 	err := root.Execute()
 	assert.NotNil(t, err)
+
+	notLinux := NewRootCmd(fakeruntime.FakeExecer{ExpectOS: "fake"})
+	notLinux.SetArgs([]string{"service", "--action", "install"})
+	err = notLinux.Execute()
+	assert.NotNil(t, err)
+
+	tmpFile, err := os.CreateTemp(os.TempDir(), "service")
+	assert.Nil(t, err)
+	defer func() {
+		os.RemoveAll(tmpFile.Name())
+	}()
+
+	targetScript := NewRootCmd(fakeruntime.FakeExecer{ExpectOS: "linux"})
+	targetScript.SetArgs([]string{"service", "--action", "install", "--script-path", tmpFile.Name()})
+	err = targetScript.Execute()
+	assert.Nil(t, err)
+	data, err := os.ReadFile(tmpFile.Name())
+	assert.Nil(t, err)
+	assert.Equal(t, script, string(data))
+
+	tests := []struct {
+		name         string
+		action       string
+		expectOutput string
+	}{{
+		name:         "action: start",
+		action:       "start",
+		expectOutput: "output1",
+	}, {
+		name:         "action: stop",
+		action:       "stop",
+		expectOutput: "output2",
+	}, {
+		name:         "action: restart",
+		action:       "restart",
+		expectOutput: "output3",
+	}, {
+		name:         "action: status",
+		action:       "status",
+		expectOutput: "output4",
+	}}
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			buf := new(bytes.Buffer)
+			normalRoot := NewRootCmd(fakeruntime.FakeExecer{ExpectOS: "linux", ExpectOutput: tt.expectOutput})
+			normalRoot.SetOut(buf)
+			normalRoot.SetArgs([]string{"service", "--action", tt.action})
+			err = normalRoot.Execute()
+			assert.Nil(t, err)
+			assert.Equal(t, tt.expectOutput+"\n", buf.String())
+		})
+	}
 }

go.mod

@@ -28,6 +28,7 @@ require (
 	github.com/imdario/mergo v0.3.11 // indirect
 	github.com/inconshreveable/mousetrap v1.0.1 // indirect
 	github.com/invopop/jsonschema v0.7.0 // indirect
+	github.com/linuxsuren/go-fake-runtime v0.0.0-20230413085645-15e77ab55dbd // indirect
 	github.com/mitchellh/copystructure v1.0.0 // indirect
 	github.com/mitchellh/reflectwalk v1.0.0 // indirect
 	github.com/pmezard/go-difflib v1.0.0 // indirect

go.sum

@@ -563,6 +563,8 @@ github.com/kr/pretty v0.1.0/go.mod h1:dAy3ld7l9f0ibDNOQOHHMYYIIbhfbHSm3C4ZsoJORN
 github.com/kr/pty v1.1.1/go.mod h1:pFQYn66WHrOpPYNljwOMqo10TkYh1fy3cYio2l3bCsQ=
 github.com/kr/text v0.1.0 h1:45sCR5RtlFHMR4UwH9sdQ5TC8v0qDQCHnXt+kaKSTVE=
 github.com/kr/text v0.1.0/go.mod h1:4Jbv+DJW3UT/LiOwJeYQe1efqtUx/iVham/4vfdArNI=
+github.com/linuxsuren/go-fake-runtime v0.0.0-20230413085645-15e77ab55dbd h1:2Avir30WOgcDqG3sA4hlW4bC4c/tgseAUntPhf5JQ6E=
+github.com/linuxsuren/go-fake-runtime v0.0.0-20230413085645-15e77ab55dbd/go.mod h1:zmh6J78hSnWZo68faMA2eKOdaEp8eFbERHi3ZB9xHCQ=
 github.com/linuxsuren/unstructured v0.0.1 h1:ilUA8MUYbR6l9ebo/YPV2bKqlf62bzQursDSE+j00iU=
 github.com/linuxsuren/unstructured v0.0.1/go.mod h1:KH6aTj+FegzGBzc1vS6mzZx3/duhTUTEVyW5sO7p4as=
 github.com/lyft/protoc-gen-star v0.6.0/go.mod h1:TGAoBVkt8w7MPG72TrKIu85MIdXwDuzJYeZuUPFPNwA=

main.go

@@ -4,10 +4,11 @@ import (
 	"os"
 
 	"github.com/linuxsuren/api-testing/cmd"
+	exec "github.com/linuxsuren/go-fake-runtime"
 )
 
 func main() {
-	c := cmd.NewRootCmd()
+	c := cmd.NewRootCmd(exec.DefaultExecer{})
 	if err := c.Execute(); err != nil {
 		os.Exit(1)
 	}