@ekodb/ekodb-client 0.12.0 → 0.13.0

package/src/client.test.ts

@@ -85,7 +85,6 @@ describe("EkoDBClient configuration", () => {
     const client = new EkoDBClient({
       baseURL: "http://localhost:8080",
       apiKey: "test-key",
-      timeout: 60000,
       maxRetries: 5,
       shouldRetry: true,
       format: SerializationFormat.Json,
@@ -190,6 +189,80 @@ describe("EkoDBClient update", () => {
   });
 });
 
+// ============================================================================
+// Atomic Field Action Tests
+// ============================================================================
+
+describe("EkoDBClient updateWithAction", () => {
+  it("increments a field", async () => {
+    const client = createTestClient();
+
+    mockTokenResponse();
+    mockJsonResponse({ id: "rec_1", views: 42 });
+
+    const result = await client.updateWithAction(
+      "counters",
+      "rec_1",
+      "increment",
+      "views",
+      1,
+    );
+
+    expect(result).toHaveProperty("views", 42);
+  });
+
+  it("pushes to an array field", async () => {
+    const client = createTestClient();
+
+    mockTokenResponse();
+    mockJsonResponse({ id: "rec_2", tags: ["rust", "new-tag"] });
+
+    const result = await client.updateWithAction(
+      "lists",
+      "rec_2",
+      "push",
+      "tags",
+      "new-tag",
+    );
+
+    expect(result.tags).toContain("new-tag");
+  });
+
+  it("clears a field without value", async () => {
+    const client = createTestClient();
+
+    mockTokenResponse();
+    mockJsonResponse({ id: "rec_3", temp: 0 });
+
+    const result = await client.updateWithAction(
+      "data",
+      "rec_3",
+      "clear",
+      "temp",
+    );
+
+    expect(result).toHaveProperty("temp", 0);
+  });
+});
+
+describe("EkoDBClient updateWithActionSequence", () => {
+  it("applies multiple actions atomically", async () => {
+    const client = createTestClient();
+
+    mockTokenResponse();
+    mockJsonResponse({ id: "player_1", score: 110, lives: 2 });
+
+    const result = await client.updateWithActionSequence("game", "player_1", [
+      ["increment", "score", 10],
+      ["decrement", "lives", 1],
+      ["push", "log", "hit"],
+    ]);
+
+    expect(result).toHaveProperty("score", 110);
+    expect(result).toHaveProperty("lives", 2);
+  });
+});
+
 // ============================================================================
 // Delete Tests
 // ============================================================================
@@ -1243,3 +1316,270 @@ describe("EkoDBClient collection utilities", () => {
     expect(result).toBe(0);
   });
 });
+
+// ============================================================================
+// Distinct Values Tests
+// ============================================================================
+
+describe("EkoDBClient distinctValues", () => {
+  it("returns distinct values for a field", async () => {
+    const client = createTestClient();
+
+    mockTokenResponse();
+    mockJsonResponse({
+      collection: "products",
+      field: "category",
+      values: ["books", "electronics", "food"],
+      count: 3,
+    });
+
+    const result = await client.distinctValues("products", "category");
+
+    expect(result.collection).toBe("products");
+    expect(result.field).toBe("category");
+    expect(result.count).toBe(3);
+    expect(result.values).toHaveLength(3);
+    expect(result.values).toContain("books");
+  });
+
+  it("sends filter in request body", async () => {
+    const client = createTestClient();
+
+    mockTokenResponse();
+    mockJsonResponse({
+      collection: "orders",
+      field: "status",
+      values: ["active", "pending"],
+      count: 2,
+    });
+
+    const filter = {
+      type: "Condition",
+      content: { field: "region", operator: "Eq", value: "us" },
+    };
+    const result = await client.distinctValues("orders", "status", { filter });
+
+    expect(result.count).toBe(2);
+    expect(result.values).toContain("active");
+  });
+
+  it("returns empty values for collection with no matching records", async () => {
+    const client = createTestClient();
+
+    mockTokenResponse();
+    mockJsonResponse({
+      collection: "empty",
+      field: "tag",
+      values: [],
+      count: 0,
+    });
+
+    const result = await client.distinctValues("empty", "tag");
+
+    expect(result.count).toBe(0);
+    expect(result.values).toHaveLength(0);
+  });
+
+  it("calls correct endpoint", async () => {
+    const client = createTestClient();
+
+    mockTokenResponse();
+    mockJsonResponse({
+      collection: "users",
+      field: "role",
+      values: ["admin", "user"],
+      count: 2,
+    });
+
+    await client.distinctValues("users", "role");
+
+    const calls = (global.fetch as ReturnType<typeof vi.fn>).mock.calls;
+    const dataCall = calls[1]; // calls[0] is token
+    expect(dataCall[0]).toContain("/api/distinct/users/role");
+    expect(dataCall[1]?.method).toBe("POST");
+  });
+});
+
+// ============================================================================
+// Raw Completion Tests
+// ============================================================================
+
+describe("EkoDBClient rawCompletion", () => {
+  it("returns content from LLM response", async () => {
+    const client = createTestClient();
+
+    mockTokenResponse();
+    mockJsonResponse({ content: "The answer is 42." });
+
+    const result = await client.rawCompletion({
+      system_prompt: "You are a helpful assistant.",
+      message: "What is the answer?",
+    });
+
+    expect(result.content).toBe("The answer is 42.");
+  });
+
+  it("sends all fields in request body", async () => {
+    const client = createTestClient();
+
+    mockTokenResponse();
+    mockJsonResponse({ content: "Response text." });
+
+    await client.rawCompletion({
+      system_prompt: "System.",
+      message: "User.",
+      provider: "openai",
+      model: "gpt-4o",
+      max_tokens: 512,
+    });
+
+    const calls = (global.fetch as ReturnType<typeof vi.fn>).mock.calls;
+    const dataCall = calls[1]; // calls[0] is token
+    const body = JSON.parse(dataCall[1]?.body as string);
+    expect(body.system_prompt).toBe("System.");
+    expect(body.message).toBe("User.");
+    expect(body.provider).toBe("openai");
+    expect(body.model).toBe("gpt-4o");
+    expect(body.max_tokens).toBe(512);
+  });
+
+  it("omits optional fields when not provided", async () => {
+    const client = createTestClient();
+
+    mockTokenResponse();
+    mockJsonResponse({ content: "Response." });
+
+    await client.rawCompletion({
+      system_prompt: "System.",
+      message: "User.",
+    });
+
+    const calls = (global.fetch as ReturnType<typeof vi.fn>).mock.calls;
+    const dataCall = calls[1];
+    const body = JSON.parse(dataCall[1]?.body as string);
+    expect(body.provider).toBeUndefined();
+    expect(body.model).toBeUndefined();
+    expect(body.max_tokens).toBeUndefined();
+  });
+
+  it("calls correct endpoint with POST method", async () => {
+    const client = createTestClient();
+
+    mockTokenResponse();
+    mockJsonResponse({ content: "Response." });
+
+    await client.rawCompletion({
+      system_prompt: "System.",
+      message: "User.",
+    });
+
+    const calls = (global.fetch as ReturnType<typeof vi.fn>).mock.calls;
+    const dataCall = calls[1];
+    expect(dataCall[0]).toContain("/api/chat/complete");
+    expect(dataCall[1]?.method).toBe("POST");
+  });
+});
+
+// ============================================================================
+// Token Management Tests
+// ============================================================================
+
+describe("refreshToken", () => {
+  it("fetches a new token", async () => {
+    const client = createTestClient();
+
+    // First token fetch (init)
+    mockTokenResponse();
+    mockJsonResponse({ status: "ok" });
+    await client.health();
+
+    // Refresh token
+    mockFetch.mockResolvedValueOnce({
+      ok: true,
+      status: 200,
+      json: async () => ({ token: "new-jwt-token" }),
+      headers: new Headers(),
+    });
+    await client.refreshToken();
+
+    // Verify it called the token endpoint again
+    const calls = (global.fetch as ReturnType<typeof vi.fn>).mock.calls;
+    const tokenCalls = calls.filter((c: unknown[]) =>
+      (c[0] as string).includes("/api/auth/token"),
+    );
+    expect(tokenCalls.length).toBe(2);
+  });
+});
+
+describe("clearTokenCache", () => {
+  it("clears the cached token", async () => {
+    const client = createTestClient();
+
+    // Init with token
+    mockTokenResponse();
+    mockJsonResponse({ status: "ok" });
+    await client.health();
+
+    // Clear cache
+    client.clearTokenCache();
+
+    // Next request should fetch a new token
+    mockTokenResponse();
+    mockJsonResponse({ status: "ok" });
+    await client.health();
+
+    const calls = (global.fetch as ReturnType<typeof vi.fn>).mock.calls;
+    const tokenCalls = calls.filter((c: unknown[]) =>
+      (c[0] as string).includes("/api/auth/token"),
+    );
+    expect(tokenCalls.length).toBe(2);
+  });
+});
+
+// ============================================================================
+// findByIdWithProjection Tests
+// ============================================================================
+
+describe("findByIdWithProjection", () => {
+  it("calls correct endpoint with select_fields", async () => {
+    const client = createTestClient();
+
+    mockTokenResponse();
+    mockJsonResponse({ id: "123", name: "Alice" });
+
+    await client.findByIdWithProjection("users", "123", ["name", "email"]);
+
+    const calls = (global.fetch as ReturnType<typeof vi.fn>).mock.calls;
+    const dataCall = calls[1];
+    expect(dataCall[0]).toContain("/api/find/users/123");
+    expect(dataCall[0]).toContain("select_fields=name%2Cemail");
+  });
+
+  it("calls correct endpoint with exclude_fields", async () => {
+    const client = createTestClient();
+
+    mockTokenResponse();
+    mockJsonResponse({ id: "123", name: "Alice" });
+
+    await client.findByIdWithProjection("users", "123", undefined, [
+      "password",
+    ]);
+
+    const calls = (global.fetch as ReturnType<typeof vi.fn>).mock.calls;
+    const dataCall = calls[1];
+    expect(dataCall[0]).toContain("exclude_fields=password");
+  });
+
+  it("calls without params when no projection", async () => {
+    const client = createTestClient();
+
+    mockTokenResponse();
+    mockJsonResponse({ id: "123", name: "Alice" });
+
+    await client.findByIdWithProjection("users", "123");
+
+    const calls = (global.fetch as ReturnType<typeof vi.fn>).mock.calls;
+    const dataCall = calls[1];
+    expect(dataCall[0]).toBe("http://localhost:8080/api/find/users/123");
+  });
+});