Chore: Refactor GoConvey in services dashboards package (#40895)
This commit is contained in:
@@ -11,11 +11,12 @@ import (
|
|||||||
"github.com/grafana/grafana/pkg/bus"
|
"github.com/grafana/grafana/pkg/bus"
|
||||||
"github.com/grafana/grafana/pkg/models"
|
"github.com/grafana/grafana/pkg/models"
|
||||||
"github.com/grafana/grafana/pkg/services/guardian"
|
"github.com/grafana/grafana/pkg/services/guardian"
|
||||||
. "github.com/smartystreets/goconvey/convey"
|
|
||||||
|
"github.com/stretchr/testify/require"
|
||||||
)
|
)
|
||||||
|
|
||||||
func TestDashboardService(t *testing.T) {
|
func TestDashboardService(t *testing.T) {
|
||||||
Convey("Dashboard service tests", t, func() {
|
t.Run("Dashboard service tests", func(t *testing.T) {
|
||||||
bus.ClearBusHandlers()
|
bus.ClearBusHandlers()
|
||||||
|
|
||||||
fakeStore := fakeDashboardStore{}
|
fakeStore := fakeDashboardStore{}
|
||||||
@@ -25,35 +26,36 @@ func TestDashboardService(t *testing.T) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
origNewDashboardGuardian := guardian.New
|
origNewDashboardGuardian := guardian.New
|
||||||
|
defer func() { guardian.New = origNewDashboardGuardian }()
|
||||||
guardian.MockDashboardGuardian(&guardian.FakeDashboardGuardian{CanSaveValue: true})
|
guardian.MockDashboardGuardian(&guardian.FakeDashboardGuardian{CanSaveValue: true})
|
||||||
|
|
||||||
Convey("Save dashboard validation", func() {
|
t.Run("Save dashboard validation", func(t *testing.T) {
|
||||||
dto := &SaveDashboardDTO{}
|
dto := &SaveDashboardDTO{}
|
||||||
|
|
||||||
Convey("When saving a dashboard with empty title it should return error", func() {
|
t.Run("When saving a dashboard with empty title it should return error", func(t *testing.T) {
|
||||||
titles := []string{"", " ", " \t "}
|
titles := []string{"", " ", " \t "}
|
||||||
|
|
||||||
for _, title := range titles {
|
for _, title := range titles {
|
||||||
dto.Dashboard = models.NewDashboard(title)
|
dto.Dashboard = models.NewDashboard(title)
|
||||||
_, err := service.SaveDashboard(dto, false)
|
_, err := service.SaveDashboard(dto, false)
|
||||||
So(err, ShouldEqual, models.ErrDashboardTitleEmpty)
|
require.Equal(t, err, models.ErrDashboardTitleEmpty)
|
||||||
}
|
}
|
||||||
})
|
})
|
||||||
|
|
||||||
Convey("Should return validation error if it's a folder and have a folder id", func() {
|
t.Run("Should return validation error if it's a folder and have a folder id", func(t *testing.T) {
|
||||||
dto.Dashboard = models.NewDashboardFolder("Folder")
|
dto.Dashboard = models.NewDashboardFolder("Folder")
|
||||||
dto.Dashboard.FolderId = 1
|
dto.Dashboard.FolderId = 1
|
||||||
_, err := service.SaveDashboard(dto, false)
|
_, err := service.SaveDashboard(dto, false)
|
||||||
So(err, ShouldEqual, models.ErrDashboardFolderCannotHaveParent)
|
require.Equal(t, err, models.ErrDashboardFolderCannotHaveParent)
|
||||||
})
|
})
|
||||||
|
|
||||||
Convey("Should return validation error if folder is named General", func() {
|
t.Run("Should return validation error if folder is named General", func(t *testing.T) {
|
||||||
dto.Dashboard = models.NewDashboardFolder("General")
|
dto.Dashboard = models.NewDashboardFolder("General")
|
||||||
_, err := service.SaveDashboard(dto, false)
|
_, err := service.SaveDashboard(dto, false)
|
||||||
So(err, ShouldEqual, models.ErrDashboardFolderNameExists)
|
require.Equal(t, err, models.ErrDashboardFolderNameExists)
|
||||||
})
|
})
|
||||||
|
|
||||||
Convey("When saving a dashboard should validate uid", func() {
|
t.Run("When saving a dashboard should validate uid", func(t *testing.T) {
|
||||||
origValidateAlerts := validateAlerts
|
origValidateAlerts := validateAlerts
|
||||||
t.Cleanup(func() {
|
t.Cleanup(func() {
|
||||||
validateAlerts = origValidateAlerts
|
validateAlerts = origValidateAlerts
|
||||||
@@ -81,11 +83,11 @@ func TestDashboardService(t *testing.T) {
|
|||||||
dto.User = &models.SignedInUser{}
|
dto.User = &models.SignedInUser{}
|
||||||
|
|
||||||
_, err := service.buildSaveDashboardCommand(dto, true, false)
|
_, err := service.buildSaveDashboardCommand(dto, true, false)
|
||||||
So(err, ShouldEqual, tc.Error)
|
require.Equal(t, err, tc.Error)
|
||||||
}
|
}
|
||||||
})
|
})
|
||||||
|
|
||||||
Convey("Should return validation error if dashboard is provisioned", func() {
|
t.Run("Should return validation error if dashboard is provisioned", func(t *testing.T) {
|
||||||
t.Cleanup(func() {
|
t.Cleanup(func() {
|
||||||
fakeStore.provisionedData = nil
|
fakeStore.provisionedData = nil
|
||||||
})
|
})
|
||||||
@@ -103,10 +105,10 @@ func TestDashboardService(t *testing.T) {
|
|||||||
dto.Dashboard.SetId(3)
|
dto.Dashboard.SetId(3)
|
||||||
dto.User = &models.SignedInUser{UserId: 1}
|
dto.User = &models.SignedInUser{UserId: 1}
|
||||||
_, err := service.SaveDashboard(dto, false)
|
_, err := service.SaveDashboard(dto, false)
|
||||||
So(err, ShouldEqual, models.ErrDashboardCannotSaveProvisionedDashboard)
|
require.Equal(t, err, models.ErrDashboardCannotSaveProvisionedDashboard)
|
||||||
})
|
})
|
||||||
|
|
||||||
Convey("Should not return validation error if dashboard is provisioned but UI updates allowed", func() {
|
t.Run("Should not return validation error if dashboard is provisioned but UI updates allowed", func(t *testing.T) {
|
||||||
origValidateAlerts := validateAlerts
|
origValidateAlerts := validateAlerts
|
||||||
t.Cleanup(func() {
|
t.Cleanup(func() {
|
||||||
validateAlerts = origValidateAlerts
|
validateAlerts = origValidateAlerts
|
||||||
@@ -119,10 +121,10 @@ func TestDashboardService(t *testing.T) {
|
|||||||
dto.Dashboard.SetId(3)
|
dto.Dashboard.SetId(3)
|
||||||
dto.User = &models.SignedInUser{UserId: 1}
|
dto.User = &models.SignedInUser{UserId: 1}
|
||||||
_, err := service.SaveDashboard(dto, true)
|
_, err := service.SaveDashboard(dto, true)
|
||||||
So(err, ShouldBeNil)
|
require.NoError(t, err)
|
||||||
})
|
})
|
||||||
|
|
||||||
Convey("Should return validation error if alert data is invalid", func() {
|
t.Run("Should return validation error if alert data is invalid", func(t *testing.T) {
|
||||||
origValidateAlerts := validateAlerts
|
origValidateAlerts := validateAlerts
|
||||||
t.Cleanup(func() {
|
t.Cleanup(func() {
|
||||||
validateAlerts = origValidateAlerts
|
validateAlerts = origValidateAlerts
|
||||||
@@ -133,14 +135,14 @@ func TestDashboardService(t *testing.T) {
|
|||||||
|
|
||||||
dto.Dashboard = models.NewDashboard("Dash")
|
dto.Dashboard = models.NewDashboard("Dash")
|
||||||
_, err := service.SaveDashboard(dto, false)
|
_, err := service.SaveDashboard(dto, false)
|
||||||
So(err.Error(), ShouldEqual, "alert validation error")
|
require.Equal(t, err.Error(), "alert validation error")
|
||||||
})
|
})
|
||||||
})
|
})
|
||||||
|
|
||||||
Convey("Save provisioned dashboard validation", func() {
|
t.Run("Save provisioned dashboard validation", func(t *testing.T) {
|
||||||
dto := &SaveDashboardDTO{}
|
dto := &SaveDashboardDTO{}
|
||||||
|
|
||||||
Convey("Should not return validation error if dashboard is provisioned", func() {
|
t.Run("Should not return validation error if dashboard is provisioned", func(t *testing.T) {
|
||||||
origUpdateAlerting := UpdateAlerting
|
origUpdateAlerting := UpdateAlerting
|
||||||
t.Cleanup(func() {
|
t.Cleanup(func() {
|
||||||
UpdateAlerting = origUpdateAlerting
|
UpdateAlerting = origUpdateAlerting
|
||||||
@@ -162,10 +164,10 @@ func TestDashboardService(t *testing.T) {
|
|||||||
dto.Dashboard.SetId(3)
|
dto.Dashboard.SetId(3)
|
||||||
dto.User = &models.SignedInUser{UserId: 1}
|
dto.User = &models.SignedInUser{UserId: 1}
|
||||||
_, err := service.SaveProvisionedDashboard(dto, nil)
|
_, err := service.SaveProvisionedDashboard(dto, nil)
|
||||||
So(err, ShouldBeNil)
|
require.NoError(t, err)
|
||||||
})
|
})
|
||||||
|
|
||||||
Convey("Should override invalid refresh interval if dashboard is provisioned", func() {
|
t.Run("Should override invalid refresh interval if dashboard is provisioned", func(t *testing.T) {
|
||||||
oldRefreshInterval := setting.MinRefreshInterval
|
oldRefreshInterval := setting.MinRefreshInterval
|
||||||
setting.MinRefreshInterval = "5m"
|
setting.MinRefreshInterval = "5m"
|
||||||
defer func() { setting.MinRefreshInterval = oldRefreshInterval }()
|
defer func() { setting.MinRefreshInterval = oldRefreshInterval }()
|
||||||
@@ -192,15 +194,15 @@ func TestDashboardService(t *testing.T) {
|
|||||||
dto.User = &models.SignedInUser{UserId: 1}
|
dto.User = &models.SignedInUser{UserId: 1}
|
||||||
dto.Dashboard.Data.Set("refresh", "1s")
|
dto.Dashboard.Data.Set("refresh", "1s")
|
||||||
_, err := service.SaveProvisionedDashboard(dto, nil)
|
_, err := service.SaveProvisionedDashboard(dto, nil)
|
||||||
So(err, ShouldBeNil)
|
require.NoError(t, err)
|
||||||
So(dto.Dashboard.Data.Get("refresh").MustString(), ShouldEqual, "5m")
|
require.Equal(t, dto.Dashboard.Data.Get("refresh").MustString(), "5m")
|
||||||
})
|
})
|
||||||
})
|
})
|
||||||
|
|
||||||
Convey("Import dashboard validation", func() {
|
t.Run("Import dashboard validation", func(t *testing.T) {
|
||||||
dto := &SaveDashboardDTO{}
|
dto := &SaveDashboardDTO{}
|
||||||
|
|
||||||
Convey("Should return validation error if dashboard is provisioned", func() {
|
t.Run("Should return validation error if dashboard is provisioned", func(t *testing.T) {
|
||||||
t.Cleanup(func() {
|
t.Cleanup(func() {
|
||||||
fakeStore.provisionedData = nil
|
fakeStore.provisionedData = nil
|
||||||
})
|
})
|
||||||
@@ -227,45 +229,41 @@ func TestDashboardService(t *testing.T) {
|
|||||||
dto.Dashboard.SetId(3)
|
dto.Dashboard.SetId(3)
|
||||||
dto.User = &models.SignedInUser{UserId: 1}
|
dto.User = &models.SignedInUser{UserId: 1}
|
||||||
_, err := service.ImportDashboard(dto)
|
_, err := service.ImportDashboard(dto)
|
||||||
So(err, ShouldEqual, models.ErrDashboardCannotSaveProvisionedDashboard)
|
require.Equal(t, err, models.ErrDashboardCannotSaveProvisionedDashboard)
|
||||||
})
|
})
|
||||||
})
|
})
|
||||||
|
|
||||||
Convey("Given provisioned dashboard", func() {
|
t.Run("Given provisioned dashboard", func(t *testing.T) {
|
||||||
result := setupDeleteHandlers(t, &fakeStore, true)
|
t.Run("DeleteProvisionedDashboard should delete it", func(t *testing.T) {
|
||||||
|
result := setupDeleteHandlers(t, &fakeStore, true)
|
||||||
Convey("DeleteProvisionedDashboard should delete it", func() {
|
|
||||||
err := service.DeleteProvisionedDashboard(1, 1)
|
err := service.DeleteProvisionedDashboard(1, 1)
|
||||||
So(err, ShouldBeNil)
|
require.NoError(t, err)
|
||||||
So(result.deleteWasCalled, ShouldBeTrue)
|
require.True(t, result.deleteWasCalled)
|
||||||
})
|
})
|
||||||
|
|
||||||
Convey("DeleteDashboard should fail to delete it", func() {
|
t.Run("DeleteDashboard should fail to delete it", func(t *testing.T) {
|
||||||
|
result := setupDeleteHandlers(t, &fakeStore, true)
|
||||||
err := service.DeleteDashboard(1, 1)
|
err := service.DeleteDashboard(1, 1)
|
||||||
So(err, ShouldEqual, models.ErrDashboardCannotDeleteProvisionedDashboard)
|
require.Equal(t, err, models.ErrDashboardCannotDeleteProvisionedDashboard)
|
||||||
So(result.deleteWasCalled, ShouldBeFalse)
|
require.False(t, result.deleteWasCalled)
|
||||||
})
|
})
|
||||||
})
|
})
|
||||||
|
|
||||||
Convey("Given non provisioned dashboard", func() {
|
t.Run("Given non provisioned dashboard", func(t *testing.T) {
|
||||||
result := setupDeleteHandlers(t, &fakeStore, false)
|
result := setupDeleteHandlers(t, &fakeStore, false)
|
||||||
|
|
||||||
Convey("DeleteProvisionedDashboard should delete it", func() {
|
t.Run("DeleteProvisionedDashboard should delete it", func(t *testing.T) {
|
||||||
err := service.DeleteProvisionedDashboard(1, 1)
|
err := service.DeleteProvisionedDashboard(1, 1)
|
||||||
So(err, ShouldBeNil)
|
require.NoError(t, err)
|
||||||
So(result.deleteWasCalled, ShouldBeTrue)
|
require.True(t, result.deleteWasCalled)
|
||||||
})
|
})
|
||||||
|
|
||||||
Convey("DeleteDashboard should delete it", func() {
|
t.Run("DeleteDashboard should delete it", func(t *testing.T) {
|
||||||
err := service.DeleteDashboard(1, 1)
|
err := service.DeleteDashboard(1, 1)
|
||||||
So(err, ShouldBeNil)
|
require.NoError(t, err)
|
||||||
So(result.deleteWasCalled, ShouldBeTrue)
|
require.True(t, result.deleteWasCalled)
|
||||||
})
|
})
|
||||||
})
|
})
|
||||||
|
|
||||||
Reset(func() {
|
|
||||||
guardian.New = origNewDashboardGuardian
|
|
||||||
})
|
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -285,8 +283,8 @@ func setupDeleteHandlers(t *testing.T, fakeStore *fakeDashboardStore, provisione
|
|||||||
|
|
||||||
result := &Result{}
|
result := &Result{}
|
||||||
bus.AddHandler("test", func(cmd *models.DeleteDashboardCommand) error {
|
bus.AddHandler("test", func(cmd *models.DeleteDashboardCommand) error {
|
||||||
So(cmd.Id, ShouldEqual, 1)
|
require.Equal(t, cmd.Id, int64(1))
|
||||||
So(cmd.OrgId, ShouldEqual, 1)
|
require.Equal(t, cmd.OrgId, int64(1))
|
||||||
result.deleteWasCalled = true
|
result.deleteWasCalled = true
|
||||||
return nil
|
return nil
|
||||||
})
|
})
|
||||||
|
|||||||
Reference in New Issue
Block a user