|
@@ -2,12 +2,12 @@ import logging
|
|
|
import sys
|
|
|
import unittest2 as unittest
|
|
|
|
|
|
+from celery import beat
|
|
|
from celery import platform
|
|
|
-from celery.beat import ClockService
|
|
|
-from celery.bin import celerybeat as beat
|
|
|
+from celery.bin import celerybeat as celerybeat
|
|
|
|
|
|
|
|
|
-class MockClockService(ClockService):
|
|
|
+class MockService(beat.Service):
|
|
|
started = False
|
|
|
in_sync = False
|
|
|
|
|
@@ -18,15 +18,15 @@ class MockClockService(ClockService):
|
|
|
self.__class__.in_sync = True
|
|
|
|
|
|
|
|
|
-class MockBeat(beat.Beat):
|
|
|
+class MockBeat(celerybeat.Beat):
|
|
|
running = False
|
|
|
|
|
|
def run(self):
|
|
|
self.__class__.running = True
|
|
|
|
|
|
|
|
|
-class MockBeat2(beat.Beat):
|
|
|
- ClockService = MockClockService
|
|
|
+class MockBeat2(celerybeat.Beat):
|
|
|
+ Service = MockService
|
|
|
|
|
|
def install_sync_handler(self, b):
|
|
|
pass
|
|
@@ -35,29 +35,29 @@ class MockBeat2(beat.Beat):
|
|
|
class test_Beat(unittest.TestCase):
|
|
|
|
|
|
def test_loglevel_string(self):
|
|
|
- b = beat.Beat(loglevel="DEBUG")
|
|
|
+ b = celerybeat.Beat(loglevel="DEBUG")
|
|
|
self.assertEqual(b.loglevel, logging.DEBUG)
|
|
|
|
|
|
- b2 = beat.Beat(loglevel=logging.DEBUG)
|
|
|
+ b2 = celerybeat.Beat(loglevel=logging.DEBUG)
|
|
|
self.assertEqual(b2.loglevel, logging.DEBUG)
|
|
|
|
|
|
def test_init_loader(self):
|
|
|
- b = beat.Beat()
|
|
|
+ b = celerybeat.Beat()
|
|
|
b.init_loader()
|
|
|
|
|
|
def test_startup_info(self):
|
|
|
- b = beat.Beat()
|
|
|
+ b = celerybeat.Beat()
|
|
|
self.assertIn("@stderr", b.startup_info())
|
|
|
|
|
|
def test_process_title(self):
|
|
|
- b = beat.Beat()
|
|
|
+ b = celerybeat.Beat()
|
|
|
b.set_process_title()
|
|
|
|
|
|
def test_run(self):
|
|
|
b = MockBeat2()
|
|
|
- MockClockService.started = False
|
|
|
+ MockService.started = False
|
|
|
b.run()
|
|
|
- self.assertTrue(MockClockService.started)
|
|
|
+ self.assertTrue(MockService.started)
|
|
|
|
|
|
def psig(self, fun, *args, **kwargs):
|
|
|
handlers = {}
|
|
@@ -73,39 +73,39 @@ class test_Beat(unittest.TestCase):
|
|
|
platform.install_signal_handler = p
|
|
|
|
|
|
def test_install_sync_handler(self):
|
|
|
- b = beat.Beat()
|
|
|
- clock = MockClockService()
|
|
|
- MockClockService.in_sync = False
|
|
|
+ b = celerybeat.Beat()
|
|
|
+ clock = MockService()
|
|
|
+ MockService.in_sync = False
|
|
|
handlers = self.psig(b.install_sync_handler, clock)
|
|
|
self.assertRaises(SystemExit, handlers["SIGINT"],
|
|
|
"SIGINT", object())
|
|
|
- self.assertTrue(MockClockService.in_sync)
|
|
|
- MockClockService.in_sync = False
|
|
|
+ self.assertTrue(MockService.in_sync)
|
|
|
+ MockService.in_sync = False
|
|
|
|
|
|
|
|
|
class test_div(unittest.TestCase):
|
|
|
|
|
|
def setUp(self):
|
|
|
- self.prev, beat.Beat = beat.Beat, MockBeat
|
|
|
+ self.prev, celerybeat.Beat = celerybeat.Beat, MockBeat
|
|
|
|
|
|
def tearDown(self):
|
|
|
- beat.Beat = self.prev
|
|
|
+ celerybeat.Beat = self.prev
|
|
|
|
|
|
def test_main(self):
|
|
|
sys.argv = [sys.argv[0], "-s", "foo"]
|
|
|
try:
|
|
|
- beat.main()
|
|
|
+ celerybeat.main()
|
|
|
self.assertTrue(MockBeat.running)
|
|
|
finally:
|
|
|
MockBeat.running = False
|
|
|
|
|
|
def test_run_celerybeat(self):
|
|
|
try:
|
|
|
- beat.run_celerybeat()
|
|
|
+ celerybeat.run_celerybeat()
|
|
|
self.assertTrue(MockBeat.running)
|
|
|
finally:
|
|
|
MockBeat.running = False
|
|
|
|
|
|
def test_parse_options(self):
|
|
|
- options = beat.parse_options(["-s", "foo"])
|
|
|
+ options = celerybeat.parse_options(["-s", "foo"])
|
|
|
self.assertEqual(options.schedule, "foo")
|