def test_get_class_with_two_registered_class(self): klass1 = self._new_class('foo') klass2 = self._new_class('foo') msg = self._new_msg('foo') cti_command_registry.register_class(klass1) cti_command_registry.register_class(klass2) result = cti_command_registry.get_class(msg) self.assertEqual(sorted(result), sorted([klass1, klass2]))
def test_get_class_with_registered_class_that_doesnt_match(self): klass = self._new_class('foo') klass.match_message.return_value = False msg = self._new_msg('foo') cti_command_registry.register_class(klass) result = cti_command_registry.get_class(msg) klass.match_message.assert_called_once_with(msg) self.assertEqual(result, [])
def add_to_registry(self): cti_command_registry.register_class(self)