Working on Actions and Testing

Author: YeisonCardona

.github/workflows/python-package.yml

@@ -35,12 +35,12 @@ jobs:
         flake8 . --count --select=E9,F63,F7,F82 --show-source --statistics
         # exit-zero treats all errors as warnings. The GitHub editor is 127 chars wide
         flake8 . --count --exit-zero --max-complexity=10 --max-line-length=127 --statistics
-    - name: Test connection IPv4
-      run: |
-        pytest test/test_connection.py::Test_Connections_for_IPv4 -v
-    - name: Test connection IPv6
-      run: |
-        pytest test/test_connection.py::Test_Connections_for_IPv6 -v
+#    - name: Test connection IPv4
+#      run: |
+#        pytest test/test_connection.py::Test_Connections_for_IPv4 -v
+#    - name: Test connection IPv6
+#      run: |
+#        pytest test/test_connection.py::Test_Connections_for_IPv6 -v
     - name: Test discovery
       run: |
         pytest test/test_discovery.py -v

chaski/node.py

@@ -356,6 +356,7 @@ class UDPProtocol(asyncio.DatagramProtocol):
 
     node: "ChaskiNode"
     on_message_received: Awaitable
+    _background_tasks = set()
 
     def datagram_received(self, message: bytes, addr: tuple[str, int]) -> None:
         """
@@ -405,6 +406,12 @@ def connection_lost(self, exc: Optional[Exception]) -> None:
         """
         logger_udp.info(f"UDP connection closed: {exc}")
 
+    def track_task(self, coro):
+        task = asyncio.create_task(coro)
+        self._background_tasks.add(task)
+        task.add_done_callback(self._background_tasks.discard)
+        return task
+
 
 class ChaskiNode:
     """

test/test_discovery.py

@@ -308,12 +308,12 @@ async def test_multiple_discovery(self):
         self.assertEqual(len(self.nodes[5].edges), 1, f"Node 5 failed before discovery")
         self.assertEqual(len(self.nodes[6].edges), 1, f"Node 6 failed before discovery")
 
-        await self.nodes[1].discovery(on_pair="none", timeout=1)
-        await self.nodes[2].discovery(on_pair="none", timeout=1)
-        await self.nodes[3].discovery(on_pair="none", timeout=1)
-        await self.nodes[4].discovery(on_pair="none", timeout=1)
-        await self.nodes[5].discovery(on_pair="none", timeout=1)
-        await self.nodes[6].discovery(on_pair="none", timeout=1)
+        await self.nodes[1].discovery(on_pair="none")
+        await self.nodes[2].discovery(on_pair="none")
+        await self.nodes[3].discovery(on_pair="none")
+        await self.nodes[4].discovery(on_pair="none")
+        await self.nodes[5].discovery(on_pair="none")
+        await self.nodes[6].discovery(on_pair="none")
 
         await self._wait_for_connections()