feat: adding resource loader interface and default implement

Author: LinuxSuRen

cmd/run.go

@@ -5,8 +5,6 @@ import (
 	"fmt"
 	"io"
 	"os"
-	"path"
-	"path/filepath"
 	"strings"
 	"sync"
 	"time"
@@ -16,7 +14,6 @@ import (
 	"github.com/linuxsuren/api-testing/pkg/render"
 	"github.com/linuxsuren/api-testing/pkg/runner"
 	"github.com/linuxsuren/api-testing/pkg/testing"
-	"github.com/linuxsuren/api-testing/pkg/util"
 	"github.com/spf13/cobra"
 	"golang.org/x/sync/semaphore"
 )
@@ -134,19 +131,14 @@ func (o *runOption) runE(cmd *cobra.Command, args []string) (err error) {
 		o.limiter.Stop()
 	}()
 
-	var suites []string
-	for _, pattern := range util.Expand(o.pattern) {
-		var files []string
-		if files, err = filepath.Glob(pattern); err == nil {
-			suites = append(suites, files...)
-		}
+	loader := testing.NewFileLoader()
+	if err = loader.Put(o.pattern); err != nil {
+		return
 	}
 
-	cmd.Println("found suites:", len(suites))
-	for i := range suites {
-		item := suites[i]
-		cmd.Println("run suite:", item)
-		if err = o.runSuiteWithDuration(item); err != nil {
+	cmd.Println("found suites:", loader.GetCount())
+	for loader.HasMore() {
+		if err = o.runSuiteWithDuration(loader); err != nil {
 			break
 		}
 	}
@@ -166,7 +158,7 @@ func (o *runOption) runE(cmd *cobra.Command, args []string) (err error) {
 	return
 }
 
-func (o *runOption) runSuiteWithDuration(suite string) (err error) {
+func (o *runOption) runSuiteWithDuration(loader testing.Loader) (err error) {
 	sem := semaphore.NewWeighted(o.thread)
 	stop := false
 	var timeout *time.Ticker
@@ -204,7 +196,7 @@ func (o *runOption) runSuiteWithDuration(suite string) (err error) {
 				}()
 
 				dataContext := getDefaultContext()
-				ch <- o.runSuite(suite, dataContext, o.context, stopSingal)
+				ch <- o.runSuite(loader, dataContext, o.context, stopSingal)
 			}(errChannel, sem)
 			if o.duration <= 0 {
 				stop = true
@@ -221,9 +213,14 @@ func (o *runOption) runSuiteWithDuration(suite string) (err error) {
 	return
 }
 
-func (o *runOption) runSuite(suite string, dataContext map[string]interface{}, ctx context.Context, stopSingal chan struct{}) (err error) {
+func (o *runOption) runSuite(loader testing.Loader, dataContext map[string]interface{}, ctx context.Context, stopSingal chan struct{}) (err error) {
+	var data []byte
+	if data, err = loader.Load(); err != nil {
+		return
+	}
+
 	var testSuite *testing.TestSuite
-	if testSuite, err = testing.Parse(suite); err != nil {
+	if testSuite, err = testing.Parse(data); err != nil {
 		return
 	}
 
@@ -253,7 +250,7 @@ func (o *runOption) runSuite(suite string, dataContext map[string]interface{}, c
 			o.limiter.Accept()
 
 			ctxWithTimeout, _ := context.WithTimeout(ctx, o.requestTimeout)
-			ctxWithTimeout = context.WithValue(ctxWithTimeout, runner.ContextKey("").ParentDir(), path.Dir(suite))
+			ctxWithTimeout = context.WithValue(ctxWithTimeout, runner.ContextKey("").ParentDir(), loader.GetContext())
 
 			simpleRunner := runner.NewSimpleTestCaseRunner()
 			simpleRunner.WithTestReporter(o.reporter)

cmd/run_test.go

@@ -12,6 +12,7 @@ import (
 
 	"github.com/h2non/gock"
 	"github.com/linuxsuren/api-testing/pkg/limit"
+	atest "github.com/linuxsuren/api-testing/pkg/testing"
 	"github.com/linuxsuren/api-testing/pkg/util"
 	fakeruntime "github.com/linuxsuren/go-fake-runtime"
 	"github.com/spf13/cobra"
@@ -58,8 +59,13 @@ func TestRunSuite(t *testing.T) {
 			opt.limiter = limit.NewDefaultRateLimiter(0, 0)
 			stopSingal := make(chan struct{}, 1)
 
-			err := opt.runSuite(tt.suiteFile, ctx, context.TODO(), stopSingal)
-			assert.Equal(t, tt.hasError, err != nil, err)
+			loader := atest.NewFileLoader()
+			err := loader.Put(tt.suiteFile)
+			assert.NoError(t, err)
+			if loader.HasMore() {
+				err = opt.runSuite(loader, ctx, context.TODO(), stopSingal)
+				assert.Equal(t, tt.hasError, err != nil, err)
+			}
 		})
 	}
 }

pkg/testing/loader.go

@@ -0,0 +1,10 @@
+package testing
+
+// Loader is an interface for test cases loader
+type Loader interface {
+	HasMore() bool
+	Load() ([]byte, error)
+	Put(string) (err error)
+	GetContext() string
+	GetCount() int
+}

pkg/testing/loader_file.go

@@ -0,0 +1,52 @@
+package testing
+
+import (
+	"os"
+	"path"
+	"path/filepath"
+
+	"github.com/linuxsuren/api-testing/pkg/util"
+)
+
+type fileLoader struct {
+	paths []string
+	index int
+}
+
+// NewFileLoader creates the instance of file loader
+func NewFileLoader() Loader {
+	return &fileLoader{index: -1}
+}
+
+// HasMore returns if there are more test cases
+func (l *fileLoader) HasMore() bool {
+	l.index++
+	return l.index < len(l.paths)
+}
+
+// Load returns the test case content
+func (l *fileLoader) Load() (data []byte, err error) {
+	data, err = os.ReadFile(l.paths[l.index])
+	return
+}
+
+// Put adds the test case path
+func (l *fileLoader) Put(item string) (err error) {
+	for _, pattern := range util.Expand(item) {
+		var files []string
+		if files, err = filepath.Glob(pattern); err == nil {
+			l.paths = append(l.paths, files...)
+		}
+	}
+	return
+}
+
+// GetContext returns the context of current test case
+func (l *fileLoader) GetContext() string {
+	return path.Dir(l.paths[l.index])
+}
+
+// GetCount returns the count of test cases
+func (l *fileLoader) GetCount() int {
+	return len(l.paths)
+}

pkg/testing/loader_file_test.go

@@ -0,0 +1,56 @@
+package testing_test
+
+import (
+	"testing"
+
+	atest "github.com/linuxsuren/api-testing/pkg/testing"
+	"github.com/stretchr/testify/assert"
+)
+
+func TestFileLoader(t *testing.T) {
+	tests := []struct {
+		name   string
+		items  []string
+		verify func(t *testing.T, loader atest.Loader)
+	}{{
+		name:  "empty",
+		items: []string{},
+		verify: func(t *testing.T, loader atest.Loader) {
+			assert.False(t, loader.HasMore())
+			assert.Empty(t, loader.GetCount())
+		},
+	}, {
+		name:   "brace expansion path",
+		items:  []string{"testdata/{invalid-,}testcase.yaml"},
+		verify: defaultVerify,
+	}, {
+		name:   "glob path",
+		items:  []string{"testdata/*testcase.yaml"},
+		verify: defaultVerify,
+	}}
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			loader := atest.NewFileLoader()
+			for _, item := range tt.items {
+				loader.Put(item)
+			}
+			tt.verify(t, loader)
+		})
+	}
+}
+
+func defaultVerify(t *testing.T, loader atest.Loader) {
+	assert.True(t, loader.HasMore())
+	data, err := loader.Load()
+	assert.Nil(t, err)
+	assert.Equal(t, invalidTestCaseContent, string(data))
+	assert.Equal(t, "testdata", loader.GetContext())
+
+	assert.True(t, loader.HasMore())
+	data, err = loader.Load()
+	assert.Nil(t, err)
+	assert.Equal(t, testCaseContent, string(data))
+	assert.Equal(t, "testdata", loader.GetContext())
+
+	assert.False(t, loader.HasMore())
+}

pkg/testing/parser.go

@@ -19,11 +19,8 @@ import (
 )
 
 // Parse parses a file and returns the test suite
-func Parse(configFile string) (testSuite *TestSuite, err error) {
-	var data []byte
-	if data, err = os.ReadFile(configFile); err == nil {
-		testSuite, err = ParseFromData(data)
-	}
+func Parse(data []byte) (testSuite *TestSuite, err error) {
+	testSuite, err = ParseFromData(data)
 
 	// schema validation
 	if err == nil {
@@ -116,7 +113,7 @@ func (r *Request) Render(ctx interface{}, dataDir string) (err error) {
 	}
 
 	// setting default values
-	r.Method = emptyThenDefault(r.Method, http.MethodGet)
+	r.Method = EmptyThenDefault(r.Method, http.MethodGet)
 	return
 }
 
@@ -153,18 +150,20 @@ func (r *Request) GetBody() (reader io.Reader, err error) {
 
 // Render renders the response
 func (r *Response) Render(ctx interface{}) (err error) {
-	r.StatusCode = zeroThenDefault(r.StatusCode, http.StatusOK)
+	r.StatusCode = ZeroThenDefault(r.StatusCode, http.StatusOK)
 	return
 }
 
-func zeroThenDefault(val, defVal int) int {
+// ZeroThenDefault return the default value if the val is zero
+func ZeroThenDefault(val, defVal int) int {
 	if val == 0 {
 		val = defVal
 	}
 	return val
 }
 
-func emptyThenDefault(val, defVal string) string {
+// EmptyThenDefault return the default value if the val is empty
+func EmptyThenDefault(val, defVal string) string {
 	if strings.TrimSpace(val) == "" {
 		val = defVal
 	}