mirror of
https://github.com/grafana/grafana.git
synced 2024-11-27 11:20:27 -06:00
158 lines
4.7 KiB
Go
158 lines
4.7 KiB
Go
package api
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/grafana/grafana/pkg/api/dtos"
|
|
"github.com/grafana/grafana/pkg/bus"
|
|
m "github.com/grafana/grafana/pkg/models"
|
|
"github.com/grafana/grafana/pkg/services/search"
|
|
|
|
. "github.com/smartystreets/goconvey/convey"
|
|
)
|
|
|
|
func TestAlertingApiEndpoint(t *testing.T) {
|
|
Convey("Given an alert in a dashboard with an acl", t, func() {
|
|
|
|
singleAlert := &m.Alert{Id: 1, DashboardId: 1, Name: "singlealert"}
|
|
|
|
bus.AddHandler("test", func(query *m.GetAlertByIdQuery) error {
|
|
query.Result = singleAlert
|
|
return nil
|
|
})
|
|
|
|
viewerRole := m.ROLE_VIEWER
|
|
editorRole := m.ROLE_EDITOR
|
|
|
|
aclMockResp := []*m.DashboardAclInfoDTO{}
|
|
bus.AddHandler("test", func(query *m.GetDashboardAclInfoListQuery) error {
|
|
query.Result = aclMockResp
|
|
return nil
|
|
})
|
|
|
|
bus.AddHandler("test", func(query *m.GetTeamsByUserQuery) error {
|
|
query.Result = []*m.TeamDTO{}
|
|
return nil
|
|
})
|
|
|
|
Convey("When user is editor and not in the ACL", func() {
|
|
Convey("Should not be able to pause the alert", func() {
|
|
cmd := dtos.PauseAlertCommand{
|
|
AlertId: 1,
|
|
Paused: true,
|
|
}
|
|
postAlertScenario("When calling POST on", "/api/alerts/1/pause", "/api/alerts/:alertId/pause", m.ROLE_EDITOR, cmd, func(sc *scenarioContext) {
|
|
CallPauseAlert(sc)
|
|
So(sc.resp.Code, ShouldEqual, 403)
|
|
})
|
|
})
|
|
})
|
|
|
|
Convey("When user is editor and dashboard has default ACL", func() {
|
|
aclMockResp = []*m.DashboardAclInfoDTO{
|
|
{Role: &viewerRole, Permission: m.PERMISSION_VIEW},
|
|
{Role: &editorRole, Permission: m.PERMISSION_EDIT},
|
|
}
|
|
|
|
Convey("Should be able to pause the alert", func() {
|
|
cmd := dtos.PauseAlertCommand{
|
|
AlertId: 1,
|
|
Paused: true,
|
|
}
|
|
postAlertScenario("When calling POST on", "/api/alerts/1/pause", "/api/alerts/:alertId/pause", m.ROLE_EDITOR, cmd, func(sc *scenarioContext) {
|
|
CallPauseAlert(sc)
|
|
So(sc.resp.Code, ShouldEqual, 200)
|
|
})
|
|
})
|
|
})
|
|
|
|
loggedInUserScenarioWithRole("When calling GET on", "GET", "/api/alerts?dashboardId=1", "/api/alerts", m.ROLE_EDITOR, func(sc *scenarioContext) {
|
|
var searchQuery *search.Query
|
|
bus.AddHandler("test", func(query *search.Query) error {
|
|
searchQuery = query
|
|
return nil
|
|
})
|
|
|
|
var getAlertsQuery *m.GetAlertsQuery
|
|
bus.AddHandler("test", func(query *m.GetAlertsQuery) error {
|
|
getAlertsQuery = query
|
|
return nil
|
|
})
|
|
|
|
sc.handlerFunc = GetAlerts
|
|
sc.fakeReqWithParams("GET", sc.url, map[string]string{}).exec()
|
|
|
|
So(searchQuery, ShouldBeNil)
|
|
So(getAlertsQuery, ShouldNotBeNil)
|
|
})
|
|
|
|
loggedInUserScenarioWithRole("When calling GET on", "GET", "/api/alerts?dashboardId=1&dashboardId=2&folderId=3&dashboardTag=abc&dashboardQuery=dbQuery&limit=5&query=alertQuery", "/api/alerts", m.ROLE_EDITOR, func(sc *scenarioContext) {
|
|
var searchQuery *search.Query
|
|
bus.AddHandler("test", func(query *search.Query) error {
|
|
searchQuery = query
|
|
query.Result = search.HitList{
|
|
&search.Hit{Id: 1},
|
|
&search.Hit{Id: 2},
|
|
}
|
|
return nil
|
|
})
|
|
|
|
var getAlertsQuery *m.GetAlertsQuery
|
|
bus.AddHandler("test", func(query *m.GetAlertsQuery) error {
|
|
getAlertsQuery = query
|
|
return nil
|
|
})
|
|
|
|
sc.handlerFunc = GetAlerts
|
|
sc.fakeReqWithParams("GET", sc.url, map[string]string{}).exec()
|
|
|
|
So(searchQuery, ShouldNotBeNil)
|
|
So(searchQuery.DashboardIds[0], ShouldEqual, 1)
|
|
So(searchQuery.DashboardIds[1], ShouldEqual, 2)
|
|
So(searchQuery.FolderIds[0], ShouldEqual, 3)
|
|
So(searchQuery.Tags[0], ShouldEqual, "abc")
|
|
So(searchQuery.Title, ShouldEqual, "dbQuery")
|
|
|
|
So(getAlertsQuery, ShouldNotBeNil)
|
|
So(getAlertsQuery.DashboardIDs[0], ShouldEqual, 1)
|
|
So(getAlertsQuery.DashboardIDs[1], ShouldEqual, 2)
|
|
So(getAlertsQuery.Limit, ShouldEqual, 5)
|
|
So(getAlertsQuery.Query, ShouldEqual, "alertQuery")
|
|
})
|
|
|
|
loggedInUserScenarioWithRole("When calling GET on", "GET", "/api/alert-notifications/1", "/alert-notifications/:notificationId", m.ROLE_ADMIN, func(sc *scenarioContext) {
|
|
sc.handlerFunc = GetAlertNotificationByID
|
|
sc.fakeReqWithParams("GET", sc.url, map[string]string{}).exec()
|
|
So(sc.resp.Code, ShouldEqual, 404)
|
|
})
|
|
})
|
|
}
|
|
|
|
func CallPauseAlert(sc *scenarioContext) {
|
|
bus.AddHandler("test", func(cmd *m.PauseAlertCommand) error {
|
|
return nil
|
|
})
|
|
|
|
sc.fakeReqWithParams("POST", sc.url, map[string]string{}).exec()
|
|
}
|
|
|
|
func postAlertScenario(desc string, url string, routePattern string, role m.RoleType, cmd dtos.PauseAlertCommand, fn scenarioFunc) {
|
|
Convey(desc+" "+url, func() {
|
|
defer bus.ClearBusHandlers()
|
|
|
|
sc := setupScenarioContext(url)
|
|
sc.defaultHandler = Wrap(func(c *m.ReqContext) Response {
|
|
sc.context = c
|
|
sc.context.UserId = TestUserID
|
|
sc.context.OrgId = TestOrgID
|
|
sc.context.OrgRole = role
|
|
|
|
return PauseAlert(c, cmd)
|
|
})
|
|
|
|
sc.m.Post(routePattern, sc.defaultHandler)
|
|
|
|
fn(sc)
|
|
})
|
|
}
|