test module connector

aircox_streamer/tests.py

@@ -1 +0,0 @@
-# Create your tests here.

aircox_streamer/tests/conftest.py

@@ -0,0 +1,19 @@
+import pytest
+from aircox_streamer.connector import Connector
+
+@pytest.fixture
+def connector():
+    return Connector()
+
+@pytest.fixture
+def connector_inet():
+    host, port = '0.0.0.0', 6767
+    address = host, port
+    connector_instance = Connector(address)
+    return connector_instance
+
+@pytest.fixture
+def connector_unix():
+    address = 'path/to/socket'
+    connector_instance = Connector(address)
+    return connector_instance

aircox_streamer/tests/test_connector.py

@@ -0,0 +1,116 @@
+import pytest
+from unittest.mock import MagicMock, patch
+from aircox_streamer.connector import Connector
+
+
+class TestConnector:
+
+    @pytest.mark.django_db
+    def test_is_open_unix(self, connector_unix):
+        mock_socket = MagicMock()    
+        with patch('aircox_streamer.connector.socket', mock_socket):
+            connector_unix.open()
+            assert connector_unix.is_open
+
+    @pytest.mark.django_db
+    def test_is_open_tcpip(self, connector_inet):
+        mock_socket = MagicMock()    
+        with patch('aircox_streamer.connector.socket', mock_socket):
+            connector_inet.open()
+            assert connector_inet.is_open
+
+    @pytest.mark.django_db
+    def test_init_with_unix_address(self, connector_unix):
+        assert connector_unix.address == "path/to/socket"
+    
+    @pytest.mark.django_db
+    def test_init_with_tcpip_address(self, connector_inet):
+        assert connector_inet.address == ('0.0.0.0', 6767)
+
+    @pytest.mark.django_db
+    def test_init_without_address(self, connector):
+        connector = Connector()
+        assert connector.address is None
+
+    #TODO Check if the mock_socket was define with the right arguments (socket.socket(family, socket.SOCK_STREAM))
+    @pytest.mark.django_db
+    def test_open_unix(self, connector_unix):
+        mock_socket = MagicMock()
+        with patch('aircox_streamer.connector.socket', mock_socket):
+            connector_unix.open()
+            connector_unix.socket.connect.assert_called_with("path/to/socket")
+
+    @pytest.mark.django_db
+    def test_open_tcpip(self, connector_inet):
+        mock_socket = MagicMock()
+        with patch('aircox_streamer.connector.socket', mock_socket):
+            connector_inet.open()
+            connector_inet.socket.connect.assert_called_with(('0.0.0.0', 6767))
+    
+    @pytest.mark.django_db
+    def test_open_with_exeption(self, connector):
+        connector.address = 123
+        assert connector.open() == -1
+    
+    @pytest.mark.django_db
+    def test_open_unix_when_isopen(self, connector_unix):
+        connector_unix.is_open == True
+        mock_socket = MagicMock()    
+        with patch('aircox_streamer.connector.socket', mock_socket):
+            assert connector_unix.open() == None
+    
+    @pytest.mark.django_db
+    def test_open_tcpip_when_isopen(self, connector_inet):
+        connector_inet.is_open == True
+        mock_socket = MagicMock()    
+        with patch('aircox_streamer.connector.socket', mock_socket):
+            assert connector_inet.open() == None
+
+    @pytest.mark.django_db
+    def test_close_unix(self, connector_unix):
+        mock_socket = MagicMock()    
+        with patch('aircox_streamer.connector.socket', mock_socket):
+            connector_unix.open()
+            connector_unix.close()
+            assert connector_unix.socket == None
+
+    @pytest.mark.django_db
+    def test_close_tcpip(self, connector_inet):
+        mock_socket = MagicMock()    
+        with patch('aircox_streamer.connector.socket', mock_socket):
+            connector_inet.open()
+            connector_inet.close()
+            assert connector_inet.socket == None
+
+    #issue, don't succeed to fake a response. Pytest always fail by being stuck at the while loop.
+    @pytest.mark.django_db
+    def test_send_tcpip(self, connector_inet):
+        pass
+    
+    @pytest.mark.django_db
+    def test_parse(self, connector):
+        input_data = 'key1=value1\nkey2=value2\nkey3=value3'
+        expected_output = {
+            'key1': 'value1',
+            'key2': 'value2',
+            'key3': 'value3'
+        }
+        assert connector.parse(input_data) == expected_output
+
+    @pytest.mark.django_db
+    def test_parse_json(self, connector):
+        input_data = '{"key1": "value1", "key2": "value2", "key3": "value3"}'
+        expected_output = {
+            'key1': 'value1',
+            'key2': 'value2',
+            'key3': 'value3'
+        }
+        assert connector.parse_json(input_data) == expected_output
+
+    
+
+
+    
+            
+        
+