mirror of
https://github.com/grafana/grafana.git
synced 2024-11-29 12:14:08 -06:00
96127dce62
This commit fixes a bug in the screenshot service where [alerting].concurrent_render_limit was used instead of [rendering].concurrent_render_request_limit, as in the docs.
93 lines
2.9 KiB
Go
93 lines
2.9 KiB
Go
package screenshot
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"testing"
|
|
|
|
"github.com/golang/mock/gomock"
|
|
"github.com/prometheus/client_golang/prometheus"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/mock"
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/grafana/grafana/pkg/services/dashboards"
|
|
"github.com/grafana/grafana/pkg/services/org"
|
|
"github.com/grafana/grafana/pkg/services/rendering"
|
|
"github.com/grafana/grafana/pkg/setting"
|
|
)
|
|
|
|
func TestHeadlessScreenshotService(t *testing.T) {
|
|
c := gomock.NewController(t)
|
|
defer c.Finish()
|
|
|
|
d := dashboards.FakeDashboardService{}
|
|
r := rendering.NewMockService(c)
|
|
cfg := setting.NewCfg()
|
|
s := NewHeadlessScreenshotService(cfg, &d, r, prometheus.NewRegistry())
|
|
|
|
// a non-existent dashboard should return error
|
|
d.On("GetDashboard", mock.Anything, mock.AnythingOfType("*dashboards.GetDashboardQuery")).Return(nil, dashboards.ErrDashboardNotFound).Once()
|
|
ctx := context.Background()
|
|
opts := ScreenshotOptions{}
|
|
screenshot, err := s.Take(ctx, opts)
|
|
assert.EqualError(t, err, "Dashboard not found")
|
|
assert.Nil(t, screenshot)
|
|
|
|
// should take a screenshot
|
|
qResult := &dashboards.Dashboard{ID: 1, UID: "foo", Slug: "bar", OrgID: 2}
|
|
d.On("GetDashboard", mock.Anything, mock.AnythingOfType("*dashboards.GetDashboardQuery")).Return(qResult, nil)
|
|
|
|
renderOpts := rendering.Opts{
|
|
AuthOpts: rendering.AuthOpts{
|
|
OrgID: 2,
|
|
OrgRole: org.RoleAdmin,
|
|
},
|
|
ErrorOpts: rendering.ErrorOpts{
|
|
ErrorConcurrentLimitReached: true,
|
|
ErrorRenderUnavailable: true,
|
|
},
|
|
TimeoutOpts: rendering.TimeoutOpts{
|
|
Timeout: DefaultTimeout,
|
|
},
|
|
Width: DefaultWidth,
|
|
Height: DefaultHeight,
|
|
Theme: DefaultTheme,
|
|
Path: "d-solo/foo/bar?from=now-6h&orgId=2&panelId=4&to=now-2h",
|
|
ConcurrentLimit: cfg.RendererConcurrentRequestLimit,
|
|
}
|
|
|
|
opts.From = "now-6h"
|
|
opts.To = "now-2h"
|
|
opts.DashboardUID = "foo"
|
|
opts.PanelID = 4
|
|
r.EXPECT().
|
|
Render(ctx, rendering.RenderPNG, renderOpts, nil).
|
|
Return(&rendering.RenderResult{FilePath: "panel.png"}, nil)
|
|
screenshot, err = s.Take(ctx, opts)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, Screenshot{Path: "panel.png"}, *screenshot)
|
|
|
|
// a timeout should return error
|
|
r.EXPECT().
|
|
Render(ctx, rendering.RenderPNG, renderOpts, nil).
|
|
Return(nil, rendering.ErrTimeout)
|
|
screenshot, err = s.Take(ctx, opts)
|
|
assert.EqualError(t, err, fmt.Sprintf("failed to take screenshot: %s", rendering.ErrTimeout))
|
|
assert.Nil(t, screenshot)
|
|
}
|
|
|
|
func TestNoOpScreenshotService(t *testing.T) {
|
|
s := NoOpScreenshotService{}
|
|
screenshot, err := s.Take(context.Background(), ScreenshotOptions{})
|
|
assert.NoError(t, err)
|
|
assert.NotNil(t, screenshot)
|
|
}
|
|
|
|
func TestScreenshotUnavailableService(t *testing.T) {
|
|
s := ScreenshotUnavailableService{}
|
|
screenshot, err := s.Take(context.Background(), ScreenshotOptions{})
|
|
assert.Equal(t, err, ErrScreenshotsUnavailable)
|
|
assert.Nil(t, screenshot)
|
|
}
|