|  | @@ -10,7 +10,7 @@ from celery.five import Empty
 | 
	
		
			
				|  |  |  from celery.platforms import EX_FAILURE
 | 
	
		
			
				|  |  |  from celery.worker import state
 | 
	
		
			
				|  |  |  from celery.worker.consumer import Consumer
 | 
	
		
			
				|  |  | -from celery.worker.loops import asynloop, synloop
 | 
	
		
			
				|  |  | +from celery.worker.loops import _quick_drain, asynloop, synloop
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  |  from celery.tests.case import AppCase, Mock, task_message_from_sig
 | 
	
		
			
				|  |  |  
 | 
	
	
		
			
				|  | @@ -126,7 +126,7 @@ class test_asynloop(AppCase):
 | 
	
		
			
				|  |  |      def test_drain_after_consume(self):
 | 
	
		
			
				|  |  |          x, _ = get_task_callback(self.app, transport_driver_type='amqp')
 | 
	
		
			
				|  |  |          self.assertIn(
 | 
	
		
			
				|  |  | -            x.connection.drain_events, [p.fun for p in x.hub._ready],
 | 
	
		
			
				|  |  | +            _quick_drain, [p.fun for p in x.hub._ready],
 | 
	
		
			
				|  |  |          )
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  |      def test_setup_heartbeat(self):
 |