Просмотр исходного кода

feat(alerting): fixed test issues

Torkel Ödegaard 9 лет назад
Родитель
Сommit
3289225b77

+ 0 - 15
pkg/services/alerting/alerting.go

@@ -1,9 +1,6 @@
 package alerting
 
 import (
-	"github.com/grafana/grafana/pkg/bus"
-	"github.com/grafana/grafana/pkg/log"
-	m "github.com/grafana/grafana/pkg/models"
 	"github.com/grafana/grafana/pkg/setting"
 	_ "github.com/grafana/grafana/pkg/tsdb/graphite"
 )
@@ -29,15 +26,3 @@ func Init() {
 	// go scheduler.executor(&ExecutorImpl{})
 	// go scheduler.handleResponses()
 }
-
-func saveState(result *AlertResult) {
-	cmd := &m.UpdateAlertStateCommand{
-		AlertId:  result.AlertJob.Rule.Id,
-		NewState: result.State,
-		Info:     result.Description,
-	}
-
-	if err := bus.Dispatch(cmd); err != nil {
-		log.Error(2, "failed to save state %v", err)
-	}
-}

+ 2 - 2
pkg/services/alerting/engine.go

@@ -114,11 +114,11 @@ func (e *Engine) resultHandler() {
 
 				result.State = alertstates.Critical
 				result.Description = fmt.Sprintf("Failed to run check after %d retires, Error: %v", maxRetries, result.Error)
-				saveState(result)
+				e.saveState(result)
 			}
 		} else {
 			result.AlertJob.RetryCount = 0
-			saveState(result)
+			e.saveState(result)
 		}
 	}
 }

+ 1 - 1
pkg/services/alerting/executor_test.go

@@ -10,7 +10,7 @@ import (
 
 func TestAlertingExecutor(t *testing.T) {
 	Convey("Test alert execution", t, func() {
-		executor := &ExecutorImpl{}
+		executor := NewExecutor()
 
 		Convey("single time serie", func() {
 			Convey("Show return ok since avg is above 2", func() {

+ 1 - 0
pkg/services/sqlstore/alert_rule_changes_test.go

@@ -48,6 +48,7 @@ func TestAlertRuleChangesDataAccess(t *testing.T) {
 
 			SaveAlerts(&cmd)
 
+			query := &m.GetAlertChangesQuery{OrgId: FakeOrgId}
 			er := GetAlertRuleChanges(query)
 			So(er, ShouldBeNil)
 			So(len(query.Result), ShouldEqual, 1)

+ 1 - 1
pkg/tsdb/graphite/graphite_test.go

@@ -16,7 +16,7 @@ func TestGraphite(t *testing.T) {
 		})
 
 		queries := tsdb.QuerySlice{
-			&tsdb.Query{Query: "apps.backend.*.counters.requests.count"},
+			&tsdb.Query{Query: "{\"target\": \"apps.backend.*.counters.requests.count\"}"},
 		}
 		context := tsdb.NewQueryContext(queries, tsdb.TimeRange{})
 

+ 14 - 16
pkg/tsdb/tsdb_test.go

@@ -55,7 +55,7 @@ func TestMetricQuery(t *testing.T) {
 	Convey("When executing request with one query", t, func() {
 		req := &Request{
 			Queries: QuerySlice{
-				{RefId: "A", Query: "asd", DataSource: &DataSourceInfo{Id: 1, Type: "test"}},
+				{RefId: "A", Query: "asd", DataSource: &DataSourceInfo{Id: 1, PluginId: "test"}},
 			},
 		}
 
@@ -74,8 +74,8 @@ func TestMetricQuery(t *testing.T) {
 	Convey("When executing one request with two queries from same data source", t, func() {
 		req := &Request{
 			Queries: QuerySlice{
-				{RefId: "A", Query: "asd", DataSource: &DataSourceInfo{Id: 1, Type: "test"}},
-				{RefId: "B", Query: "asd", DataSource: &DataSourceInfo{Id: 1, Type: "test"}},
+				{RefId: "A", Query: "asd", DataSource: &DataSourceInfo{Id: 1, PluginId: "test"}},
+				{RefId: "B", Query: "asd", DataSource: &DataSourceInfo{Id: 1, PluginId: "test"}},
 			},
 		}
 
@@ -100,9 +100,9 @@ func TestMetricQuery(t *testing.T) {
 	Convey("When executing one request with three queries from different datasources", t, func() {
 		req := &Request{
 			Queries: QuerySlice{
-				{RefId: "A", Query: "asd", DataSource: &DataSourceInfo{Id: 1, Type: "test"}},
-				{RefId: "B", Query: "asd", DataSource: &DataSourceInfo{Id: 1, Type: "test"}},
-				{RefId: "C", Query: "asd", DataSource: &DataSourceInfo{Id: 2, Type: "test"}},
+				{RefId: "A", Query: "asd", DataSource: &DataSourceInfo{Id: 1, PluginId: "test"}},
+				{RefId: "B", Query: "asd", DataSource: &DataSourceInfo{Id: 1, PluginId: "test"}},
+				{RefId: "C", Query: "asd", DataSource: &DataSourceInfo{Id: 2, PluginId: "test"}},
 			},
 		}
 
@@ -117,24 +117,22 @@ func TestMetricQuery(t *testing.T) {
 	Convey("When query uses data source of unknown type", t, func() {
 		req := &Request{
 			Queries: QuerySlice{
-				{RefId: "A", Query: "asd", DataSource: &DataSourceInfo{Id: 1, Type: "asdasdas"}},
+				{RefId: "A", Query: "asd", DataSource: &DataSourceInfo{Id: 1, PluginId: "asdasdas"}},
 			},
 		}
 
-		res, err := HandleRequest(req)
-		So(err, ShouldBeNil)
-
-		Convey("Should return error", func() {
-			So(res.Results["A"].Error.Error(), ShouldContainSubstring, "not find")
-		})
+		_, err := HandleRequest(req)
+		So(err, ShouldNotBeNil)
 	})
 
 	Convey("When executing request that depend on other query", t, func() {
 		req := &Request{
 			Queries: QuerySlice{
-				{RefId: "A", Query: "asd", DataSource: &DataSourceInfo{Id: 1, Type: "test"}},
-				{RefId: "B", Query: "#A / 2", DataSource: &DataSourceInfo{Id: 2, Type: "test"},
-					Depends: []string{"A"},
+				{
+					RefId: "A", Query: "asd", DataSource: &DataSourceInfo{Id: 1, PluginId: "test"},
+				},
+				{
+					RefId: "B", Query: "#A / 2", DataSource: &DataSourceInfo{Id: 2, PluginId: "test"}, Depends: []string{"A"},
 				},
 			},
 		}