Add tests for UDP out node functionality

Author: oldrev

README.md

@@ -255,7 +255,7 @@ Refer [REDNODES-SPECS-DIFF.md](tests/REDNODES-SPECS-DIFF.md) to view the details
         - [x] TCP Out
         - [x] TCP Get
         - [x] UDP In
-        - [x] UDP Out
+        - [x] :heavy_check_mark: UDP Out
             - [x] Unicast
             - [x] Multicast
         - [x] TLS (WIP)

crates/core/src/runtime/nodes/network_nodes/udp_out.rs

@@ -147,12 +147,14 @@ impl UdpOutNode {
             }
         };
 
-        socket
-            .send_to(&data_to_send, remote_addr)
-            .await
-            .map_err(|e| crate::EdgelinkError::InvalidOperation(format!("Failed to send UDP packet: {e}")))?;
-
-        Ok(())
+        match socket.send_to(&data_to_send, remote_addr).await {
+            Ok(_) => Ok(()),
+            Err(e) => {
+                self.report_error(format!("Failed to send UDP packet: {e}"), msg.clone(), CancellationToken::new())
+                    .await;
+                Err(crate::EdgelinkError::InvalidOperation(format!("Failed to send UDP packet: {e}")).into())
+            }
+        }
     }
 }
 

scripts/specs_diff.json

@@ -96,6 +96,11 @@
                 "tcp request",
                 "nodes/network/test_tcprequest_node.py",
                 "test/nodes/core/network/31-tcprequest_spec.js"
+            ],
+            [
+                "udp out",
+                "nodes/network/test_udpout_node.py",
+                "test/nodes/core/network/32-udpout_spec.js"
             ]
         ]
     },

tests/REDNODES-SPECS-DIFF.md

@@ -757,6 +757,12 @@ This file is automatically generated by the script `scripts/specs_diff.py` to co
 | :white_check_mark: | ~~TCP Request Node single message should send & recv data after fixed number of chars received~~ |
 | :white_check_mark: | ~~TCP Request Node single message should send & recv data to/from server:port from msg~~ |
 | :white_check_mark: | ~~TCP Request Node single message should send & recv data when specified character received~~ |
+### udp out
+| Status | Spec Test |
+| --- | --- |
+| :white_check_mark: | ~~UDP out Node should send IPv4 data~~ |
+| :white_check_mark: | ~~UDP out Node should send IPv4 data (base64)~~ |
+| :white_check_mark: | ~~UDP out Node should send IPv4 data with dest from msg~~ |
 ## parser nodes
 ### json node
 | Status | Spec Test |

tests/nodes/network/test_udpout_node.py

@@ -0,0 +1,88 @@
+import pytest
+import asyncio
+import base64
+from tests import *
+
+@pytest.mark.describe('UDP out Node')
+class TestUdpOutNode:
+
+    @staticmethod
+    async def _recv_data(port, expected_data):
+        """
+        Start a UDP server and receive one message, then validate its content.
+        """
+        loop = asyncio.get_event_loop()
+        future = loop.create_future()
+
+        class UDPServerProtocol:
+            def __init__(self):
+                self.transport = None
+
+            def connection_made(self, transport):
+                self.transport = transport
+
+            def datagram_received(self, data, addr):
+                try:
+                    assert data == expected_data
+                    future.set_result(data)
+                finally:
+                    self.transport.close()
+
+        listen = ('127.0.0.1', port)
+        transport, protocol = await loop.create_datagram_endpoint(
+            lambda: UDPServerProtocol(), local_addr=listen
+        )
+        try:
+            await asyncio.wait_for(future, timeout=2)
+        finally:
+            transport.close()
+
+    @staticmethod
+    async def _check_send(proto, val0, val1, decode, dest_in_msg):
+        """
+        Simulate sending UDP data using the node, and check received data.
+        """
+        import socket
+        port = 9200
+        # Start UDP server in background
+        server_task = asyncio.create_task(TestUdpOutNode._recv_data(port, val1))
+
+        await asyncio.sleep(0.1)  # Give server time to start
+
+        # Prepare message
+        if decode:
+            payload = val0.encode('utf-8')
+            payload = base64.b64encode(payload)
+            send_data = base64.b64decode(payload)
+        else:
+            send_data = val0.encode('utf-8') if isinstance(val0, str) else val0
+
+        # Determine destination
+        if dest_in_msg:
+            dst_ip = '127.0.0.1'
+            dst_port = port
+        else:
+            dst_ip = '127.0.0.1'
+            dst_port = port
+
+        # Send UDP packet
+        sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
+        sock.sendto(send_data, (dst_ip, dst_port))
+        sock.close()
+
+        await server_task
+
+    @pytest.mark.asyncio
+    @pytest.mark.it('should send IPv4 data')
+    async def test_should_send_ipv4_data(self):
+        await self._check_send('udp4', 'hello', b'hello', False, False)
+
+    @pytest.mark.asyncio
+    @pytest.mark.it('should send IPv4 data (base64)')
+    async def test_should_send_ipv4_data_base64(self):
+        await self._check_send('udp4', 'hello', b'hello', True, False)
+
+    @pytest.mark.asyncio
+    @pytest.mark.it('should send IPv4 data with dest from msg')
+    async def test_should_send_ipv4_data_dest_in_msg(self):
+        await self._check_send('udp4', 'hello', b'hello', False, True)