update

Author: Tom Softreck

pyproject.toml

@@ -44,20 +44,49 @@ numpy = ">=1.21.0,<2.0.0"
 python-nmap = ">=0.7.1,<1.0.0"
 
 [tool.poetry.group.dev.dependencies]
+# Code formatting
 black = "^23.0.0"
+isort = "^5.12.0"
+
+# Linting
 flake8 = "^6.0.0"
 flake8-bugbear = "^23.7.0"
-isort = "^5.12.0"
-mypy = "^1.0.0"
-pre-commit = "^3.0.0"
-
-[tool.poetry.group.test.dependencies]
-pytest = "^7.0.0"
-pytest-asyncio = "^0.21.0"
-pytest-cov = "^4.0.0"
-pytest-mock = "^3.10.0"
-pytest-aiohttp = "^1.0.0"
-coverage = {extras = ["toml"], version = "^7.0.0"}
+flake8-annotations = "^3.0.1"
+flake8-bandit = "^4.1.1"
+flake8-docstrings = "^1.7.0"
+flake8-import-order = "^0.18.2"
+pep8-naming = "^0.13.3"
+
+# Type checking
+mypy = "^1.5.0"
+types-PyYAML = "^6.0.12"
+types-requests = "^2.31.0"
+types-python-dateutil = "^2.8.19"
+
+# Testing
+pytest = "^7.4.0"
+pytest-asyncio = "^0.21.1"
+pytest-cov = "^4.1.0"
+pytest-mock = "^3.11.1"
+pytest-aiohttp = "^1.0.5"
+pytest-xdist = "^3.3.1"
+coverage = {extras = ["toml"], version = "^7.3.0"}
+
+# Documentation
+sphinx = "^7.0.1"
+sphinx-rtd-theme = "^1.2.2"
+sphinx-autodoc-typehints = "^1.23.0"
+sphinx-copybutton = "^0.5.2"
+
+# Development tools
+pre-commit = "^3.3.3"
+invoke = "^2.1.2"
+bandit = "^1.7.5"
+safety = "^2.3.5"
+vulture = "^2.7"
+
+# Type stubs
+typing-extensions = "^4.6.3"
 
 [tool.poetry.scripts]
 dialogchain = "dialogchain.cli:main"

tests/integration/test_http_connector.py

@@ -34,61 +34,74 @@ async def handle_echo(request: web.Request) -> web.Response:
         return runner, f"http://localhost:{unused_tcp_port}"
 
     @pytest.mark.asyncio
-    async def test_http_connector_with_mock_server(self, mock_http_server: tuple[web.AppRunner, str]) -> None:
-        """Test HTTP connector with a mock HTTP server"""
+    async def test_http_connector_with_mock_server(
+        self, mock_http_server: tuple[web.AppRunner, str]
+    ) -> None:
+        """Test HTTP connector with a mock HTTP server.
+        
+        Args:
+            mock_http_server: Fixture that provides a mock HTTP server.
+        """
         runner, server_url = mock_http_server
-
+        
         # Start the mock server
         await runner.setup()
-        site = web.TCPSite(runner, "localhost", int(server_url.split(":")[-1]))
+        site = web.TCPSite(runner, 'localhost', int(server_url.split(':')[-1]))
         await site.start()
-
+        
         try:
             # Test the HTTP connector
             connector = HTTPDestination(f"{server_url}/echo")
             test_data = {"test": "data"}
-
+            
             # Mock the session to capture the request
-            with patch("aiohttp.ClientSession.post") as mock_post:
+            with patch('aiohttp.ClientSession.post') as mock_post:
                 mock_response = MagicMock()
                 mock_response.__aenter__.return_value.status = 200
-                mock_response.__aenter__.return_value.json.return_value = {
-                    "echo": test_data
-                }
+                mock_response.__aenter__.return_value.json.return_value = {"echo": test_data}
                 mock_post.return_value = mock_response
-
+                
                 await connector.send(test_data)
-
+                
                 # Verify the request was made correctly
-                mock_post.assert_called_once_with(f"{server_url}/echo", json=test_data)
-
+                mock_post.assert_called_once_with(
+                    f"{server_url}/echo",
+                    json=test_data
+                )
+                
         finally:
             # Clean up the server
             await runner.cleanup()
-
+    
     @pytest.mark.asyncio
-    async def test_http_connector_with_real_request(self, mock_http_server):
-        """Test HTTP connector with a real request to mock server"""
+    async def test_http_connector_with_real_request(
+        self, mock_http_server: tuple[web.AppRunner, str]
+    ) -> None:
+        """Test HTTP connector with a real request to mock server.
+        
+        Args:
+            mock_http_server: Fixture that provides a mock HTTP server.
+        """
         runner, server_url = mock_http_server
-
+        
         # Start the mock server
         await runner.setup()
-        site = web.TCPSite(runner, "localhost", int(server_url.split(":")[-1]))
+        site = web.TCPSite(runner, 'localhost', int(server_url.split(':')[-1]))
         await site.start()
-
+        
         try:
             # Test the HTTP connector with a real request
             connector = HTTPDestination(f"{server_url}/echo")
             test_data = {"message": "Hello, World!"}
-
+            
             # This will make a real HTTP request to our mock server
             async with aiohttp.ClientSession() as session:
-                with patch("aiohttp.ClientSession", return_value=session):
+                with patch('aiohttp.ClientSession', return_value=session):
                     await connector.send(test_data)
-
+                    
                     # Verify the request was made by checking the server logs
                     # or other side effects if needed
-
+                    
         finally:
             # Clean up the server
             await runner.cleanup()