Merge remote-tracking branch 'esp8266/master'

Author: Me No Dev

tests/device/libraries/BSTest/runner.py

@@ -42,15 +42,18 @@ def __init__(self, spawn_obj, name, mocks):
 
     def get_test_list(self):
         self.sp.sendline('-1')
-        timeout = 10
+        self.tests = []
+        timeout = 100
         while timeout > 0:
             res = self.sp.expect(['>>>>>bs_test_menu_begin', EOF, TIMEOUT])
             if res == 0:
                 break
             timeout-=1
             time.sleep(0.1)
+        if timeout <= 0:
+            debug_print('begin timeout')
+            return
         debug_print('got begin')
-        self.tests = []
         while True:
             res = self.sp.expect(['>>>>>bs_test_item id\=(\d+) name\="([^\"]*?)" desc="([^"]*?)"',
                                   '>>>>>bs_test_menu_end',

tests/device/test_WiFiServer/test_WiFiServer.ino

@@ -0,0 +1,54 @@
+#include <Arduino.h>
+#include <ESP8266WiFi.h>
+#include <ESP8266mDNS.h>
+#include <WiFiClient.h>
+#include <BSTest.h>
+#include <test_config.h>
+
+
+BS_ENV_DECLARE();
+
+void setup()
+{
+    Serial.begin(115200);
+    WiFi.persistent(false);
+    WiFi.begin(STA_SSID, STA_PASS);
+    while (WiFi.status() != WL_CONNECTED) {
+        delay(500);
+    }
+    MDNS.begin("esp8266-wfs-test");
+    BS_RUN(Serial);
+}
+
+TEST_CASE("Simple echo server", "[WiFiServer]")
+{
+    const uint32_t timeout = 10000;
+    const uint16_t port = 5000;
+    const int maxRequests = 5;
+    const int minRequestLength = 128;
+    WiFiServer server(port);
+    server.begin();
+    auto start = millis();
+    
+    int replyCount = 0;
+    while (millis() - start < timeout) {
+        delay(50);
+        WiFiClient client = server.available();
+        if (!client) {
+            continue;
+        }
+        String request = client.readStringUntil('\n');
+        CHECK(request.length() >= minRequestLength);
+        client.print(request);
+        client.print('\n');
+        if (++replyCount == maxRequests) {
+            break;
+        }
+    }
+    CHECK(replyCount == maxRequests);
+}
+
+void loop()
+{
+}
+

tests/device/test_WiFiServer/test_WiFiServer.py

@@ -0,0 +1,42 @@
+from mock_decorators import setup, teardown
+from threading import Thread
+import socket
+import time
+
+stop_client_thread = False
+client_thread = None
+
+@setup('Simple echo server')
+def setup_echo_server(e):
+    global stop_client_thread
+    global client_thread
+    def echo_client_thread():
+        server_address = socket.gethostbyname('esp8266-wfs-test.local')
+        count = 0
+        while count < 5 and not stop_client_thread:
+            sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
+            sock.connect((server_address, 5000))
+            sock.settimeout(1.0)
+            buf = 'a' * 1023 + '\n'
+            sock.sendall(buf)
+            data = ''
+            retries = 0
+            while len(data) < 1024 and retries < 3:
+                data += sock.recv(1024)
+                retries += 1
+            print 'Received {} bytes'.format(len(data))
+            if len(data) != 1024:
+                raise RuntimeError('client failed to receive response')
+            count += 1
+        
+    stop_client_thread = False
+    client_thread = Thread(target=echo_client_thread)
+    client_thread.start()
+
+@teardown('Simple echo server')
+def teardown_echo_server(e):
+    global stop_client_thread
+    stop_client_thread = True
+    client_thread.join()
+
+