mirror of https://github.com/grafana/grafana
Plugins: Add plugin resource tests (#62014)
* remove plugin context from response * remove integration test indicatorpull/62079/head^2
parent
bbf7c0a147
commit
eb1fed7929
@ -0,0 +1,131 @@ |
||||
package api |
||||
|
||||
import ( |
||||
"context" |
||||
"encoding/json" |
||||
"errors" |
||||
"io" |
||||
"path/filepath" |
||||
"strings" |
||||
"testing" |
||||
|
||||
"github.com/stretchr/testify/require" |
||||
|
||||
"github.com/grafana/grafana-azure-sdk-go/azsettings" |
||||
"github.com/grafana/grafana-plugin-sdk-go/backend" |
||||
|
||||
"github.com/grafana/grafana/pkg/infra/db" |
||||
"github.com/grafana/grafana/pkg/infra/localcache" |
||||
"github.com/grafana/grafana/pkg/plugins" |
||||
"github.com/grafana/grafana/pkg/plugins/backendplugin/coreplugin" |
||||
"github.com/grafana/grafana/pkg/plugins/backendplugin/provider" |
||||
"github.com/grafana/grafana/pkg/plugins/config" |
||||
pluginClient "github.com/grafana/grafana/pkg/plugins/manager/client" |
||||
"github.com/grafana/grafana/pkg/plugins/manager/fakes" |
||||
"github.com/grafana/grafana/pkg/plugins/manager/loader" |
||||
"github.com/grafana/grafana/pkg/plugins/manager/registry" |
||||
"github.com/grafana/grafana/pkg/plugins/manager/signature" |
||||
"github.com/grafana/grafana/pkg/plugins/manager/store" |
||||
"github.com/grafana/grafana/pkg/plugins/plugincontext" |
||||
"github.com/grafana/grafana/pkg/services/accesscontrol" |
||||
datasources "github.com/grafana/grafana/pkg/services/datasources/fakes" |
||||
"github.com/grafana/grafana/pkg/services/featuremgmt" |
||||
"github.com/grafana/grafana/pkg/services/oauthtoken/oauthtokentest" |
||||
pluginSettings "github.com/grafana/grafana/pkg/services/pluginsettings/service" |
||||
"github.com/grafana/grafana/pkg/services/pluginsintegration" |
||||
"github.com/grafana/grafana/pkg/services/quota/quotatest" |
||||
"github.com/grafana/grafana/pkg/services/user" |
||||
"github.com/grafana/grafana/pkg/setting" |
||||
"github.com/grafana/grafana/pkg/tsdb/cloudwatch" |
||||
"github.com/grafana/grafana/pkg/tsdb/testdatasource" |
||||
"github.com/grafana/grafana/pkg/web/webtest" |
||||
) |
||||
|
||||
func TestCallResource(t *testing.T) { |
||||
staticRootPath, err := filepath.Abs("../../public/") |
||||
require.NoError(t, err) |
||||
|
||||
cfg := setting.NewCfg() |
||||
cfg.StaticRootPath = staticRootPath |
||||
cfg.IsFeatureToggleEnabled = func(_ string) bool { |
||||
return false |
||||
} |
||||
cfg.Azure = &azsettings.AzureSettings{} |
||||
|
||||
coreRegistry := coreplugin.ProvideCoreRegistry(nil, &cloudwatch.CloudWatchService{}, nil, nil, nil, nil, |
||||
nil, nil, nil, nil, testdatasource.ProvideService(cfg, featuremgmt.WithFeatures()), nil, nil, nil, nil, nil, nil) |
||||
pCfg := config.ProvideConfig(setting.ProvideProvider(cfg), cfg) |
||||
reg := registry.ProvideService() |
||||
l := loader.ProvideService(pCfg, fakes.NewFakeLicensingService(), signature.NewUnsignedAuthorizer(pCfg), |
||||
reg, provider.ProvideService(coreRegistry), fakes.NewFakeRoleRegistry()) |
||||
ps, err := store.ProvideService(cfg, pCfg, reg, l) |
||||
require.NoError(t, err) |
||||
|
||||
pcp := plugincontext.ProvideService(localcache.ProvideService(), ps, &datasources.FakeCacheService{}, &datasources.FakeDataSourceService{}, pluginSettings.ProvideService(db.InitTestDB(t), nil)) |
||||
|
||||
srv := SetupAPITestServer(t, func(hs *HTTPServer) { |
||||
hs.Cfg = cfg |
||||
hs.PluginContextProvider = pcp |
||||
hs.QuotaService = quotatest.New(false, nil) |
||||
hs.pluginStore = ps |
||||
hs.pluginClient = pluginClient.ProvideService(reg, pCfg) |
||||
}) |
||||
|
||||
t.Run("Test successful response is received for valid request", func(t *testing.T) { |
||||
req := srv.NewPostRequest("/api/plugins/testdata/resources/test", strings.NewReader("{ \"test\": true }")) |
||||
webtest.RequestWithSignedInUser(req, &user.SignedInUser{UserID: 1, OrgID: 1, Permissions: map[int64]map[string][]string{ |
||||
1: accesscontrol.GroupScopesByAction([]accesscontrol.Permission{ |
||||
{Action: plugins.ActionAppAccess, Scope: plugins.ScopeProvider.GetResourceAllScope()}, |
||||
}), |
||||
}}) |
||||
resp, err := srv.SendJSON(req) |
||||
require.NoError(t, err) |
||||
|
||||
b, err := io.ReadAll(resp.Body) |
||||
require.NoError(t, err) |
||||
|
||||
var body = make(map[string]interface{}) |
||||
err = json.Unmarshal(b, &body) |
||||
require.NoError(t, err) |
||||
|
||||
require.Equal(t, "Hello world from test datasource!", body["message"]) |
||||
require.NoError(t, resp.Body.Close()) |
||||
require.Equal(t, 200, resp.StatusCode) |
||||
}) |
||||
|
||||
pc, err := pluginClient.NewDecorator(&fakes.FakePluginClient{ |
||||
CallResourceHandlerFunc: backend.CallResourceHandlerFunc(func(ctx context.Context, |
||||
req *backend.CallResourceRequest, sender backend.CallResourceResponseSender) error { |
||||
return errors.New("something went wrong") |
||||
}), |
||||
}, pluginsintegration.CreateMiddlewares(cfg, &oauthtokentest.Service{})...) |
||||
require.NoError(t, err) |
||||
|
||||
srv = SetupAPITestServer(t, func(hs *HTTPServer) { |
||||
hs.Cfg = cfg |
||||
hs.PluginContextProvider = pcp |
||||
hs.QuotaService = quotatest.New(false, nil) |
||||
hs.pluginStore = ps |
||||
hs.pluginClient = pc |
||||
}) |
||||
|
||||
t.Run("Test error is properly propagated to API response", func(t *testing.T) { |
||||
req := srv.NewGetRequest("/api/plugins/testdata/resources/scenarios") |
||||
webtest.RequestWithSignedInUser(req, &user.SignedInUser{UserID: 1, OrgID: 1, Permissions: map[int64]map[string][]string{ |
||||
1: accesscontrol.GroupScopesByAction([]accesscontrol.Permission{ |
||||
{Action: plugins.ActionAppAccess, Scope: plugins.ScopeProvider.GetResourceAllScope()}, |
||||
}), |
||||
}}) |
||||
resp, err := srv.SendJSON(req) |
||||
require.NoError(t, err) |
||||
|
||||
body := new(strings.Builder) |
||||
_, err = io.Copy(body, resp.Body) |
||||
require.NoError(t, err) |
||||
|
||||
expectedBody := `{ "error": "something went wrong", "message": "Failed to call resource", "traceID": "" }` |
||||
require.JSONEq(t, expectedBody, body.String()) |
||||
require.NoError(t, resp.Body.Close()) |
||||
require.Equal(t, 500, resp.StatusCode) |
||||
}) |
||||
} |
Loading…
Reference in new issue