Chore: Add CloudWatch HTTP API tests (#29691)

* CloudWatch: Add HTTP API tests

Signed-off-by: Arve Knudsen <arve.knudsen@gmail.com>

pkg/tsdb/models.go

@@ -1,7 +1,9 @@
 package tsdb
 
 import (
+	"encoding/base64"
 	"encoding/json"
+	"fmt"
 
 	"github.com/grafana/grafana-plugin-sdk-go/data"
 	"github.com/grafana/grafana/pkg/components/null"
@@ -42,6 +44,121 @@ type QueryResult struct {
 	Dataframes  DataFrames       `json:"dataframes"`
 }
 
+// UnmarshalJSON deserializes a QueryResult from JSON.
+//
+// Deserialization support is required by tests.
+func (r *QueryResult) UnmarshalJSON(b []byte) error {
+	m := map[string]interface{}{}
+	// TODO: Use JSON decoder
+	if err := json.Unmarshal(b, &m); err != nil {
+		return err
+	}
+
+	refID, ok := m["refId"].(string)
+	if !ok {
+		return fmt.Errorf("can't decode field refId - not a string")
+	}
+	var meta *simplejson.Json
+	if m["meta"] != nil {
+		mm, ok := m["meta"].(map[string]interface{})
+		if !ok {
+			return fmt.Errorf("can't decode field meta - not a JSON object")
+		}
+		meta = simplejson.NewFromAny(mm)
+	}
+	var series TimeSeriesSlice
+	/* TODO
+	if m["series"] != nil {
+	}
+	*/
+	var tables []*Table
+	if m["tables"] != nil {
+		ts, ok := m["tables"].([]interface{})
+		if !ok {
+			return fmt.Errorf("can't decode field tables - not an array of Tables")
+		}
+		for _, ti := range ts {
+			tm, ok := ti.(map[string]interface{})
+			if !ok {
+				return fmt.Errorf("can't decode field tables - not an array of Tables")
+			}
+			var columns []TableColumn
+			cs, ok := tm["columns"].([]interface{})
+			if !ok {
+				return fmt.Errorf("can't decode field tables - not an array of Tables")
+			}
+			for _, ci := range cs {
+				cm, ok := ci.(map[string]interface{})
+				if !ok {
+					return fmt.Errorf("can't decode field tables - not an array of Tables")
+				}
+				val, ok := cm["text"].(string)
+				if !ok {
+					return fmt.Errorf("can't decode field tables - not an array of Tables")
+				}
+
+				columns = append(columns, TableColumn{Text: val})
+			}
+
+			rs, ok := tm["rows"].([]interface{})
+			if !ok {
+				return fmt.Errorf("can't decode field tables - not an array of Tables")
+			}
+			var rows []RowValues
+			for _, ri := range rs {
+				vals, ok := ri.([]interface{})
+				if !ok {
+					return fmt.Errorf("can't decode field tables - not an array of Tables")
+				}
+				rows = append(rows, vals)
+			}
+
+			tables = append(tables, &Table{
+				Columns: columns,
+				Rows:    rows,
+			})
+		}
+	}
+
+	var dfs *dataFrames
+	if m["dataframes"] != nil {
+		raw, ok := m["dataframes"].([]interface{})
+		if !ok {
+			return fmt.Errorf("can't decode field dataframes - not an array of byte arrays")
+		}
+
+		var encoded [][]byte
+		for _, ra := range raw {
+			encS, ok := ra.(string)
+			if !ok {
+				return fmt.Errorf("can't decode field dataframes - not an array of byte arrays")
+			}
+			enc, err := base64.StdEncoding.DecodeString(encS)
+			if err != nil {
+				return fmt.Errorf("can't decode field dataframes - not an array of arrow frames")
+			}
+			encoded = append(encoded, enc)
+		}
+		decoded, err := data.UnmarshalArrowFrames(encoded)
+		if err != nil {
+			return err
+		}
+		dfs = &dataFrames{
+			decoded: decoded,
+			encoded: encoded,
+		}
+	}
+
+	r.RefId = refID
+	r.Meta = meta
+	r.Series = series
+	r.Tables = tables
+	if dfs != nil {
+		r.Dataframes = dfs
+	}
+	return nil
+}
+
 type TimeSeries struct {
 	Name   string            `json:"name"`
 	Points TimeSeriesPoints  `json:"points"`