|  | @@ -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")
 |