From ba45d2f6018e931af5fcded127a507df1a490c7d Mon Sep 17 00:00:00 2001
From: Eason Goodale <easong@openai.com>
Date: Sat, 26 Apr 2025 12:15:34 -0700
Subject: [PATCH] test cache discount

---
 codex-cli/tests/estimate-cost.test.ts | 21 ++++++++++++++++++++-
 1 file changed, 20 insertions(+), 1 deletion(-)

diff --git a/codex-cli/tests/estimate-cost.test.ts b/codex-cli/tests/estimate-cost.test.ts
index 18100b7568..2bd7f26d4e 100644
--- a/codex-cli/tests/estimate-cost.test.ts
+++ b/codex-cli/tests/estimate-cost.test.ts
@@ -1,6 +1,9 @@
 import { describe, expect, test } from "vitest";
 
-import { estimateCostUSD } from "../src/utils/estimate-cost.js";
+import {
+  estimateCostUSD,
+  estimateCostFromUsage,
+} from "../src/utils/estimate-cost.js";
 import { SessionCostTracker } from "../src/utils/session-cost.js";
 import type { ResponseItem } from "openai/resources/responses/responses.mjs";
 
@@ -36,6 +39,22 @@ describe("estimateCostUSD", () => {
     );
     expect(cost2!).toBeGreaterThan(cost!);
   });
+
+  test("cost calculation honours cached input token discount", () => {
+    const usage = {
+      input_tokens: 1000,
+      input_tokens_details: { cached_tokens: 600 },
+      output_tokens: 500,
+      total_tokens: 1500,
+    } as any; // simple literal structure for test
+
+    const cost = estimateCostFromUsage(usage, "gpt-4.1");
+
+    // Expected: (1000-600)*0.000002 + 600*0.0000005 + 500*0.000008
+    const expected = 400 * 0.000002 + 600 * 0.0000005 + 500 * 0.000008;
+    expect(cost).not.toBeNull();
+    expect(cost!).toBeCloseTo(expected, 8);
+  });
 });
 
 describe("SessionCostTracker", () => {