example: add socket scripts again

examples/protocols/sockets/udp_client/pytest_udp_client.py

@@ -3,7 +3,6 @@
 
 import logging
 import socket
-from threading import Event, Thread
 
 import pytest
 from common_test_methods import (get_env_config_variable, get_host_ip4_by_dest_ip, get_host_ip6_by_dest_ip,
@@ -11,60 +10,19 @@ from common_test_methods import (get_env_config_variable, get_host_ip4_by_dest_i
 from pexpect.exceptions import TIMEOUT
 from pytest_embedded import Dut
 
+try:
+    from run_udp_server import UdpServer
+except ImportError:
+    import os
+    import sys
+    sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), '..', 'scripts')))
+    from run_udp_server import UdpServer
+
+
 PORT = 3333
 MAX_RETRIES = 3
 
 
-class UdpServer:
-
-    def __init__(self, port, family_addr, persist=False):  # type: ignore
-        self.port = port
-        self.family_addr = family_addr
-        self.socket = socket.socket(family_addr, socket.SOCK_DGRAM)
-        self.socket.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
-        self.socket.settimeout(60.0)
-        self.shutdown = Event()
-        self.persist = persist
-
-    def __enter__(self):  # type: ignore
-        try:
-            self.socket.bind(('', self.port))
-        except socket.error as e:
-            print('Bind failed:{}'.format(e))
-            raise
-
-        print('Starting server on port={} family_addr={}'.format(self.port, self.family_addr))
-        self.server_thread = Thread(target=self.run_server)
-        self.server_thread.start()
-        return self
-
-    def __exit__(self, exc_type, exc_value, traceback):  # type: ignore
-        if self.persist:
-            sock = socket.socket(self.family_addr, socket.SOCK_DGRAM)
-            sock.sendto(b'Stop', ('localhost', self.port))
-            sock.close()
-            self.shutdown.set()
-        self.server_thread.join()
-        self.socket.close()
-
-    def run_server(self) -> None:
-        while not self.shutdown.is_set():
-            try:
-                data, addr = self.socket.recvfrom(1024)
-                print(addr)
-                if not data:
-                    return
-                data = data.decode()
-                print('Reply[' + addr[0] + ':' + str(addr[1]) + '] - ' + data)
-                reply = 'OK: ' + data
-                self.socket.sendto(reply.encode(), addr)
-            except socket.error as e:
-                print('Running server failed:{}'.format(e))
-                raise
-            if not self.persist:
-                break
-
-
 @pytest.mark.esp32
 @pytest.mark.wifi_router
 def test_examples_udp_client_ipv4(dut: Dut) -> None: