Andres Martinez Gotor
115 lines
3.8 KiB
Go
115 lines
3.8 KiB
Go
package datasourcecheck
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/grafana/grafana-plugin-sdk-go/backend"
|
|
advisor "github.com/grafana/grafana/apps/advisor/pkg/apis/advisor/v0alpha1"
|
|
"github.com/grafana/grafana/pkg/plugins"
|
|
"github.com/grafana/grafana/pkg/registry/apis/datasource"
|
|
"github.com/grafana/grafana/pkg/services/datasources"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestCheck_Run(t *testing.T) {
|
|
t.Run("should return no errors when all datasources are healthy", func(t *testing.T) {
|
|
datasources := []*datasources.DataSource{
|
|
{UID: "valid-uid-1", Type: "prometheus", Name: "Prometheus"},
|
|
{UID: "valid-uid-2", Type: "mysql", Name: "MySQL"},
|
|
}
|
|
|
|
mockDatasourceSvc := &MockDatasourceSvc{dss: datasources}
|
|
mockPluginContextProvider := &MockPluginContextProvider{pCtx: backend.PluginContext{}}
|
|
mockPluginClient := &MockPluginClient{res: &backend.CheckHealthResult{Status: backend.HealthStatusOk}}
|
|
|
|
check := &check{
|
|
DatasourceSvc: mockDatasourceSvc,
|
|
PluginContextProvider: mockPluginContextProvider,
|
|
PluginClient: mockPluginClient,
|
|
}
|
|
|
|
report, err := check.Run(context.Background(), &advisor.CheckSpec{})
|
|
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, int64(2), report.Count)
|
|
assert.Empty(t, report.Errors)
|
|
})
|
|
|
|
t.Run("should return errors when datasource UID is invalid", func(t *testing.T) {
|
|
datasources := []*datasources.DataSource{
|
|
{UID: "invalid uid", Type: "prometheus", Name: "Prometheus"},
|
|
}
|
|
|
|
mockDatasourceSvc := &MockDatasourceSvc{dss: datasources}
|
|
mockPluginContextProvider := &MockPluginContextProvider{pCtx: backend.PluginContext{}}
|
|
mockPluginClient := &MockPluginClient{res: &backend.CheckHealthResult{Status: backend.HealthStatusOk}}
|
|
|
|
check := &check{
|
|
DatasourceSvc: mockDatasourceSvc,
|
|
PluginContextProvider: mockPluginContextProvider,
|
|
PluginClient: mockPluginClient,
|
|
}
|
|
|
|
report, err := check.Run(context.Background(), &advisor.CheckSpec{})
|
|
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, int64(1), report.Count)
|
|
assert.Len(t, report.Errors, 1)
|
|
assert.Equal(t, "Invalid UID 'invalid uid' for data source Prometheus", report.Errors[0].Reason)
|
|
})
|
|
|
|
t.Run("should return errors when datasource health check fails", func(t *testing.T) {
|
|
datasources := []*datasources.DataSource{
|
|
{UID: "valid-uid-1", Type: "prometheus", Name: "Prometheus"},
|
|
}
|
|
|
|
mockDatasourceSvc := &MockDatasourceSvc{dss: datasources}
|
|
mockPluginContextProvider := &MockPluginContextProvider{pCtx: backend.PluginContext{}}
|
|
mockPluginClient := &MockPluginClient{res: &backend.CheckHealthResult{Status: backend.HealthStatusError}}
|
|
|
|
check := &check{
|
|
DatasourceSvc: mockDatasourceSvc,
|
|
PluginContextProvider: mockPluginContextProvider,
|
|
PluginClient: mockPluginClient,
|
|
}
|
|
|
|
report, err := check.Run(context.Background(), &advisor.CheckSpec{})
|
|
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, int64(1), report.Count)
|
|
assert.Len(t, report.Errors, 1)
|
|
assert.Equal(t, "Health check failed for Prometheus", report.Errors[0].Reason)
|
|
})
|
|
}
|
|
|
|
type MockDatasourceSvc struct {
|
|
datasources.DataSourceService
|
|
|
|
dss []*datasources.DataSource
|
|
}
|
|
|
|
func (m *MockDatasourceSvc) GetAllDataSources(ctx context.Context, query *datasources.GetAllDataSourcesQuery) ([]*datasources.DataSource, error) {
|
|
return m.dss, nil
|
|
}
|
|
|
|
type MockPluginContextProvider struct {
|
|
datasource.PluginContextWrapper
|
|
|
|
pCtx backend.PluginContext
|
|
}
|
|
|
|
func (m *MockPluginContextProvider) PluginContextForDataSource(ctx context.Context, datasourceSettings *backend.DataSourceInstanceSettings) (backend.PluginContext, error) {
|
|
return m.pCtx, nil
|
|
}
|
|
|
|
type MockPluginClient struct {
|
|
plugins.Client
|
|
|
|
res *backend.CheckHealthResult
|
|
}
|
|
|
|
func (m *MockPluginClient) CheckHealth(ctx context.Context, req *backend.CheckHealthRequest) (*backend.CheckHealthResult, error) {
|
|
return m.res, nil
|
|
}
|