From e7a38eb3f9964492dfc8ad23244b108711f9b5e3 Mon Sep 17 00:00:00 2001
From: rick <1450685+LinuxSuRen@users.noreply.github.com>
Date: Mon, 10 Apr 2023 10:53:32 +0800
Subject: [PATCH] feat: support with nested parent test cases

---
 pkg/server/remote_server.go      | 59 ++++++++++++++++++++++++++++----
 pkg/server/remote_server_test.go | 43 +++++++++++++++++++++++
 2 files changed, 96 insertions(+), 6 deletions(-)

diff --git a/pkg/server/remote_server.go b/pkg/server/remote_server.go
index b272af1b..1ff91b24 100644
--- a/pkg/server/remote_server.go
+++ b/pkg/server/remote_server.go
@@ -46,6 +46,7 @@ func (s *server) Run(ctx context.Context, task *TestTask) (reply *HelloReply, er
 		for key, val := range oldEnv {
 			os.Setenv(key, val)
 		}
+		fmt.Println(reply, err)
 	}()
 
 	switch task.Kind {
@@ -82,6 +83,7 @@ func (s *server) Run(ctx context.Context, task *TestTask) (reply *HelloReply, er
 
 		if targetTestcase != nil {
 			parentCases := findParentTestCases(targetTestcase, suite)
+			fmt.Printf("find %d parent cases\n", len(parentCases))
 			suite.Items = append(parentCases, *targetTestcase)
 		} else {
 			err = fmt.Errorf("cannot found testcase %s", task.CaseName)
@@ -101,6 +103,8 @@ func (s *server) Run(ctx context.Context, task *TestTask) (reply *HelloReply, er
 		suite.API = result
 		suite.API = strings.TrimSuffix(suite.API, "/")
 	} else {
+		reply.Error = err.Error()
+		err = nil
 		return
 	}
 
@@ -138,28 +142,71 @@ func findParentTestCases(testcase *testing.TestCase, suite *testing.TestSuite) (
 	reg, matchErr := regexp.Compile(`.*\{\{.*\.\w*.*}\}.*`)
 	targetReg, targetErr := regexp.Compile(`\.\w*`)
 
+	expectNames := new(UniqueSlice[string])
 	if matchErr == nil && targetErr == nil {
-		expectName := ""
+		var expectName string
 		for _, val := range testcase.Request.Header {
 			if matched := reg.MatchString(val); matched {
 				expectName = targetReg.FindString(val)
 				expectName = strings.TrimPrefix(expectName, ".")
-				break
+				expectNames.Push(expectName)
 			}
 		}
 
-		if expectName == "" {
-			if mached := reg.MatchString(testcase.Request.API); mached {
-				expectName = targetReg.FindString(testcase.Request.API)
+		if mached := reg.MatchString(testcase.Request.API); mached {
+			// remove {{ and }}
+			if left, leftErr := regexp.Compile(`.*\{\{`); leftErr == nil {
+				api := left.ReplaceAllString(testcase.Request.API, "")
+
+				expectName = targetReg.FindString(api)
 				expectName = strings.TrimPrefix(expectName, ".")
+				expectNames.Push(expectName)
 			}
 		}
 
+		fmt.Println("expect test case names", expectNames.GetAll())
 		for _, item := range suite.Items {
-			if item.Name == expectName {
+			if expectNames.Exist(item.Name) {
 				testcases = append(testcases, item)
 			}
 		}
 	}
 	return
 }
+
+// UniqueSlice represents an unique slice
+type UniqueSlice[T comparable] struct {
+	data []T
+}
+
+// Push pushes an item if it's not exist
+func (s *UniqueSlice[T]) Push(item T) *UniqueSlice[T] {
+	if s.data == nil {
+		s.data = []T{item}
+	} else {
+		for _, it := range s.data {
+			if it == item {
+				return s
+			}
+		}
+		s.data = append(s.data, item)
+	}
+	return s
+}
+
+// Exist checks if the item exist, return true it exists
+func (s *UniqueSlice[T]) Exist(item T) bool {
+	if s.data != nil {
+		for _, it := range s.data {
+			if it == item {
+				return true
+			}
+		}
+	}
+	return false
+}
+
+// GetAll returns all the items
+func (s *UniqueSlice[T]) GetAll() []T {
+	return s.data
+}
diff --git a/pkg/server/remote_server_test.go b/pkg/server/remote_server_test.go
index dba0b089..40b1686e 100644
--- a/pkg/server/remote_server_test.go
+++ b/pkg/server/remote_server_test.go
@@ -107,6 +107,43 @@ func TestFindParentTestCases(t *testing.T) {
 		expect: []atesting.TestCase{{
 			Name: "login",
 		}},
+	}, {
+		name: "nest dep",
+		testcase: &atesting.TestCase{
+			Name: "user",
+			Request: atesting.Request{
+				API: "/users/{{(index .users 0).name}}",
+				Header: map[string]string{
+					"Authorization": "Bearer {{.login.data.access_token}}",
+				},
+			},
+		},
+		suite: &atesting.TestSuite{
+			Items: []atesting.TestCase{{
+				Name: "login",
+			}, {
+				Name: "users",
+				Request: atesting.Request{
+					API: "/users",
+				},
+			}, {
+				Name: "user",
+				Request: atesting.Request{
+					API: "/users/{{(index .users 0).name}}",
+					Header: map[string]string{
+						"Authorization": "Bearer {{.login.data.access_token}}",
+					},
+				},
+			}},
+		},
+		expect: []atesting.TestCase{{
+			Name: "login",
+		}, {
+			Name: "users",
+			Request: atesting.Request{
+				API: "/users",
+			},
+		}},
 	}}
 	for _, tt := range tests {
 		t.Run(tt.name, func(t *testing.T) {
@@ -116,6 +153,12 @@ func TestFindParentTestCases(t *testing.T) {
 	}
 }
 
+func TestUniqueSlice(t *testing.T) {
+	uniqueSlice := new(UniqueSlice[string])
+	uniqueSlice.Push("a").Push("a").Push("b")
+	assert.Equal(t, []string{"a", "b"}, uniqueSlice.GetAll())
+}
+
 //go:embed testdata/simple.yaml
 var simpleSuite string