test: add test to check wifi conneciton between sta and softap

components/esp32/test/test_wifi.c

@@ -1,6 +1,7 @@
 /*
  Tests for the Wi-Fi
 */
+#include "string.h"
 #include "esp_system.h"
 #include "unity.h"
 #include "esp_system.h"
@@ -11,12 +12,22 @@
 #include "nvs_flash.h"
 #include "test_utils.h"
 #include "freertos/task.h"
+#include "freertos/event_groups.h"
 
 static const char* TAG = "test_wifi";
 
 #define DEFAULT_SSID "TEST_SSID"
 #define DEFAULT_PWD "TEST_PASS"
 
+#define GOT_IP_EVENT        0x00000001
+#define DISCONNECT_EVENT    0x00000002
+
+#define EVENT_HANDLER_FLAG_DO_NOT_AUTO_RECONNECT 0x00000001
+
+static uint32_t wifi_event_handler_flag;
+
+static EventGroupHandle_t wifi_events;
+
 static esp_err_t event_handler(void *ctx, system_event_t *event)
 {
     printf("ev_handle_called.\n");
@@ -30,10 +41,18 @@ static esp_err_t event_handler(void *ctx, system_event_t *event)
             ESP_LOGI(TAG, "SYSTEM_EVENT_STA_GOT_IP");
             ESP_LOGI(TAG, "got ip:%s\n",
             ip4addr_ntoa(&event->event_info.got_ip.ip_info.ip));
+            if (wifi_events) {
+                xEventGroupSetBits(wifi_events, GOT_IP_EVENT);
+            }
             break;
         case SYSTEM_EVENT_STA_DISCONNECTED:
             ESP_LOGI(TAG, "SYSTEM_EVENT_STA_DISCONNECTED");
+            if (! (EVENT_HANDLER_FLAG_DO_NOT_AUTO_RECONNECT & wifi_event_handler_flag) ) {
                 TEST_ESP_OK(esp_wifi_connect());
+            }
+            if (wifi_events) {
+                xEventGroupSetBits(wifi_events, DISCONNECT_EVENT);
+            }
             break;
         default:
             break;
@@ -129,8 +148,8 @@ static void start_wifi_as_softap(void)
     cfg.nvs_enable = false;
 
     wifi_config_t w_config = {
-        .ap.ssid = "default_ssid",
+        .ap.ssid = DEFAULT_SSID,
-        .ap.password = "default_password",
+        .ap.password = DEFAULT_PWD,
         .ap.ssid_len = 0,
         .ap.channel = 1,
         .ap.authmode = WIFI_AUTH_WPA2_PSK,
@@ -139,21 +158,60 @@ static void start_wifi_as_softap(void)
         .ap.beacon_interval = 100,
     };
 
+    TEST_ESP_OK(esp_event_loop_init(event_handler, NULL));
+
+    // can't deinit event loop, need to reset leak check
+    unity_reset_leak_checks();
+
+    if (wifi_events == NULL) {
+        wifi_events = xEventGroupCreate();
+    }
+
     TEST_ESP_OK(esp_wifi_init(&cfg));
     TEST_ESP_OK(esp_wifi_set_mode(WIFI_MODE_AP));
     TEST_ESP_OK(esp_wifi_set_config(WIFI_IF_AP, &w_config));
     TEST_ESP_OK(esp_wifi_start());
+}
+
+static void start_wifi_as_sta(void)
+{
+    wifi_init_config_t cfg = WIFI_INIT_CONFIG_DEFAULT();
+    cfg.nvs_enable = false;
+
+    // do not auto connect
+    wifi_event_handler_flag |= EVENT_HANDLER_FLAG_DO_NOT_AUTO_RECONNECT;
+    TEST_ESP_OK(esp_event_loop_init(event_handler, NULL));
+
+    // can't deinit event loop, need to reset leak check
+    unity_reset_leak_checks();
+
+    if (wifi_events == NULL) {
+        wifi_events = xEventGroupCreate();
+    } else {
+        xEventGroupClearBits(wifi_events, 0x00ffffff);
+    }
+
+    TEST_ESP_OK(esp_wifi_init(&cfg));
+    TEST_ESP_OK(esp_wifi_set_mode(WIFI_MODE_STA));
+    TEST_ESP_OK(esp_wifi_start());
 
 }
 
 static void stop_wifi(void)
 {
+    printf("stop wifi\n");
     TEST_ESP_OK(esp_wifi_stop());
     TEST_ESP_OK(esp_wifi_deinit());
+    if (wifi_events) {
+        vEventGroupDelete(wifi_events);
+        wifi_events = NULL;
+    }
+    vTaskDelay(1000/portTICK_PERIOD_MS);
 }
 
 static void receive_ds2ds_packet(void)
 {
+    test_case_uses_tcpip();
     start_wifi_as_softap();
     unity_wait_for_signal("sender ready");
     unity_send_signal("receiver ready");
@@ -161,7 +219,6 @@ static void receive_ds2ds_packet(void)
     // wait for sender to send packets
     vTaskDelay(1000/portTICK_PERIOD_MS);
     stop_wifi();
-
 }
 
 static const char ds2ds_pdu[] = {
@@ -174,6 +231,7 @@ static const char ds2ds_pdu[] = {
 
 static void send_ds2ds_packet(void)
 {
+    test_case_uses_tcpip();
     start_wifi_as_softap();
     unity_send_signal("sender ready");
     unity_wait_for_signal("receiver ready");
@@ -187,3 +245,70 @@ static void send_ds2ds_packet(void)
 }
 
 TEST_CASE_MULTIPLE_DEVICES("receive ds2ds packet without exception", "[wifi][test_env=UT_T2_1]", receive_ds2ds_packet, send_ds2ds_packet);
+
+static void wifi_connect_by_bssid(uint8_t *bssid)
+{
+    EventBits_t bits;
+
+    wifi_config_t w_config = {
+        .sta.ssid = DEFAULT_SSID,
+        .sta.password = DEFAULT_PWD,
+        .sta.bssid_set = true,
+    };
+
+    memcpy(w_config.sta.bssid, bssid, 6);
+
+    TEST_ESP_OK(esp_wifi_set_config(WIFI_IF_STA, &w_config));
+    TEST_ESP_OK(esp_wifi_connect());
+    bits = xEventGroupWaitBits(wifi_events, GOT_IP_EVENT, 1, 0, 5000/portTICK_RATE_MS);
+    TEST_ASSERT(bits == GOT_IP_EVENT);
+}
+
+static void test_wifi_connection_sta(void)
+{
+    char mac_str[19];
+    uint8_t mac[6];
+    EventBits_t bits;
+
+    test_case_uses_tcpip();
+
+    start_wifi_as_sta();
+
+    unity_wait_for_signal_param("SoftAP mac", mac_str, 19);
+
+    TEST_ASSERT_TRUE(unity_util_convert_mac_from_string(mac_str, mac));
+
+    wifi_connect_by_bssid(mac);
+
+    unity_send_signal("STA connected");
+
+    bits = xEventGroupWaitBits(wifi_events, DISCONNECT_EVENT, 1, 0, 60000 / portTICK_RATE_MS);
+    // disconnect event not triggered
+    printf("wait finish\n");
+    TEST_ASSERT(bits == 0);
+
+    stop_wifi();
+}
+
+static void test_wifi_connection_softap(void)
+{
+    char mac_str[19] = {0};
+    uint8_t mac[6];
+
+    test_case_uses_tcpip();
+
+    start_wifi_as_softap();
+
+    TEST_ESP_OK(esp_wifi_get_mac(ESP_IF_WIFI_AP, mac));
+    sprintf(mac_str, MACSTR, MAC2STR(mac));
+
+    unity_send_signal_param("SoftAP mac", mac_str);
+
+    unity_wait_for_signal("STA connected");
+
+    vTaskDelay(60000 / portTICK_PERIOD_MS);
+
+    stop_wifi();
+}
+
+TEST_CASE_MULTIPLE_DEVICES("test wifi retain connection for 60s", "[wifi][test_env=UT_T2_1][timeout=90]", test_wifi_connection_sta, test_wifi_connection_softap);