#include #include "embeddings.hpp" #include "errors.hpp" #include #include 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(); }