Add updating cached allocations

nomad/api_querier_mock.go

@@ -1,4 +1,4 @@
-// Code generated by mockery v0.0.0-dev. DO NOT EDIT.
+// Code generated by mockery v2.8.0. DO NOT EDIT.
 
 package nomad
 
@@ -19,6 +19,29 @@ type apiQuerierMock struct {
 	mock.Mock
 }
 
+// AllocationStream provides a mock function with given fields: ctx
+func (_m *apiQuerierMock) AllocationStream(ctx context.Context) (<-chan *api.Events, error) {
+	ret := _m.Called(ctx)
+
+	var r0 <-chan *api.Events
+	if rf, ok := ret.Get(0).(func(context.Context) <-chan *api.Events); ok {
+		r0 = rf(ctx)
+	} else {
+		if ret.Get(0) != nil {
+			r0 = ret.Get(0).(<-chan *api.Events)
+		}
+	}
+
+	var r1 error
+	if rf, ok := ret.Get(1).(func(context.Context) error); ok {
+		r1 = rf(ctx)
+	} else {
+		r1 = ret.Error(1)
+	}
+
+	return r0, r1
+}
+
 // DeleteRunner provides a mock function with given fields: runnerId
 func (_m *apiQuerierMock) DeleteRunner(runnerId string) error {
 	ret := _m.Called(runnerId)