1 files changed, 128 insertions, 0 deletions

diff --git a/tests/unit/test_display.c b/tests/unit/test_display.c
new file mode 100644
index 0000000..81f67a7
--- /dev/null
+++ b/tests/unit/test_display.c
@@ -0,0 +1,128 @@
+#include "test_framework.h"
+#include <string.h>
+#include <stdio.h>
+
+static int escape_wifi_field(const char *src, char *dst, int dst_size) {
+    int si = 0, di = 0;
+    while (src[si] && di < dst_size - 2) {
+        char c = src[si];
+        if (c == '\\' || c == ';' || c == ':' || c == ',' || c == '"') {
+            if (di + 2 >= dst_size) break;
+            dst[di++] = '\\';
+            dst[di++] = c;
+        } else {
+            dst[di++] = c;
+        }
+        si++;
+    }
+    dst[di] = '\0';
+    return di;
+}
+
+static int test_escape_no_special(void) {
+    char dst[64];
+    int len = escape_wifi_field("HelloWorld", dst, sizeof(dst));
+    ASSERT(strcmp(dst, "HelloWorld") == 0, "no special chars unchanged");
+    ASSERT(len == 10, "no special chars length correct");
+    return 0;
+}
+
+static int test_escape_semicolon(void) {
+    char dst[64];
+    int len = escape_wifi_field("Hello;World", dst, sizeof(dst));
+    ASSERT(strcmp(dst, "Hello\\;World") == 0, "semicolon escaped");
+    ASSERT(len == 12, "semicolon escaped length correct");
+    return 0;
+}
+
+static int test_escape_colon(void) {
+    char dst[64];
+    int len = escape_wifi_field("Hello:World", dst, sizeof(dst));
+    ASSERT(strcmp(dst, "Hello\\:World") == 0, "colon escaped");
+    ASSERT(len == 12, "colon escaped length correct");
+    return 0;
+}
+
+static int test_escape_backslash(void) {
+    char dst[64];
+    int len = escape_wifi_field("Hello\\World", dst, sizeof(dst));
+    ASSERT(strcmp(dst, "Hello\\\\World") == 0, "backslash escaped");
+    ASSERT(len == 12, "backslash escaped length correct");
+    return 0;
+}
+
+static int test_escape_comma(void) {
+    char dst[64];
+    int len = escape_wifi_field("Hello,World", dst, sizeof(dst));
+    ASSERT(strcmp(dst, "Hello\\,World") == 0, "comma escaped");
+    ASSERT(len == 12, "comma escaped length correct");
+    return 0;
+}
+
+static int test_escape_quote(void) {
+    char dst[64];
+    int len = escape_wifi_field("Hello\"World", dst, sizeof(dst));
+    ASSERT(strcmp(dst, "Hello\\\"World") == 0, "quote escaped");
+    ASSERT(len == 12, "quote escaped length correct");
+    return 0;
+}
+
+static int test_escape_multiple(void) {
+    char dst[64];
+    int len = escape_wifi_field("a;b:c\\d", dst, sizeof(dst));
+    ASSERT(strcmp(dst, "a\\;b\\:c\\\\d") == 0, "multiple special chars escaped");
+    ASSERT(len == 10, "multiple special chars length correct");
+    return 0;
+}
+
+static int test_escape_empty(void) {
+    char dst[64];
+    int len = escape_wifi_field("", dst, sizeof(dst));
+    ASSERT(strcmp(dst, "") == 0, "empty string stays empty");
+    ASSERT(len == 0, "empty string length is 0");
+    return 0;
+}
+
+static int test_escape_overflow(void) {
+    char dst[5];
+    int len = escape_wifi_field("Hello;World", dst, sizeof(dst));
+    ASSERT(len < (int)sizeof(dst), "output truncated on overflow");
+    ASSERT(dst[len] == '\0', "still null-terminated after truncation");
+    return 0;
+}
+
+static int test_escape_ssid_like(void) {
+    char dst[64];
+    int len = escape_wifi_field("TollGate-C0E9CA", dst, sizeof(dst));
+    ASSERT(strcmp(dst, "TollGate-C0E9CA") == 0, "TollGate SSID no escaping needed");
+    ASSERT(len == 15, "TollGate SSID length correct");
+    return 0;
+}
+
+static int test_escape_all_special_in_one(void) {
+    char dst[64];
+    int len = escape_wifi_field("\\;:,\"", dst, sizeof(dst));
+    ASSERT(strcmp(dst, "\\\\\\;\\:\\,\\\"") == 0, "all special chars in sequence");
+    ASSERT(len == 10, "all special chars length correct");
+    return 0;
+}
+
+int main(void) {
+    int failed = 0;
+    failed += test_escape_no_special();
+    failed += test_escape_semicolon();
+    failed += test_escape_colon();
+    failed += test_escape_backslash();
+    failed += test_escape_comma();
+    failed += test_escape_quote();
+    failed += test_escape_multiple();
+    failed += test_escape_empty();
+    failed += test_escape_overflow();
+    failed += test_escape_ssid_like();
+    failed += test_escape_all_special_in_one();
+
+    if (failed == 0) {
+        printf("\n=== ALL DISPLAY TESTS PASSED ===\n");
+    }
+    return failed;
+}