def test_message_handler_noinstall(self, mock_method):
y = YateBase()
y.register_message_handler("call.execute", lambda: True, install=False)
self.assertIn("call.execute", y._message_handlers)
handler = y._message_handlers["call.execute"]
self.assertEqual("call.execute", handler.message)
mock_method.assert_not_called()
def test_message_handler_install_done_callback(self):
y = YateBase()
done_callback_mock = MagicMock()
y.register_message_handler("call.execute",
lambda: True,
100,
"myattrib",
"myvalue",
done_callback=done_callback_mock)
done_callback_mock.assert_not_called()
y._recv_message_raw(b"%%<install:100:call.execute:true")
done_callback_mock.assert_called_with(True)
def test_message_handler_install(self, mock_method):
y = YateBase()
y.register_message_handler("call.execute", lambda: True, 100,
"myattrib", "myvalue")
self.assertIn("call.execute", y._message_handlers)
handler = y._message_handlers["call.execute"]
self.assertEqual("call.execute", handler.message)
self.assertEqual("100", str(handler.priority))
self.assertEqual("myattrib", handler.filter_attribute)
self.assertEqual("myvalue", handler.filter_value)
self.assertFalse(handler.installed)
self.assertFalse(handler.uninstalled)
mock_method.assert_called_with(
b"%%>install:100:call.execute:myattrib:myvalue")
# now simulate that yate responded correctly
y._recv_message_raw(b"%%<install:100:call.execute:true")
self.assertTrue(handler.installed)