dstalk/tests/service_registry_test.cpp

// ============================================================================
// service_registry_test.cpp — ServiceRegistry 单元测试（补充覆盖，不与 host_api_test 重叠）
// ============================================================================
// host_api_test 已覆盖: 重复注册(同名同版/同名异版)、查询不存在服务、版本不满足、
//                      shutdown 后查询。本测试补充边界与生命周期路径。
// ============================================================================

#include <cstring>
#include <iostream>

#include "service_registry.hpp"

// ---- 轻量断言 ----
static int g_failures = 0;
#define TCHECK(cond, msg) do {                            \
    if (cond) {                                           \
        std::cout << "[OK] " << (msg) << "\n";            \
    } else {                                              \
        std::cerr << "[FAIL] " << (msg) << "\n";          \
        g_failures++;                                     \
    }                                                     \
} while (0)

// ============================================================
int main()
{
    std::cout << "=== dstalk service_registry unit tests (supplement) ===\n\n";

    // ====================================================================
    // Test 1: register_service(nullptr name) → -1
    // ====================================================================
    {
        dstalk::ServiceRegistry reg;
        void* vt = reinterpret_cast<void*>(0x10);
        int r = reg.register_service(nullptr, 1, vt);
        TCHECK(r == -1, "register_service(nullptr name) returns -1");
    }

    // ====================================================================
    // Test 2: register_service(nullptr vtable) → -1
    // ====================================================================
    {
        dstalk::ServiceRegistry reg;
        int r = reg.register_service("valid_name", 1, nullptr);
        TCHECK(r == -1, "register_service(nullptr vtable) returns -1");
    }

    // ====================================================================
    // Test 3: 完整生命周期 — register → query → unregister → query(nullptr)
    // ====================================================================
    {
        dstalk::ServiceRegistry reg;
        void* vt = reinterpret_cast<void*>(0xDEAD);

        int r = reg.register_service("life", 3, vt);
        TCHECK(r == 0, "register_service(\"life\",3) returns 0");

        void* q1 = reg.query_service("life", 2);
        TCHECK(q1 == vt, "query_service(\"life\",2) returns vtable after register");

        reg.unregister_service("life");

        void* q2 = reg.query_service("life", 1);
        TCHECK(q2 == nullptr, "query_service(\"life\",1) returns nullptr after unregister");
    }

    // ====================================================================
    // Test 4: unregister_service(nullptr name) 不崩溃（安全空操作）
    // ====================================================================
    {
        dstalk::ServiceRegistry reg;
        reg.unregister_service(nullptr);
        std::cout << "[OK] unregister_service(nullptr) did not crash\n";
    }

    // ====================================================================
    // Test 5: 注册后重新注册同名 → 先 unregister 再 register 成功
    // ====================================================================
    {
        dstalk::ServiceRegistry reg;
        void* vt1 = reinterpret_cast<void*>(0xA);
        void* vt2 = reinterpret_cast<void*>(0xB);

        reg.register_service("reborn", 1, vt1);
        reg.unregister_service("reborn");

        int r = reg.register_service("reborn", 2, vt2);
        TCHECK(r == 0, "register_service after unregister same name returns 0");

        void* q = reg.query_service("reborn", 2);
        TCHECK(q == vt2, "query_service after re-register returns new vtable");
    }

    // ====================================================================
    // Test 6: query_service(nullptr name) → nullptr
    // ====================================================================
    {
        dstalk::ServiceRegistry reg;
        void* q = reg.query_service(nullptr, 1);
        TCHECK(q == nullptr, "query_service(nullptr name) returns nullptr");
    }

    // ====================================================================
    // 结果
    // ====================================================================
    std::cout << "\n";
    if (g_failures == 0) {
        std::cout << "=== All service_registry tests passed ===\n";
        return 0;
    } else {
        std::cerr << "=== " << g_failures << " service_registry test(s) FAILED ===\n";
        return 1;
    }
}