56 lines
1.4 KiB
Go
56 lines
1.4 KiB
Go
package api
|
|
|
|
import (
|
|
"context"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"testing"
|
|
|
|
"github.com/grafana/grafana-plugin-sdk-go/data"
|
|
"github.com/stretchr/testify/assert"
|
|
|
|
"github.com/grafana/grafana/pkg/infra/log"
|
|
contextmodel "github.com/grafana/grafana/pkg/services/contexthandler/model"
|
|
"github.com/grafana/grafana/pkg/services/ngalert/models"
|
|
"github.com/grafana/grafana/pkg/services/user"
|
|
"github.com/grafana/grafana/pkg/web"
|
|
)
|
|
|
|
type mockHistorian struct{}
|
|
|
|
func (m *mockHistorian) Query(ctx context.Context, query models.HistoryQuery) (*data.Frame, error) {
|
|
return &data.Frame{Name: "history"}, nil
|
|
}
|
|
|
|
func TestRouteQueryStateHistory(t *testing.T) {
|
|
testCases := []struct {
|
|
name string
|
|
queryParams string
|
|
expectedCode int
|
|
}{
|
|
{"valid states", "previous=Normal¤t=Alerting", http.StatusOK},
|
|
{"invalid previous", "previous=InvalidState", http.StatusBadRequest},
|
|
{"invalid current", "current=InvalidState", http.StatusBadRequest},
|
|
}
|
|
|
|
srv := &HistorySrv{
|
|
logger: log.NewNopLogger(),
|
|
hist: &mockHistorian{},
|
|
}
|
|
|
|
for _, tt := range testCases {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
req := httptest.NewRequest(http.MethodGet, "/?"+tt.queryParams, nil)
|
|
|
|
c := &contextmodel.ReqContext{
|
|
Context: &web.Context{Req: req},
|
|
SignedInUser: &user.SignedInUser{OrgID: 1},
|
|
}
|
|
|
|
resp := srv.RouteQueryStateHistory(c)
|
|
|
|
assert.Equal(t, tt.expectedCode, resp.Status())
|
|
})
|
|
}
|
|
}
|