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")