mockCron.go 957 B

123456789101112131415161718192021222324252627282930313233
  1. // Copyright 2023 EMQ Technologies Co., Ltd.
  2. //
  3. // Licensed under the Apache License, Version 2.0 (the "License");
  4. // you may not use this file except in compliance with the License.
  5. // You may obtain a copy of the License at
  6. //
  7. // http://www.apache.org/licenses/LICENSE-2.0
  8. //
  9. // Unless required by applicable law or agreed to in writing, software
  10. // distributed under the License is distributed on an "AS IS" BASIS,
  11. // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  12. // See the License for the specific language governing permissions and
  13. // limitations under the License.
  14. package rule
  15. import "github.com/robfig/cron/v3"
  16. type MockCron struct {
  17. index int
  18. }
  19. func (m *MockCron) Start() {}
  20. // AddFunc
  21. // MockCron execute function immediately at once
  22. func (m *MockCron) AddFunc(_ string, cmd func()) (cron.EntryID, error) {
  23. cmd()
  24. m.index++
  25. return cron.EntryID(m.index), nil
  26. }
  27. func (m *MockCron) Remove(id cron.EntryID) {}