Add labels to request response pair

SpectoLabs · Jul 11, 2024 · b133acb · b133acb
1 parent d0a5551
commit b133acb
Show file tree

Hide file tree

Showing 5 changed files with 58 additions and 49 deletions.
diff --git a/core/handlers/v2/schema.json b/core/handlers/v2/schema.json
@@ -161,6 +161,12 @@
 		},
 		"request-response-pair": {
 			"properties": {
+				"labels": {
+					"items": {
+						"type": "string"
+					},
+					"type": "array"
+				},
 				"request": {
 					"$ref": "#/definitions/request"
 				},

diff --git a/core/handlers/v2/simulation_views.go b/core/handlers/v2/simulation_views.go
@@ -32,7 +32,7 @@ func NewSimulationViewFromRequestBody(requestBody []byte) (SimulationViewV5, err
 
 	schemaVersion := jsonMap["meta"].(map[string]interface{})["schemaVersion"].(string)
 
-	if schemaVersion == "v5" || schemaVersion == "v5.1" || schemaVersion == "v5.2" {
+	if strings.HasPrefix(schemaVersion, "v5") {
 
 		err := ValidateSimulationSchemaFromFile(jsonMap, SimulationViewV5Schema)
 		if err != nil {

diff --git a/core/handlers/v2/simulation_views_v5.go b/core/handlers/v2/simulation_views_v5.go
@@ -17,6 +17,7 @@ type DataViewV5 struct {
 }
 
 type RequestMatcherResponsePairViewV5 struct {
+	Labels		   []string				 `json:"labels,omitempty"`
 	RequestMatcher RequestMatcherViewV5  `json:"request"`
 	Response       ResponseDetailsViewV5 `json:"response"`
 }

diff --git a/core/models/request_matcher.go b/core/models/request_matcher.go
@@ -142,6 +142,7 @@ func getViewFromRequestFieldMatcher(matcher *RequestFieldMatchers) *v2.MatcherVi
 }
 
 type RequestMatcherResponsePair struct {
+	Labels		   []string
 	RequestMatcher RequestMatcher
 	Response       ResponseDetails
 }
@@ -229,6 +230,7 @@ func (this *RequestMatcherResponsePair) BuildView() v2.RequestMatcherResponsePai
 	}
 
 	return v2.RequestMatcherResponsePairViewV5{
+		Labels: this.Labels,
 		RequestMatcher: v2.RequestMatcherViewV5{
 			Path:          path,
 			Method:        method,