mirror of
https://github.com/grafana/grafana.git
synced 2025-02-25 18:55:37 -06:00
dd77ff6bcd
* Set up frontend linting for Azure - Fix final frontend import - Fix other lint issues * Add Azure Monitor to backend linting * Remove featuremgmt dependency * Add intervalv2 to list of disallowed imports * Remove config dependency - Replace with function from azure-sdk * Remove util dependency * Duplicate interval functionality from core * Add required backend wrappers * Update frontend * Add testing helper * Add missing package * Bump minimum grafana dependency * Fix dependency * Regen cue * Fix lint * Update expected response file * Update import and dependency
111 lines
2.8 KiB
Go
111 lines
2.8 KiB
Go
package azuremonitor
|
|
|
|
import (
|
|
"context"
|
|
"crypto/tls"
|
|
"encoding/json"
|
|
"net/http"
|
|
"testing"
|
|
|
|
"github.com/grafana/grafana-azure-sdk-go/azcredentials"
|
|
"github.com/grafana/grafana-azure-sdk-go/azsettings"
|
|
"github.com/grafana/grafana-plugin-sdk-go/backend"
|
|
"github.com/grafana/grafana-plugin-sdk-go/backend/httpclient"
|
|
|
|
"github.com/grafana/grafana/pkg/tsdb/azuremonitor/types"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestHttpClient_AzureCredentials(t *testing.T) {
|
|
model := types.DatasourceInfo{
|
|
Credentials: &azcredentials.AzureManagedIdentityCredentials{},
|
|
}
|
|
|
|
jsonData, _ := json.Marshal(map[string]any{
|
|
"httpHeaderName1": "GrafanaHeader",
|
|
})
|
|
settings := &backend.DataSourceInstanceSettings{
|
|
JSONData: jsonData,
|
|
DecryptedSecureJSONData: map[string]string{
|
|
"httpHeaderValue1": "GrafanaValue",
|
|
},
|
|
}
|
|
|
|
azureSettings := &azsettings.AzureSettings{
|
|
Cloud: azsettings.AzurePublic,
|
|
}
|
|
provider := &fakeHttpClientProvider{}
|
|
|
|
t.Run("should have Azure middleware when scopes provided", func(t *testing.T) {
|
|
route := types.AzRoute{
|
|
Scopes: []string{"https://management.azure.com/.default"},
|
|
}
|
|
|
|
_, err := newHTTPClient(context.Background(), route, model, settings, azureSettings, provider)
|
|
require.NoError(t, err)
|
|
|
|
require.NotNil(t, provider.opts)
|
|
require.NotNil(t, provider.opts.Middlewares)
|
|
assert.Len(t, provider.opts.Middlewares, 1)
|
|
})
|
|
|
|
t.Run("should not have Azure middleware when scopes are not provided", func(t *testing.T) {
|
|
route := types.AzRoute{
|
|
Scopes: []string{},
|
|
}
|
|
|
|
_, err := newHTTPClient(context.Background(), route, model, settings, azureSettings, provider)
|
|
require.NoError(t, err)
|
|
|
|
assert.NotNil(t, provider.opts)
|
|
|
|
if provider.opts.Middlewares != nil {
|
|
assert.Len(t, provider.opts.Middlewares, 0)
|
|
}
|
|
})
|
|
|
|
t.Run("should combine custom azure and custom grafana headers", func(t *testing.T) {
|
|
route := types.AzRoute{
|
|
Headers: map[string]string{
|
|
"AzureHeader": "AzureValue",
|
|
},
|
|
}
|
|
|
|
res := map[string]string{
|
|
"GrafanaHeader": "GrafanaValue",
|
|
"AzureHeader": "AzureValue",
|
|
}
|
|
_, err := newHTTPClient(context.Background(), route, model, settings, azureSettings, provider)
|
|
require.NoError(t, err)
|
|
|
|
assert.NotNil(t, provider.opts)
|
|
|
|
if provider.opts.Headers != nil {
|
|
assert.Len(t, provider.opts.Headers, 2)
|
|
assert.Equal(t, res, provider.opts.Headers)
|
|
}
|
|
})
|
|
}
|
|
|
|
type fakeHttpClientProvider struct {
|
|
httpclient.Provider
|
|
|
|
opts httpclient.Options
|
|
}
|
|
|
|
func (p *fakeHttpClientProvider) New(opts ...httpclient.Options) (*http.Client, error) {
|
|
p.opts = opts[0]
|
|
return nil, nil
|
|
}
|
|
|
|
func (p *fakeHttpClientProvider) GetTransport(opts ...httpclient.Options) (http.RoundTripper, error) {
|
|
p.opts = opts[0]
|
|
return nil, nil
|
|
}
|
|
|
|
func (p *fakeHttpClientProvider) GetTLSConfig(opts ...httpclient.Options) (*tls.Config, error) {
|
|
p.opts = opts[0]
|
|
return nil, nil
|
|
}
|