example: add socket scripts again

examples/protocols/sockets/tcp_client/pytest_tcp_client.py

@@ -3,71 +3,24 @@
 
 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,
                                  get_my_interface_by_dest_ip)
 from pytest_embedded import Dut
 
+try:
+    from run_tcp_server import TcpServer
+except ImportError:
+    import os
+    import sys
+    sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), '..', 'scripts')))
+    from run_tcp_server import TcpServer
+
+
 PORT = 3333
 
 
-class TcpServer(object):
-    def __init__(self, port, family_addr, persist=False):  # type: ignore
-        self.port = port
-        self.socket = socket.socket(family_addr, socket.SOCK_STREAM)
-        self.socket.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
-        self.socket.settimeout(60.0)
-        self.shutdown = Event()
-        self.persist = persist
-        self.family_addr = family_addr
-
-    def __enter__(self):  # type: ignore
-        try:
-            self.socket.bind(('', self.port))
-        except socket.error as e:
-            print('Bind failed:{}'.format(e))
-            raise
-        self.socket.listen(1)
-
-        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) -> None:  # type: ignore
-        if self.persist:
-            sock = socket.socket(self.family_addr, socket.SOCK_STREAM)
-            sock.connect(('localhost', self.port))
-            sock.sendall(b'Stop', )
-            sock.close()
-            self.shutdown.set()
-        self.shutdown.set()
-        self.server_thread.join()
-        self.socket.close()
-
-    def run_server(self) -> None:
-        while not self.shutdown.is_set():
-            try:
-                conn, address = self.socket.accept()  # accept new connection
-                print('Connection from: {}'.format(address))
-                conn.setblocking(1)
-                data = conn.recv(1024)
-                if not data:
-                    return
-                data = data.decode()
-                print('Received data: ' + data)
-                reply = 'OK: ' + data
-                conn.send(reply.encode())
-                conn.close()
-            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_tcp_client_ipv4(dut: Dut) -> None: