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, 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();
}