|
@@ -3,10 +3,11 @@ package sqlstore
|
|
|
import (
|
|
import (
|
|
|
"testing"
|
|
"testing"
|
|
|
|
|
|
|
|
|
|
+ "time"
|
|
|
|
|
+
|
|
|
"github.com/grafana/grafana/pkg/components/simplejson"
|
|
"github.com/grafana/grafana/pkg/components/simplejson"
|
|
|
m "github.com/grafana/grafana/pkg/models"
|
|
m "github.com/grafana/grafana/pkg/models"
|
|
|
. "github.com/smartystreets/goconvey/convey"
|
|
. "github.com/smartystreets/goconvey/convey"
|
|
|
- "time"
|
|
|
|
|
)
|
|
)
|
|
|
|
|
|
|
|
func mockTimeNow() {
|
|
func mockTimeNow() {
|
|
@@ -99,7 +100,7 @@ func TestAlertingDataAccess(t *testing.T) {
|
|
|
})
|
|
})
|
|
|
|
|
|
|
|
Convey("Can read properties", func() {
|
|
Convey("Can read properties", func() {
|
|
|
- alertQuery := m.GetAlertsQuery{DashboardId: testDash.Id, PanelId: 1, OrgId: 1, User: &m.SignedInUser{OrgRole: m.ROLE_ADMIN}}
|
|
|
|
|
|
|
+ alertQuery := m.GetAlertsQuery{DashboardIDs: []int64{testDash.Id}, PanelId: 1, OrgId: 1, User: &m.SignedInUser{OrgRole: m.ROLE_ADMIN}}
|
|
|
err2 := HandleAlertsQuery(&alertQuery)
|
|
err2 := HandleAlertsQuery(&alertQuery)
|
|
|
|
|
|
|
|
alert := alertQuery.Result[0]
|
|
alert := alertQuery.Result[0]
|
|
@@ -109,7 +110,7 @@ func TestAlertingDataAccess(t *testing.T) {
|
|
|
})
|
|
})
|
|
|
|
|
|
|
|
Convey("Viewer cannot read alerts", func() {
|
|
Convey("Viewer cannot read alerts", func() {
|
|
|
- alertQuery := m.GetAlertsQuery{DashboardId: testDash.Id, PanelId: 1, OrgId: 1, User: &m.SignedInUser{OrgRole: m.ROLE_VIEWER}}
|
|
|
|
|
|
|
+ alertQuery := m.GetAlertsQuery{DashboardIDs: []int64{testDash.Id}, PanelId: 1, OrgId: 1, User: &m.SignedInUser{OrgRole: m.ROLE_VIEWER}}
|
|
|
err2 := HandleAlertsQuery(&alertQuery)
|
|
err2 := HandleAlertsQuery(&alertQuery)
|
|
|
|
|
|
|
|
So(err2, ShouldBeNil)
|
|
So(err2, ShouldBeNil)
|
|
@@ -134,7 +135,7 @@ func TestAlertingDataAccess(t *testing.T) {
|
|
|
})
|
|
})
|
|
|
|
|
|
|
|
Convey("Alerts should be updated", func() {
|
|
Convey("Alerts should be updated", func() {
|
|
|
- query := m.GetAlertsQuery{DashboardId: testDash.Id, OrgId: 1, User: &m.SignedInUser{OrgRole: m.ROLE_ADMIN}}
|
|
|
|
|
|
|
+ query := m.GetAlertsQuery{DashboardIDs: []int64{testDash.Id}, OrgId: 1, User: &m.SignedInUser{OrgRole: m.ROLE_ADMIN}}
|
|
|
err2 := HandleAlertsQuery(&query)
|
|
err2 := HandleAlertsQuery(&query)
|
|
|
|
|
|
|
|
So(err2, ShouldBeNil)
|
|
So(err2, ShouldBeNil)
|
|
@@ -183,7 +184,7 @@ func TestAlertingDataAccess(t *testing.T) {
|
|
|
Convey("Should save 3 dashboards", func() {
|
|
Convey("Should save 3 dashboards", func() {
|
|
|
So(err, ShouldBeNil)
|
|
So(err, ShouldBeNil)
|
|
|
|
|
|
|
|
- queryForDashboard := m.GetAlertsQuery{DashboardId: testDash.Id, OrgId: 1, User: &m.SignedInUser{OrgRole: m.ROLE_ADMIN}}
|
|
|
|
|
|
|
+ queryForDashboard := m.GetAlertsQuery{DashboardIDs: []int64{testDash.Id}, OrgId: 1, User: &m.SignedInUser{OrgRole: m.ROLE_ADMIN}}
|
|
|
err2 := HandleAlertsQuery(&queryForDashboard)
|
|
err2 := HandleAlertsQuery(&queryForDashboard)
|
|
|
|
|
|
|
|
So(err2, ShouldBeNil)
|
|
So(err2, ShouldBeNil)
|
|
@@ -197,7 +198,7 @@ func TestAlertingDataAccess(t *testing.T) {
|
|
|
err = SaveAlerts(&cmd)
|
|
err = SaveAlerts(&cmd)
|
|
|
|
|
|
|
|
Convey("should delete the missing alert", func() {
|
|
Convey("should delete the missing alert", func() {
|
|
|
- query := m.GetAlertsQuery{DashboardId: testDash.Id, OrgId: 1, User: &m.SignedInUser{OrgRole: m.ROLE_ADMIN}}
|
|
|
|
|
|
|
+ query := m.GetAlertsQuery{DashboardIDs: []int64{testDash.Id}, OrgId: 1, User: &m.SignedInUser{OrgRole: m.ROLE_ADMIN}}
|
|
|
err2 := HandleAlertsQuery(&query)
|
|
err2 := HandleAlertsQuery(&query)
|
|
|
So(err2, ShouldBeNil)
|
|
So(err2, ShouldBeNil)
|
|
|
So(len(query.Result), ShouldEqual, 2)
|
|
So(len(query.Result), ShouldEqual, 2)
|
|
@@ -232,7 +233,7 @@ func TestAlertingDataAccess(t *testing.T) {
|
|
|
So(err, ShouldBeNil)
|
|
So(err, ShouldBeNil)
|
|
|
|
|
|
|
|
Convey("Alerts should be removed", func() {
|
|
Convey("Alerts should be removed", func() {
|
|
|
- query := m.GetAlertsQuery{DashboardId: testDash.Id, OrgId: 1, User: &m.SignedInUser{OrgRole: m.ROLE_ADMIN}}
|
|
|
|
|
|
|
+ query := m.GetAlertsQuery{DashboardIDs: []int64{testDash.Id}, OrgId: 1, User: &m.SignedInUser{OrgRole: m.ROLE_ADMIN}}
|
|
|
err2 := HandleAlertsQuery(&query)
|
|
err2 := HandleAlertsQuery(&query)
|
|
|
|
|
|
|
|
So(testDash.Id, ShouldEqual, 1)
|
|
So(testDash.Id, ShouldEqual, 1)
|