@ -21,12 +21,6 @@ class ConsumerRegistryTestCase(base.BaseTestCase):
def test__get_manager_is_singleton ( self ) :
self . assertIs ( registry . _get_manager ( ) , registry . _get_manager ( ) )
@mock.patch.object ( registry , ' _get_manager ' )
def test_subscribe ( self , manager_mock ) :
callback = lambda : None
registry . subscribe ( callback , ' TYPE ' )
manager_mock ( ) . register . assert_called_with ( callback , ' TYPE ' )
@mock.patch.object ( registry , ' _get_manager ' )
def test_register ( self , manager_mock ) :
callback = lambda : None
@ -53,14 +47,11 @@ class ConsumerRegistryTestCase(base.BaseTestCase):
context = mock . Mock ( )
callback1 = mock . Mock ( )
callback2 = mock . Mock ( )
legacy_callback = mock . Mock ( )
registry . register ( callback1 , ' x ' )
registry . register ( callback2 , ' x ' )
registry . subscribe ( legacy_callback , ' x ' )
callbacks = { callback1 , callback2 , legacy_callback }
callbacks = { callback1 , callback2 }
manager_mock ( ) . get_callbacks . return_value = callbacks
registry . push ( context , resource_type_ , [ resource_ ] , event_type_ )
for callback in ( callback1 , callback2 ) :
callback . assert_called_with ( context , resource_type_ ,
[ resource_ ] , event_type_ )
legacy_callback . assert_called_with ( [ resource_ ] , event_type_ )