Entity: Added entity to deserialize response

Used by GetDetectorController to deserialize response from gateway to type.
opendistro-for-elasticsearch · Aug 12, 2020 · 0848abc · 0848abc
1 parent 17da589
commit 0848abc
Show file tree

Hide file tree

Showing 3 changed files with 109 additions and 0 deletions.
diff --git a/cli/internal/controller/ad/testdata/get_response.json b/cli/internal/controller/ad/testdata/get_response.json
diff --git a/cli/internal/entity/ad/ad.go b/cli/internal/entity/ad/ad.go
@@ -143,3 +143,32 @@ type Container struct {
 type SearchResponse struct {
 	Hits Container `json:"hits"`
 }
+
+type Metadata CreateDetector
+
+type AnomalyDetector struct {
+	Metadata
+	SchemaVersion  int32  `json:"schema_version"`
+	LastUpdateTime uint64 `json:"last_update_time"`
+}
+
+//DetectorResponse represents detector's setting
+type DetectorResponse struct {
+	ID              string          `json:"_id"`
+	AnomalyDetector AnomalyDetector `json:"anomaly_detector"`
+}
+
+//DetectorOutput represents detector's setting displayed to user
+type DetectorOutput struct {
+	ID            string
+	Name          string          `json:"name"`
+	Description   string          `json:"description"`
+	TimeField     string          `json:"time_field"`
+	Index         []string        `json:"indices"`
+	Features      []Feature       `json:"features"`
+	Filter        json.RawMessage `json:"filter_query"`
+	Interval      string          `json:"detection_interval"`
+	Delay         string          `json:"window_delay"`
+	LastUpdatedAt uint64          `json:"last_update_time"`
+	SchemaVersion int32           `json:"schema_version"`
+}
diff --git a/cli/internal/entity/ad/ad_test.go b/cli/internal/entity/ad/ad_test.go
@@ -174,3 +174,83 @@ func getSearchRequest() interface{} {
 		},
 	}
 }
+
+func TestGetDetectorResponse(t *testing.T) {
+	t.Run("deserialization success", func(t *testing.T) {
+		var actual DetectorResponse
+		expected := DetectorResponse{
+			ID: "m4ccEnIBTXsGi3mvMt9p",
+			AnomalyDetector: AnomalyDetector{
+				Metadata: Metadata{
+					Name:        "test-detector",
+					Description: "Test detector",
+					TimeField:   "timestamp",
+					Index:       []string{"order*"},
+					Features: []Feature{
+						{
+							Name:             "total_order",
+							Enabled:          true,
+							AggregationQuery: []byte(`{"total_order":{"sum":{"field":"value"}}}`),
+						},
+					},
+					Filter: []byte(`{"bool" : {"filter" : [{"exists" : {"field" : "value","boost" : 1.0}}],"adjust_pure_negative" : true,"boost" : 1.0}}`),
+					Interval: Interval{
+						Period: Period{
+							Duration: 1,
+							Unit:     "Minutes",
+						},
+					},
+					Delay: Interval{
+						Period: Period{
+							Duration: 1,
+							Unit:     "Minutes",
+						},
+					},
+				},
+				SchemaVersion:  0,
+				LastUpdateTime: 1589441737319,
+			},
+		}
+		responseJSON := `
+		{
+		  "_id" : "m4ccEnIBTXsGi3mvMt9p",
+		  "_version" : 1,
+		  "_primary_term" : 1,
+		  "_seq_no" : 3,
+		  "anomaly_detector" : {
+			"name" : "test-detector",
+			"description" : "Test detector",
+			"time_field" : "timestamp",
+			"indices" : [
+			  "order*"
+			],
+			"filter_query" : {"bool" : {"filter" : [{"exists" : {"field" : "value","boost" : 1.0}}],"adjust_pure_negative" : true,"boost" : 1.0}},
+			"detection_interval" : {
+			  "period" : {
+				"interval" : 1,
+				"unit" : "Minutes"
+			  }
+			},
+			"window_delay" : {
+			  "period" : {
+				"interval" : 1,
+				"unit" : "Minutes"
+			  }
+			},
+			"schema_version" : 0,
+			"feature_attributes" : [
+			  {
+				"feature_id" : "mYccEnIBTXsGi3mvMd8_",
+				"feature_name" : "total_order",
+				"feature_enabled" : true,
+				"aggregation_query" : {"total_order":{"sum":{"field":"value"}}}
+			  }
+			],
+			"last_update_time" : 1589441737319
+		  }
+		}
+		`
+		_ = json.Unmarshal([]byte(responseJSON), &actual)
+		assert.EqualValues(t, expected, actual)
+	})
+}