Azure Monitor: Implement CallResourceHandler in the backend (#35581)

pkg/tsdb/azuremonitor/azuremonitor-resource-handler_test.go

@@ -0,0 +1,122 @@
+package azuremonitor
+
+import (
+	"io/ioutil"
+	"net/http"
+	"net/http/httptest"
+	"testing"
+
+	"github.com/grafana/grafana/pkg/setting"
+	"github.com/stretchr/testify/require"
+)
+
+func Test_parseResourcePath(t *testing.T) {
+	tests := []struct {
+		name           string
+		original       string
+		expectedTarget string
+		Err            require.ErrorAssertionFunc
+	}{
+		{
+			"Path with a subscription",
+			"/azuremonitor/subscriptions/44693801",
+			"/subscriptions/44693801",
+			require.NoError,
+		},
+		{
+			"Malformed path",
+			"/subscriptions?44693801",
+			"",
+			require.Error,
+		},
+	}
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			target, err := getTarget(tt.original)
+			if target != tt.expectedTarget {
+				t.Errorf("Unexpected target %s expecting %s", target, tt.expectedTarget)
+			}
+			tt.Err(t, err)
+		})
+	}
+}
+
+func Test_proxyRequest(t *testing.T) {
+	tests := []struct {
+		name string
+	}{
+		{"forwards headers and body"},
+	}
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+				w.Header().Add("foo", "bar")
+				_, err := w.Write([]byte("result"))
+				if err != nil {
+					t.Fatal(err)
+				}
+			}))
+			req, err := http.NewRequest(http.MethodGet, srv.URL, nil)
+			if err != nil {
+				t.Error(err)
+			}
+			rw := httptest.NewRecorder()
+			proxy := httpServiceProxy{}
+			res := proxy.Do(rw, req, srv.Client())
+			if res.Header().Get("foo") != "bar" {
+				t.Errorf("Unexpected headers: %v", res.Header())
+			}
+			result := rw.Result()
+			body, err := ioutil.ReadAll(result.Body)
+			if err != nil {
+				t.Error(err)
+			}
+			err = result.Body.Close()
+			if err != nil {
+				t.Error(err)
+			}
+			if string(body) != "result" {
+				t.Errorf("Unexpected body: %v", string(body))
+			}
+		})
+	}
+}
+
+type fakeProxy struct {
+	requestedURL string
+}
+
+func (s *fakeProxy) Do(rw http.ResponseWriter, req *http.Request, cli *http.Client) http.ResponseWriter {
+	s.requestedURL = req.URL.String()
+	return nil
+}
+
+func Test_resourceHandler(t *testing.T) {
+	proxy := &fakeProxy{}
+	s := Service{
+		im: &fakeInstance{
+			services: map[string]datasourceService{
+				azureMonitor: {
+					URL:        routes[setting.AzurePublic][azureMonitor].URL,
+					HTTPClient: &http.Client{},
+				},
+			},
+		},
+		Cfg: &setting.Cfg{},
+		executors: map[string]azDatasourceExecutor{
+			azureMonitor: &AzureMonitorDatasource{
+				proxy: proxy,
+			},
+		},
+	}
+	rw := httptest.NewRecorder()
+	req, err := http.NewRequest(http.MethodGet, "http://foo/azuremonitor/subscriptions/44693801", nil)
+	if err != nil {
+		t.Fatalf("Unexpected error %v", err)
+	}
+	s.resourceHandler(azureMonitor)(rw, req)
+	expectedURL := "https://management.azure.com/subscriptions/44693801"
+	if proxy.requestedURL != expectedURL {
+		t.Errorf("Unexpected result URL. Got %s, expecting %s", proxy.requestedURL, expectedURL)
+	}
+}