Merge "Remove registry.subscribe()"

This commit is contained in:
Zuul 2018-02-15 04:29:22 +00:00 committed by Gerrit Code Review
commit 87eaf8ffc5
2 changed files with 1 additions and 25 deletions

View File

@ -10,8 +10,6 @@
# License for the specific language governing permissions and limitations
# under the License.
import debtcollector
from neutron.api.rpc.callbacks import resource_manager
@ -21,19 +19,6 @@ def _get_manager():
return resource_manager.ConsumerResourceCallbacksManager()
@debtcollector.removals.remove(
message="This will be removed in the future. Please register callbacks "
"using the 'register' method in this model and adjust the "
"callback to accept the context and resource type as arguments.",
version="Ocata"
)
def subscribe(callback, resource_type):
# temporary hack to differentiate between callback types until the
# 'subscribe' method is removed
callback.__dict__['_ACCEPTS_CONTEXT'] = False
_get_manager().register(callback, resource_type)
def register(callback, resource_type):
# TODO(kevinbenton): remove this on debt collection
callback.__dict__['_ACCEPTS_CONTEXT'] = True

View File

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