fix the duplicates

cmd/run_test.go

@@ -8,6 +8,7 @@ import (
 
 	"github.com/h2non/gock"
 	"github.com/linuxsuren/api-testing/pkg/limit"
+	"github.com/linuxsuren/api-testing/pkg/util"
 	fakeruntime "github.com/linuxsuren/go-fake-runtime"
 	"github.com/spf13/cobra"
 	"github.com/stretchr/testify/assert"
@@ -41,14 +42,12 @@ func TestRunSuite(t *testing.T) {
 	}, {
 		name:      "not found file",
 		suiteFile: "testdata/fake.yaml",
-		prepare:   func() {},
 		hasError:  true,
 	}}
 	for _, tt := range tests {
 		t.Run(tt.name, func(t *testing.T) {
 			defer gock.Clean()
-
-			tt.prepare()
+			util.MakeSureNotNil(tt.prepare)()
 			ctx := getDefaultContext()
 			opt := newDiskCardRunOption()
 			opt.requestTimeout = 30 * time.Second
@@ -75,10 +74,9 @@ func TestRunCommand(t *testing.T) {
 		},
 		hasErr: true,
 	}, {
-		name:    "file not found",
-		args:    []string{"--pattern", "fake"},
-		prepare: func() {},
-		hasErr:  false,
+		name:   "file not found",
+		args:   []string{"--pattern", "fake"},
+		hasErr: false,
 	}, {
 		name: "normal case",
 		args: []string{"-p", simpleSuite},
@@ -90,8 +88,7 @@ func TestRunCommand(t *testing.T) {
 	for _, tt := range tests {
 		t.Run(tt.name, func(t *testing.T) {
 			defer gock.Clean()
-			tt.prepare()
-
+			util.MakeSureNotNil(tt.prepare)()
 			root := &cobra.Command{Use: "root"}
 			root.AddCommand(createRunCommand())
 

cmd/server.go

@@ -71,4 +71,5 @@ func (s *fakeGRPCServer) Serve(net.Listener) error {
 
 // RegisterService is a fake method
 func (s *fakeGRPCServer) RegisterService(desc *grpc.ServiceDesc, impl interface{}) {
+	// Do nothing due to this is a fake method
 }

cmd/service_test.go

@@ -16,7 +16,7 @@ func TestService(t *testing.T) {
 	assert.NotNil(t, err)
 
 	notLinux := NewRootCmd(fakeruntime.FakeExecer{ExpectOS: "fake"}, NewFakeGRPCServer())
-	notLinux.SetArgs([]string{"service", "--action", "install"})
+	notLinux.SetArgs([]string{"service", paramAction, "install"})
 	err = notLinux.Execute()
 	assert.NotNil(t, err)
 
@@ -27,7 +27,7 @@ func TestService(t *testing.T) {
 	}()
 
 	targetScript := NewRootCmd(fakeruntime.FakeExecer{ExpectOS: "linux"}, NewFakeGRPCServer())
-	targetScript.SetArgs([]string{"service", "--action", "install", "--script-path", tmpFile.Name()})
+	targetScript.SetArgs([]string{"service", paramAction, "install", "--script-path", tmpFile.Name()})
 	err = targetScript.Execute()
 	assert.Nil(t, err)
 	data, err := os.ReadFile(tmpFile.Name())
@@ -67,3 +67,5 @@ func TestService(t *testing.T) {
 		})
 	}
 }
+
+const paramAction = "--action"

pkg/runner/kubernetes/client_test.go

@@ -27,7 +27,7 @@ func TestGetPod(t *testing.T) {
 			name:      "fake",
 		},
 		prepare: func() {
-			gock.New("http://foo").
+			gock.New(urlFoo).
 				Get("/api/v1/namespaces/ns/pods/fake").
 				Reply(http.StatusOK).
 				JSON(`{"kind":"pod"}`)
@@ -46,7 +46,7 @@ func TestGetPod(t *testing.T) {
 			name:      "fake",
 		},
 		prepare: func() {
-			gock.New("http://foo").
+			gock.New(urlFoo).
 				Get("/apis/apps/v1/namespaces/ns/deployments/fake").
 				Reply(http.StatusOK).
 				JSON(`{"kind":"deployment"}`)
@@ -60,7 +60,7 @@ func TestGetPod(t *testing.T) {
 		t.Run(tt.name, func(t *testing.T) {
 			defer gock.Clean()
 			tt.prepare()
-			reader := kubernetes.NewDefaultReader("http://foo", "")
+			reader := kubernetes.NewDefaultReader(urlFoo, "")
 			result, err := reader.GetResource(tt.group, tt.kind, tt.version, tt.namespacedName.namespace, tt.namespacedName.name)
 			assert.Equal(t, tt.expect, result)
 			assert.Nil(t, err)

pkg/runner/kubernetes/verify_test.go

@@ -8,6 +8,7 @@ import (
 	"github.com/antonmedv/expr"
 	"github.com/h2non/gock"
 	"github.com/linuxsuren/api-testing/pkg/runner/kubernetes"
+	"github.com/linuxsuren/api-testing/pkg/util"
 	"github.com/stretchr/testify/assert"
 )
 
@@ -47,7 +48,6 @@ func TestKubernetesValidatorFunc(t *testing.T) {
 	}, {
 		name:       "no enough params",
 		expression: `k8s('crd')`,
-		prepare:    emptyPrepare,
 		expectBool: false,
 		expectErr:  true,
 	}, {
@@ -73,11 +73,11 @@ func TestKubernetesValidatorFunc(t *testing.T) {
 	}, {
 		name:       "no kind",
 		expression: `k8s({"foo": "bar"}, "ns", "foo").Exist()`,
-		prepare:    emptyPrepare,
 		expectErr:  true,
 	}}
 	for _, tt := range tests {
 		t.Run(tt.name, func(t *testing.T) {
+			tt.prepare = util.MakeSureNotNil(tt.prepare)
 			tt.prepare()
 			vm, err := expr.Compile(tt.expression, kubernetes.KubernetesValidatorFunc(),
 				kubernetes.PodValidatorFunc())
@@ -92,10 +92,6 @@ func TestKubernetesValidatorFunc(t *testing.T) {
 	}
 }
 
-func emptyPrepare() {
-	// only for testing
-}
-
 func preparePod() {
 	gock.New(urlFoo).
 		Get("/api/v1/namespaces/ns/pods/foo").

pkg/runner/simple.go

@@ -1,16 +1,12 @@
 package runner
 
 import (
-	"bytes"
 	"context"
 	"crypto/tls"
 	"encoding/json"
 	"fmt"
 	"io"
-	"mime/multipart"
 	"net/http"
-	"net/url"
-	"os"
 	"reflect"
 	"strings"
 	"time"
@@ -202,40 +198,14 @@ func (r *simpleTestCaseRunner) RunTestCase(testcase *testing.TestCase, dataConte
 	}
 
 	var requestBody io.Reader
-	if testcase.Request.Body != "" {
-		requestBody = bytes.NewBufferString(testcase.Request.Body)
-	} else if testcase.Request.BodyFromFile != "" {
-		var data []byte
-		if data, err = os.ReadFile(testcase.Request.BodyFromFile); err != nil {
-			return
-		}
-		requestBody = bytes.NewBufferString(string(data))
+	if requestBody, err = testcase.Request.GetBody(); err != nil {
+		return
 	}
 
 	if err = testcase.Request.Render(dataContext); err != nil {
 		return
 	}
 
-	if len(testcase.Request.Form) > 0 {
-		if testcase.Request.Header[contentType] == "multipart/form-data" {
-			multiBody := &bytes.Buffer{}
-			writer := multipart.NewWriter(multiBody)
-			for key, val := range testcase.Request.Form {
-				writer.WriteField(key, val)
-			}
-
-			_ = writer.Close()
-			requestBody = multiBody
-			testcase.Request.Header[contentType] = writer.FormDataContentType()
-		} else if testcase.Request.Header[contentType] == "application/x-www-form-urlencoded" {
-			data := url.Values{}
-			for key, val := range testcase.Request.Form {
-				data.Set(key, val)
-			}
-			requestBody = strings.NewReader(data.Encode())
-		}
-	}
-
 	var request *http.Request
 	if request, err = http.NewRequestWithContext(ctx, testcase.Request.Method, testcase.Request.API, requestBody); err != nil {
 		return
@@ -436,5 +406,3 @@ func jsonSchemaValidation(schema string, body []byte) (err error) {
 	}
 	return
 }
-
-const contentType = "Content-Type"

pkg/runner/simple_test.go

@@ -12,6 +12,7 @@ import (
 
 	"github.com/h2non/gock"
 	atest "github.com/linuxsuren/api-testing/pkg/testing"
+	"github.com/linuxsuren/api-testing/pkg/util"
 	fakeruntime "github.com/linuxsuren/go-fake-runtime"
 	"github.com/stretchr/testify/assert"
 )
@@ -300,7 +301,7 @@ func TestTestCase(t *testing.T) {
 					API:    urlFoo,
 					Method: http.MethodPost,
 					Header: map[string]string{
-						contentType: "multipart/form-data",
+						util.ContentType: "multipart/form-data",
 					},
 					Form: map[string]string{
 						"key": "value",
@@ -319,7 +320,7 @@ func TestTestCase(t *testing.T) {
 					API:    urlFoo,
 					Method: http.MethodPost,
 					Header: map[string]string{
-						contentType: "application/x-www-form-urlencoded",
+						util.ContentType: "application/x-www-form-urlencoded",
 					},
 					Form: map[string]string{
 						"key": "value",

pkg/server/remote_server_test.go

@@ -19,30 +19,30 @@ func TestRemoteServer(t *testing.T) {
 	})
 	assert.NotNil(t, err)
 
-	gock.New("http://foo").Get("/").Reply(http.StatusOK).JSON(&server)
-	gock.New("http://foo").Get("/").Reply(http.StatusOK).JSON(&server)
+	gock.New(urlFoo).Get("/").Reply(http.StatusOK).JSON(&server)
+	gock.New(urlFoo).Get("/").Reply(http.StatusOK).JSON(&server)
 	_, err = server.Run(context.TODO(), &TestTask{
 		Kind: "suite",
 		Data: simpleSuite,
 	})
 	assert.Nil(t, err)
 
-	gock.New("http://bar").Get("/").Reply(http.StatusOK).JSON(&server)
+	gock.New(urlFoo).Get("/").Reply(http.StatusOK).JSON(&server)
 	_, err = server.Run(context.TODO(), &TestTask{
 		Kind: "testcase",
 		Data: simpleTestCase,
 	})
 	assert.Nil(t, err)
 
-	gock.New("http://foo").Get("/").Reply(http.StatusOK).JSON(&server)
+	gock.New(urlFoo).Get("/").Reply(http.StatusOK).JSON(&server)
 	_, err = server.Run(context.TODO(), &TestTask{
 		Kind:     "testcaseInSuite",
 		Data:     simpleSuite,
 		CaseName: "get",
 	})
 	assert.Nil(t, err)
 
-	gock.New("http://foo").Get("/").Reply(http.StatusOK).JSON(&server)
+	gock.New(urlFoo).Get("/").Reply(http.StatusOK).JSON(&server)
 	_, err = server.Run(context.TODO(), &TestTask{
 		Kind:     "testcaseInSuite",
 		Data:     simpleSuite,
@@ -171,3 +171,5 @@ var simpleSuite string
 
 //go:embed testdata/simple_testcase.yaml
 var simpleTestCase string
+
+const urlFoo = "http://foo"

pkg/testing/parser.go

@@ -1,12 +1,17 @@
 package testing
 
 import (
+	"bytes"
 	"fmt"
+	"io"
+	"mime/multipart"
 	"net/http"
+	"net/url"
 	"os"
 	"strings"
 
 	"github.com/linuxsuren/api-testing/pkg/render"
+	"github.com/linuxsuren/api-testing/pkg/util"
 	"gopkg.in/yaml.v2"
 )
 
@@ -83,6 +88,37 @@ func (r *Request) Render(ctx interface{}) (err error) {
 	return
 }
 
+// GetBody returns the request body
+func (r *Request) GetBody() (reader io.Reader, err error) {
+	if len(r.Form) > 0 {
+		if r.Header[util.ContentType] == util.MultiPartFormData {
+			multiBody := &bytes.Buffer{}
+			writer := multipart.NewWriter(multiBody)
+			for key, val := range r.Form {
+				writer.WriteField(key, val)
+			}
+
+			_ = writer.Close()
+			reader = multiBody
+			r.Header[util.ContentType] = writer.FormDataContentType()
+		} else if r.Header[util.ContentType] == util.Form {
+			data := url.Values{}
+			for key, val := range r.Form {
+				data.Set(key, val)
+			}
+			reader = strings.NewReader(data.Encode())
+		}
+	} else if r.Body != "" {
+		reader = bytes.NewBufferString(r.Body)
+	} else if r.BodyFromFile != "" {
+		var data []byte
+		if data, err = os.ReadFile(r.BodyFromFile); err == nil {
+			reader = bytes.NewBufferString(string(data))
+		}
+	}
+	return
+}
+
 // Render renders the response
 func (r *Response) Render(ctx interface{}) (err error) {
 	r.StatusCode = zeroThenDefault(r.StatusCode, http.StatusOK)