diff --git a/start_bumper.py b/start_bumper.py index 4ec5414..3fc4fc1 100644 --- a/start_bumper.py +++ b/start_bumper.py @@ -48,7 +48,7 @@ async def main(): xmpp_address = (listen_host, 5223) mqtt_address = (listen_host, 8883) - xmpp_server = bumper.XMPPServer(xmpp_address) + mqtt_server = bumper.MQTTServer(mqtt_address) mqtt_helperbot = bumper.MQTTHelperBot(mqtt_address) conf_server = bumper.ConfServer( diff --git a/tests/test_xmppserver.py b/tests/test_xmppserver.py new file mode 100644 index 0000000..3de9298 --- /dev/null +++ b/tests/test_xmppserver.py @@ -0,0 +1,48 @@ +import mock +import bumper +import asyncio +import pytest +import os +import json +import tinydb +import pytest_asyncio + + +def return_send_data(data, *args, **kwargs): + return data + + +def mock_transport_extra_info(*args, **kwargs): + return ("127.0.0.1", 1234) + + +async def test_client(*args, **kwargs): + test_transport = asyncio.Transport() + test_transport.get_extra_info = mock.Mock(return_value=mock_transport_extra_info()) + test_transport.write = mock.Mock(return_value=return_send_data) + xmppclient = bumper.xmppserver.XMPPAsyncClient(test_transport) + xmppclient.state = xmppclient.INIT # Set client state to INIT + mock_send = xmppclient.send = mock.Mock(side_effect=return_send_data) + + # Send init connect stream from "client" + test_data = "".encode( + "utf-8" + ) + xmppclient._parse_data(test_data) + + # Expect 2 calls to send + assert mock_send.call_count == 2 + # Server opens stream + assert ( + mock_send.mock_calls[0].args[0] + == '' + ) + # Server tells client available features + assert ( + mock_send.mock_calls[1].args[0] + == '' + ) + + # Reset mock calls + mock_send.reset_mock() +