def testtransport_connect_to_server(self):
"""
Test to check the connect to server of the transport instance of the signal class.
"""
sig_dict, signal = self.setup_vars_mocks()
transport = XmppTransport.factory(
1, sig_dict["Signal"]["Overlays"]["A0FB389"], signal, None, None,
None)
transport._sig.sig_log = MagicMock()
transport.connect = MagicMock()
XmppTransport.connect_to_server(transport)
transport._sig.sig_log.assert_called_once()
transport.connect.assert_called_once()
print("Passed : testtransport_connect_to_server")
def testsignal_create_transport(self, mock_factory):
"""
Test to check the create transport method of the signal class.
"""
cfx_handle = Mock()
cfx_handle.query_param.return_value = 30
module = importlib.import_module("modules.{0}".format("Signal"))
module_class = getattr(module, "Signal")
sig_dict = {
"Signal": {
"Enabled": True,
"Overlays": {
"A0FB389": {
"HostAddress": "1.1.1.1",
"Port": "5222",
"Username": "******",
"Password": "******",
"AuthenticationMethod": "PASSWORD"
}
},
"NodeId": "1234434323"
}
}
signal = module_class(cfx_handle, sig_dict["Signal"], "Signal")
cfx_handle._cm_instance = signal
cfx_handle._cm_config = sig_dict
transport = XmppTransport(
sig_dict["Signal"]["Overlays"]["A0FB389"]["Username"],
sig_dict["Signal"]["Overlays"]["A0FB389"]["Password"],
sasl_mech="PLAIN")
mock_factory.return_value = transport
transport.connect_to_server = MagicMock()
assert signal._create_transport_instance(
"1", sig_dict["Signal"]["Overlays"]["A0FB389"], None,
None) == transport
print("Passed : testsignal_create_transport")
