From 763dfebfb30e4250e99d8001fadaaf79d6a6dd75 Mon Sep 17 00:00:00 2001
From: "marwen.azouzi" <marwen.azouzi@datadoghq.com>
Date: Mon, 2 Mar 2020 13:46:45 +0100
Subject: [PATCH] [k9] Add test for new request type creation

---
 integration/dashboards_test.go | 62 ++++++++++++++++++++++++++++++++++
 1 file changed, 62 insertions(+)

diff --git a/integration/dashboards_test.go b/integration/dashboards_test.go
index 45d331d..af1e468 100644
--- a/integration/dashboards_test.go
+++ b/integration/dashboards_test.go
@@ -334,6 +334,68 @@ func createGraphWithLogQuery() []datadog.Graph {
 	return graphs
 }
 
+func createGraphWithRumQuery() []datadog.Graph {
+	gd := &datadog.GraphDefinition{}
+	gd.SetViz("timeseries")
+
+	r := gd.Requests
+	gd.Requests = append(r, datadog.GraphDefinitionRequest{
+		Type: datadog.String("bars"),
+		RumQuery: &datadog.GraphApmOrLogQuery{
+			Index: datadog.String("rum"),
+			Compute: &datadog.GraphApmOrLogQueryCompute{
+				Aggregation: datadog.String("count"),
+			},
+			Search:  &datadog.GraphApmOrLogQuerySearch{
+				Query: datadog.String("status:info"),
+			},
+			GroupBy: []datadog.GraphApmOrLogQueryGroupBy{{
+				Facet: datadog.String("service"),
+			}},
+		},
+	})
+
+	graph := datadog.Graph{
+		Title:      datadog.String("Mandatory graph 4"),
+		Definition: gd,
+	}
+
+	graphs := []datadog.Graph{}
+	graphs = append(graphs, graph)
+	return graphs
+}
+
+func createGraphWithSecurityQuery() []datadog.Graph {
+	gd := &datadog.GraphDefinition{}
+	gd.SetViz("timeseries")
+
+	r := gd.Requests
+	gd.Requests = append(r, datadog.GraphDefinitionRequest{
+		Type: datadog.String("bars"),
+		SecurityQuery: &datadog.GraphApmOrLogQuery{
+			Index: datadog.String("signal"),
+			Compute: &datadog.GraphApmOrLogQueryCompute{
+				Aggregation: datadog.String("count"),
+			},
+			Search:  &datadog.GraphApmOrLogQuerySearch{
+				Query: datadog.String("status:(high OR critical)"),
+			},
+			GroupBy: []datadog.GraphApmOrLogQueryGroupBy{{
+				Facet: datadog.String("host"),
+			}},
+		},
+	})
+
+	graph := datadog.Graph{
+		Title:      datadog.String("Mandatory graph 5"),
+		Definition: gd,
+	}
+
+	graphs := []datadog.Graph{}
+	graphs = append(graphs, graph)
+	return graphs
+}
+
 func createGraphWithProcessQuery() []datadog.Graph {
 	gd := &datadog.GraphDefinition{}
 	gd.SetViz("timeseries")