smartbotic-vectorapi/tests/test_embeddings.cpp

#include <gtest/gtest.h>
#include "embeddings.hpp"
#include "errors.hpp"
#include <httplib.h>
#include <thread>
using namespace svapi;

TEST(Embeddings, ParsesWellFormedResponse) {
    auto v = parseEmbeddingResponse(nlohmann::json::parse(R"({"data":[{"embedding":[0.5,1.0,1.5]}]})"));
    ASSERT_EQ(v.size(), 3u); EXPECT_FLOAT_EQ(v[0], 0.5f); EXPECT_FLOAT_EQ(v[2], 1.5f);
}
TEST(Embeddings, MalformedThrows) {
    EXPECT_THROW(parseEmbeddingResponse(nlohmann::json::parse(R"({"data":[]})")), ApiError);
    EXPECT_THROW(parseEmbeddingResponse(nlohmann::json::object()), ApiError);
}
TEST(Embeddings, SplitBaseHandlesPathPrefix) {
    EXPECT_EQ(splitEmbeddingBase("https://api.openai.com").origin, "https://api.openai.com");
    EXPECT_EQ(splitEmbeddingBase("https://api.openai.com").path, "");
    EXPECT_EQ(splitEmbeddingBase("https://api.openai.com/").path, "");
    auto orr = splitEmbeddingBase("https://openrouter.ai/api");
    EXPECT_EQ(orr.origin, "https://openrouter.ai");
    EXPECT_EQ(orr.path, "/api");
    EXPECT_EQ(splitEmbeddingBase("https://openrouter.ai/api/").path, "/api");
    EXPECT_EQ(splitEmbeddingBase("http://h:8080/a/b").origin, "http://h:8080");
    EXPECT_EQ(splitEmbeddingBase("http://h:8080/a/b").path, "/a/b");
}
TEST(Embeddings, ClientHitsMockServer) {
    httplib::Server mock;
    mock.Post("/v1/embeddings", [](const httplib::Request& req, httplib::Response& res) {
        auto j = nlohmann::json::parse(req.body);
        EXPECT_EQ(j["model"], "m"); EXPECT_EQ(j["input"], "hello");
        res.set_content(R"({"data":[{"embedding":[1,2,3,4]}]})", "application/json");
    });
    int port = mock.bind_to_any_port("127.0.0.1");
    std::thread t([&]{ mock.listen_after_bind(); });
    EmbeddingClient cli("http://127.0.0.1:" + std::to_string(port), "k");
    auto v = cli.embed("m", "hello");
    EXPECT_EQ(v.size(), 4u); EXPECT_FLOAT_EQ(v[3], 4.0f);
    mock.stop(); t.join();
}
TEST(Embeddings, ClientHonorsBasePathPrefix) {
    // Provider whose embeddings live under a path prefix (e.g. OpenRouter /api/v1/embeddings)
    httplib::Server mock;
    mock.Post("/api/v1/embeddings", [](const httplib::Request&, httplib::Response& res) {
        res.set_content(R"({"data":[{"embedding":[9,9]}]})", "application/json");
    });
    int port = mock.bind_to_any_port("127.0.0.1");
    std::thread t([&]{ mock.listen_after_bind(); });
    EmbeddingClient cli("http://127.0.0.1:" + std::to_string(port) + "/api", "k");
    auto v = cli.embed("qwen/qwen3-embedding-8b", "hello");
    EXPECT_EQ(v.size(), 2u); EXPECT_FLOAT_EQ(v[0], 9.0f);
    mock.stop(); t.join();
}