Add some mock clients
@@ -31,3 +31,15 @@ func (c *MockClient) ListChecks(org string) ([]types.CheckConfig, error) {
args := c.Called(org)
return args.Get(0).([]types.CheckConfig), args.Error(1)
}
+
+// AddCheckHook for use with mock lib
+func (c *MockClient) AddCheckHook(check string, checkHook *types.CheckHook) error {
+ args := c.Called(check, checkHook)
+ return args.Error(0)
+}
+
+// RemoveCheckHook for use with mock lib
+func (c *MockClient) RemoveCheckHook(checkName string, hookType string, hookName string) error {
+ args := c.Called(checkName, hookType, hookName)
+ return args.Error(0)
+}
