Alerting: Condition evaluator with cached pipeline (#57479)

* create rule evaluator
* load header from the context
* init one factory
* update scheduler

pkg/services/ngalert/eval/eval_mocks/ConditionEvaluator.go

@@ -0,0 +1,137 @@
+// Code generated by mockery v2.14.0. DO NOT EDIT.
+
+package eval_mocks
+
+import (
+	context "context"
+
+	backend "github.com/grafana/grafana-plugin-sdk-go/backend"
+
+	eval "github.com/grafana/grafana/pkg/services/ngalert/eval"
+
+	mock "github.com/stretchr/testify/mock"
+
+	time "time"
+)
+
+// ConditionEvaluatorMock is an autogenerated mock type for the ConditionEvaluator type
+type ConditionEvaluatorMock struct {
+	mock.Mock
+}
+
+type ConditionEvaluatorMock_Expecter struct {
+	mock *mock.Mock
+}
+
+func (_m *ConditionEvaluatorMock) EXPECT() *ConditionEvaluatorMock_Expecter {
+	return &ConditionEvaluatorMock_Expecter{mock: &_m.Mock}
+}
+
+// Evaluate provides a mock function with given fields: ctx, now
+func (_m *ConditionEvaluatorMock) Evaluate(ctx context.Context, now time.Time) (eval.Results, error) {
+	ret := _m.Called(ctx, now)
+
+	var r0 eval.Results
+	if rf, ok := ret.Get(0).(func(context.Context, time.Time) eval.Results); ok {
+		r0 = rf(ctx, now)
+	} else {
+		if ret.Get(0) != nil {
+			r0 = ret.Get(0).(eval.Results)
+		}
+	}
+
+	var r1 error
+	if rf, ok := ret.Get(1).(func(context.Context, time.Time) error); ok {
+		r1 = rf(ctx, now)
+	} else {
+		r1 = ret.Error(1)
+	}
+
+	return r0, r1
+}
+
+// ConditionEvaluatorMock_Evaluate_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'Evaluate'
+type ConditionEvaluatorMock_Evaluate_Call struct {
+	*mock.Call
+}
+
+// Evaluate is a helper method to define mock.On call
+//   - ctx context.Context
+//   - now time.Time
+func (_e *ConditionEvaluatorMock_Expecter) Evaluate(ctx interface{}, now interface{}) *ConditionEvaluatorMock_Evaluate_Call {
+	return &ConditionEvaluatorMock_Evaluate_Call{Call: _e.mock.On("Evaluate", ctx, now)}
+}
+
+func (_c *ConditionEvaluatorMock_Evaluate_Call) Run(run func(ctx context.Context, now time.Time)) *ConditionEvaluatorMock_Evaluate_Call {
+	_c.Call.Run(func(args mock.Arguments) {
+		run(args[0].(context.Context), args[1].(time.Time))
+	})
+	return _c
+}
+
+func (_c *ConditionEvaluatorMock_Evaluate_Call) Return(_a0 eval.Results, _a1 error) *ConditionEvaluatorMock_Evaluate_Call {
+	_c.Call.Return(_a0, _a1)
+	return _c
+}
+
+// EvaluateRaw provides a mock function with given fields: ctx, now
+func (_m *ConditionEvaluatorMock) EvaluateRaw(ctx context.Context, now time.Time) (*backend.QueryDataResponse, error) {
+	ret := _m.Called(ctx, now)
+
+	var r0 *backend.QueryDataResponse
+	if rf, ok := ret.Get(0).(func(context.Context, time.Time) *backend.QueryDataResponse); ok {
+		r0 = rf(ctx, now)
+	} else {
+		if ret.Get(0) != nil {
+			r0 = ret.Get(0).(*backend.QueryDataResponse)
+		}
+	}
+
+	var r1 error
+	if rf, ok := ret.Get(1).(func(context.Context, time.Time) error); ok {
+		r1 = rf(ctx, now)
+	} else {
+		r1 = ret.Error(1)
+	}
+
+	return r0, r1
+}
+
+// ConditionEvaluatorMock_EvaluateRaw_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'EvaluateRaw'
+type ConditionEvaluatorMock_EvaluateRaw_Call struct {
+	*mock.Call
+}
+
+// EvaluateRaw is a helper method to define mock.On call
+//   - ctx context.Context
+//   - now time.Time
+func (_e *ConditionEvaluatorMock_Expecter) EvaluateRaw(ctx interface{}, now interface{}) *ConditionEvaluatorMock_EvaluateRaw_Call {
+	return &ConditionEvaluatorMock_EvaluateRaw_Call{Call: _e.mock.On("EvaluateRaw", ctx, now)}
+}
+
+func (_c *ConditionEvaluatorMock_EvaluateRaw_Call) Run(run func(ctx context.Context, now time.Time)) *ConditionEvaluatorMock_EvaluateRaw_Call {
+	_c.Call.Run(func(args mock.Arguments) {
+		run(args[0].(context.Context), args[1].(time.Time))
+	})
+	return _c
+}
+
+func (_c *ConditionEvaluatorMock_EvaluateRaw_Call) Return(resp *backend.QueryDataResponse, err error) *ConditionEvaluatorMock_EvaluateRaw_Call {
+	_c.Call.Return(resp, err)
+	return _c
+}
+
+type mockConstructorTestingTNewConditionEvaluatorMock interface {
+	mock.TestingT
+	Cleanup(func())
+}
+
+// NewConditionEvaluatorMock creates a new instance of ConditionEvaluatorMock. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations.
+func NewConditionEvaluatorMock(t mockConstructorTestingTNewConditionEvaluatorMock) *ConditionEvaluatorMock {
+	mock := &ConditionEvaluatorMock{}
+	mock.Mock.Test(t)
+
+	t.Cleanup(func() { mock.AssertExpectations(t) })
+
+	return mock
+}

pkg/services/ngalert/eval/eval_mocks/FakeFactory.go

@@ -0,0 +1,32 @@
+package eval_mocks
+
+import (
+	"errors"
+
+	"github.com/grafana/grafana/pkg/services/ngalert/eval"
+	"github.com/grafana/grafana/pkg/services/ngalert/models"
+)
+
+type fakeEvaluatorFactory struct {
+	err       error
+	evaluator eval.ConditionEvaluator
+}
+
+func NewEvaluatorFactory(evaluator eval.ConditionEvaluator) eval.EvaluatorFactory {
+	return &fakeEvaluatorFactory{evaluator: evaluator}
+}
+
+func NewFailingEvaluatorFactory(err error) eval.EvaluatorFactory {
+	if err == nil {
+		err = errors.New("test")
+	}
+	return &fakeEvaluatorFactory{err: err}
+}
+
+func (f fakeEvaluatorFactory) Validate(ctx eval.EvaluationContext, condition models.Condition) error {
+	return f.err
+}
+
+func (f fakeEvaluatorFactory) Create(ctx eval.EvaluationContext, condition models.Condition) (eval.ConditionEvaluator, error) {
+	return f.evaluator, f.err
+}