|  | @@ -25,7 +25,8 @@ from celery.bin.celeryd import WorkerCommand, windows_main, \
 | 
	
		
			
				|  |  |  from celery.exceptions import ImproperlyConfigured
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  |  from celery.tests.compat import catch_warnings
 | 
	
		
			
				|  |  | -from celery.tests.utils import AppCase, StringIO, mask_modules, reset_modules
 | 
	
		
			
				|  |  | +from celery.tests.utils import (AppCase, StringIO, mask_modules,
 | 
	
		
			
				|  |  | +                                reset_modules, skip_unless_module)
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  |  from celery.utils.patch import ensure_process_aware_logger
 | 
	
	
		
			
				|  | @@ -73,12 +74,17 @@ class test_compilation(AppCase):
 | 
	
		
			
				|  |  |                  from celery.apps.worker import cpu_count
 | 
	
		
			
				|  |  |                  self.assertEqual(cpu_count(), 2)
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  | -    @patch("multiprocessing.cpu_count")
 | 
	
		
			
				|  |  | +    @skip_unless_module("multiprocessing")
 | 
	
		
			
				|  |  |      def test_no_cpu_count(self, pcount):
 | 
	
		
			
				|  |  | -        pcount.side_effect = NotImplementedError("cpu_count")
 | 
	
		
			
				|  |  | -        from celery.apps.worker import cpu_count
 | 
	
		
			
				|  |  | -        self.assertEqual(cpu_count(), 2)
 | 
	
		
			
				|  |  | -        pcount.assert_called_with()
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  | +        @patch("multiprocessing.cpu_count")
 | 
	
		
			
				|  |  | +        def _do_test():
 | 
	
		
			
				|  |  | +            pcount.side_effect = NotImplementedError("cpu_count")
 | 
	
		
			
				|  |  | +            from celery.apps.worker import cpu_count
 | 
	
		
			
				|  |  | +            self.assertEqual(cpu_count(), 2)
 | 
	
		
			
				|  |  | +            pcount.assert_called_with()
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  | +        _do_test()
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  |      def test_process_name_wo_mp(self):
 | 
	
		
			
				|  |  |          with mask_modules("multiprocessing"):
 | 
	
	
		
			
				|  | @@ -86,11 +92,16 @@ class test_compilation(AppCase):
 | 
	
		
			
				|  |  |                  from celery.apps.worker import get_process_name
 | 
	
		
			
				|  |  |                  self.assertIsNone(get_process_name())
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  | -    @patch("multiprocessing.current_process")
 | 
	
		
			
				|  |  | +    @skip_unless_module("multiprocessing")
 | 
	
		
			
				|  |  |      def test_process_name_w_mp(self, current_process):
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  | +        @patch("multiprocessing.current_process")
 | 
	
		
			
				|  |  | +        def _do_test():
 | 
	
		
			
				|  |  |              from celery.apps.worker import get_process_name
 | 
	
		
			
				|  |  |              self.assertTrue(get_process_name())
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  | +        _do_test()
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  |  class test_Worker(AppCase):
 | 
	
		
			
				|  |  |      Worker = Worker
 |