mirror of
https://github.com/grafana/grafana.git
synced 2024-12-01 21:19:28 -06:00
7376eee6ff
This commit adds support for the From and To parameters that set the time range for the graph. The default is 6 hours, which can sometimes be too large of a time range.
95 lines
2.9 KiB
Go
95 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/models"
|
|
"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)
|
|
s := NewHeadlessScreenshotService(&d, r, prometheus.NewRegistry())
|
|
|
|
// a non-existent dashboard should return error
|
|
d.On("GetDashboard", mock.Anything, mock.AnythingOfType("*models.GetDashboardQuery")).Return(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
|
|
d.On("GetDashboard", mock.Anything, mock.AnythingOfType("*models.GetDashboardQuery")).Run(func(args mock.Arguments) {
|
|
q := args.Get(1).(*models.GetDashboardQuery)
|
|
q.Result = &models.Dashboard{Id: 1, Uid: "foo", Slug: "bar", OrgId: 2}
|
|
}).Return(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: setting.AlertingRenderLimit,
|
|
}
|
|
|
|
opts.From = "now-6h"
|
|
opts.To = "now-2h"
|
|
opts.DashboardUID = "foo"
|
|
opts.PanelID = 4
|
|
r.EXPECT().
|
|
Render(ctx, 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, 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)
|
|
}
|