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]))
Exemple #2
0
    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, [])
Exemple #4
0
    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, [])
Exemple #5
0
 def add_to_registry(self):
     cti_command_registry.register_class(self)