1 files changed, 125 insertions, 0 deletions

diff --git a/tests/unit/test_stratum_client.c b/tests/unit/test_stratum_client.c
new file mode 100644
index 0000000..302fcba
--- /dev/null
+++ b/tests/unit/test_stratum_client.c
@@ -0,0 +1,125 @@
+#include "test_framework.h"
+#include "../../components/tollgate_core/src/tollgate_core_stratum_client.h"
+#include <cjson/cJSON.h>
+#include <string.h>
+#include <stdio.h>
+#include <stdlib.h>
+
+int main(void)
+{
+    printf("=== test_stratum_client ===\n");
+
+    printf("\n--- hex_to_bytes ---\n");
+    {
+        const char *hex = "0a1b2c3d";
+        uint8_t out[4] = {0};
+        tollgate_core_stratum_hex_to_bytes(hex, out, 4);
+        ASSERT_EQ_INT(0x0a, (int)out[0], "byte 0");
+        ASSERT_EQ_INT(0x1b, (int)out[1], "byte 1");
+        ASSERT_EQ_INT(0x2c, (int)out[2], "byte 2");
+        ASSERT_EQ_INT(0x3d, (int)out[3], "byte 3");
+    }
+
+    printf("\n--- hex_to_bytes zeros ---\n");
+    {
+        const char *hex = "00000000";
+        uint8_t out[4] = {0xff, 0xff, 0xff, 0xff};
+        tollgate_core_stratum_hex_to_bytes(hex, out, 4);
+        ASSERT_EQ_INT(0, (int)out[0], "zero byte 0");
+        ASSERT_EQ_INT(0, (int)out[1], "zero byte 1");
+    }
+
+    printf("\n--- parse_notify valid ---\n");
+    {
+        cJSON *params = cJSON_Parse("[\"123\",\"0000000000000000000000000000000000000000000000000000000000000000\","
+                                     "\"\""
+                                     ",\"\""
+                                     ",\"\""
+                                     ",\"20000000\""
+                                     ",\"1705e3c4\""
+                                     ",\"65a3b2c1\"]");
+        tollgate_stratum_job_t job = {0};
+        uint32_t nbits = 0;
+        bool ok = tollgate_core_stratum_parse_notify(params, &job, &nbits);
+        ASSERT(ok, "parse_notify returns true");
+        ASSERT_EQ_INT(123, (int)job.job_id, "job_id parsed");
+        ASSERT(job.valid, "job is valid");
+        ASSERT_EQ_INT(0x20000000, (int)job.version, "version parsed");
+        ASSERT_EQ_INT(0x1705e3c4, (int)job.nbits, "nbits parsed");
+        ASSERT_EQ_INT(0x65a3b2c1, (int)job.ntime, "ntime parsed");
+        ASSERT_EQ_INT(0x1705e3c4, (int)nbits, "out_nbits set");
+        ASSERT_EQ_INT(32, job.target_len, "target_len is 32");
+        cJSON_Delete(params);
+    }
+
+    printf("\n--- parse_notify too few params ---\n");
+    {
+        cJSON *params = cJSON_Parse("[\"123\",\"abcd\"]");
+        tollgate_stratum_job_t job = {0};
+        bool ok = tollgate_core_stratum_parse_notify(params, &job, NULL);
+        ASSERT(!ok, "parse_notify returns false for insufficient params");
+        cJSON_Delete(params);
+    }
+
+    printf("\n--- parse_notify NULL ---\n");
+    {
+        tollgate_stratum_job_t job = {0};
+        bool ok = tollgate_core_stratum_parse_notify(NULL, &job, NULL);
+        ASSERT(!ok, "parse_notify returns false for NULL");
+    }
+
+    printf("\n--- parse_difficulty valid ---\n");
+    {
+        cJSON *params = cJSON_Parse("[512]");
+        uint64_t diff = 0;
+        bool ok = tollgate_core_stratum_parse_difficulty(params, &diff);
+        ASSERT(ok, "parse_difficulty returns true");
+        ASSERT_EQ_UINT64(512, (unsigned long long)diff, "difficulty parsed");
+        cJSON_Delete(params);
+    }
+
+    printf("\n--- parse_difficulty invalid ---\n");
+    {
+        cJSON *params = cJSON_Parse("[]");
+        uint64_t diff = 0;
+        bool ok = tollgate_core_stratum_parse_difficulty(params, &diff);
+        ASSERT(!ok, "parse_difficulty returns false for empty array");
+        cJSON_Delete(params);
+    }
+
+    printf("\n--- build_subscribe ---\n");
+    {
+        char buf[256];
+        int len = tollgate_core_stratum_build_subscribe(buf, sizeof(buf), 1);
+        ASSERT(len > 0, "build_subscribe returns positive length");
+        ASSERT(strstr(buf, "mining.subscribe") != NULL, "contains method");
+        ASSERT(strstr(buf, "\"id\":1") != NULL, "contains id");
+        ASSERT(strstr(buf, "TollGate/1.0") != NULL, "contains user agent");
+        ASSERT(buf[len - 1] == '\n', "ends with newline");
+    }
+
+    printf("\n--- build_authorize ---\n");
+    {
+        char buf[512];
+        int len = tollgate_core_stratum_build_authorize(buf, sizeof(buf), 2, "user1", "pass1");
+        ASSERT(len > 0, "build_authorize returns positive length");
+        ASSERT(strstr(buf, "mining.authorize") != NULL, "contains method");
+        ASSERT(strstr(buf, "user1") != NULL, "contains user");
+        ASSERT(strstr(buf, "pass1") != NULL, "contains pass");
+    }
+
+    printf("\n--- build_submit ---\n");
+    {
+        char buf[512];
+        int len = tollgate_core_stratum_build_submit(buf, sizeof(buf), 3, "worker1",
+                                                       100, 0x65a3b2c1, 0x12345678, 0x20000000);
+        ASSERT(len > 0, "build_submit returns positive length");
+        ASSERT(strstr(buf, "mining.submit") != NULL, "contains method");
+        ASSERT(strstr(buf, "worker1") != NULL, "contains worker");
+        ASSERT(strstr(buf, "100") != NULL, "contains job_id");
+        ASSERT(strstr(buf, "65a3b2c1") != NULL, "contains ntime");
+        ASSERT(strstr(buf, "12345678") != NULL, "contains nonce");
+    }
+
+    TEST_SUMMARY();
+}