array

Author: skudasov

framework/components/fake/record.go

@@ -23,14 +23,14 @@ type Record struct {
 }
 
 type Records struct {
-	r map[string]*Record
+	r map[string][]*Record
 }
 
 func NewRecords() *Records {
-	return &Records{make(map[string]*Record)}
+	return &Records{make(map[string][]*Record)}
 }
 
-func (r *Records) Get(path string) (*Record, error) {
+func (r *Records) Get(path string) ([]*Record, error) {
 	rec, ok := r.r[path]
 	if !ok {
 		return nil, fmt.Errorf("no record was found for path: %s", path)
@@ -74,13 +74,17 @@ func recordMiddleware() gin.HandlerFunc {
 		// Capture response data
 		resBody := customWriter.body.String()
 		status := c.Writer.Status()
-		R.r[c.Request.URL.Path] = &Record{
+		if R.r[c.Request.URL.Path] == nil {
+			R.r[c.Request.URL.Path] = make([]*Record, 0)
+		}
+		R.r[c.Request.URL.Path] = append(R.r[c.Request.URL.Path], &Record{
 			Method:  c.Request.Method,
 			Path:    c.Request.URL.Path,
 			Headers: c.Request.Header,
 			ReqBody: reqBody,
 			ResBody: resBody,
 			Status:  status,
-		}
+		},
+		)
 	}
 }

framework/components/fake/record_test.go

@@ -50,18 +50,20 @@ func TestComponentFake(t *testing.T) {
 
 		// get request and response
 		recordedData, err := fake.R.Get(apiPath)
-		require.Equal(t, &fake.Record{
-			Method: "POST",
-			Path:   apiPath,
-			Headers: http.Header{
-				"Accept-Encoding": []string{"gzip"},
-				"Content-Type":    []string{"application/json"},
-				"Content-Length":  []string{"25"},
-				"User-Agent":      []string{"go-resty/2.15.3 (https://github.com/go-resty/resty)"},
+		require.Equal(t, []*fake.Record{
+			{
+				Method: "POST",
+				Path:   apiPath,
+				Headers: http.Header{
+					"Accept-Encoding": []string{"gzip"},
+					"Content-Type":    []string{"application/json"},
+					"Content-Length":  []string{"25"},
+					"User-Agent":      []string{"go-resty/2.15.3 (https://github.com/go-resty/resty)"},
+				},
+				ReqBody: `{"some_data":"some_data"}`,
+				ResBody: `{"status":"ok"}`,
+				Status:  200,
 			},
-			ReqBody: `{"some_data":"some_data"}`,
-			ResBody: `{"status":"ok"}`,
-			Status:  200,
 		}, recordedData)
 	})
 }