Roo-Code/src/api/transform/__tests__/stream.test.ts

import { ApiStreamChunk } from "../stream"

describe("API Stream Types", () => {
	describe("ApiStreamChunk", () => {
		it("should correctly handle text chunks", () => {
			const textChunk: ApiStreamChunk = {
				type: "text",
				text: "Hello world",
			}

			expect(textChunk.type).toBe("text")
			expect(textChunk.text).toBe("Hello world")
		})

		it("should correctly handle usage chunks with cache information", () => {
			const usageChunk: ApiStreamChunk = {
				type: "usage",
				inputTokens: 100,
				outputTokens: 50,
				cacheWriteTokens: 20,
				cacheReadTokens: 10,
			}

			expect(usageChunk.type).toBe("usage")
			expect(usageChunk.inputTokens).toBe(100)
			expect(usageChunk.outputTokens).toBe(50)
			expect(usageChunk.cacheWriteTokens).toBe(20)
			expect(usageChunk.cacheReadTokens).toBe(10)
		})

		it("should handle usage chunks without cache tokens", () => {
			const usageChunk: ApiStreamChunk = {
				type: "usage",
				inputTokens: 100,
				outputTokens: 50,
			}

			expect(usageChunk.type).toBe("usage")
			expect(usageChunk.inputTokens).toBe(100)
			expect(usageChunk.outputTokens).toBe(50)
			expect(usageChunk.cacheWriteTokens).toBeUndefined()
			expect(usageChunk.cacheReadTokens).toBeUndefined()
		})

		it("should handle text chunks with empty strings", () => {
			const emptyTextChunk: ApiStreamChunk = {
				type: "text",
				text: "",
			}

			expect(emptyTextChunk.type).toBe("text")
			expect(emptyTextChunk.text).toBe("")
		})

		it("should handle usage chunks with zero tokens", () => {
			const zeroUsageChunk: ApiStreamChunk = {
				type: "usage",
				inputTokens: 0,
				outputTokens: 0,
			}

			expect(zeroUsageChunk.type).toBe("usage")
			expect(zeroUsageChunk.inputTokens).toBe(0)
			expect(zeroUsageChunk.outputTokens).toBe(0)
		})

		it("should handle usage chunks with large token counts", () => {
			const largeUsageChunk: ApiStreamChunk = {
				type: "usage",
				inputTokens: 1000000,
				outputTokens: 500000,
				cacheWriteTokens: 200000,
				cacheReadTokens: 100000,
			}

			expect(largeUsageChunk.type).toBe("usage")
			expect(largeUsageChunk.inputTokens).toBe(1000000)
			expect(largeUsageChunk.outputTokens).toBe(500000)
			expect(largeUsageChunk.cacheWriteTokens).toBe(200000)
			expect(largeUsageChunk.cacheReadTokens).toBe(100000)
		})

		it("should handle text chunks with special characters", () => {
			const specialCharsChunk: ApiStreamChunk = {
				type: "text",
				text: "!@#$%^&*()_+-=[]{}|;:,.<>?`~",
			}

			expect(specialCharsChunk.type).toBe("text")
			expect(specialCharsChunk.text).toBe("!@#$%^&*()_+-=[]{}|;:,.<>?`~")
		})

		it("should handle text chunks with unicode characters", () => {
			const unicodeChunk: ApiStreamChunk = {
				type: "text",
				text: "你好世界👋🌍",
			}

			expect(unicodeChunk.type).toBe("text")
			expect(unicodeChunk.text).toBe("你好世界👋🌍")
		})

		it("should handle text chunks with multiline content", () => {
			const multilineChunk: ApiStreamChunk = {
				type: "text",
				text: "Line 1\nLine 2\nLine 3",
			}

			expect(multilineChunk.type).toBe("text")
			expect(multilineChunk.text).toBe("Line 1\nLine 2\nLine 3")
			expect(multilineChunk.text.split("\n")).toHaveLength(3)
		})
	})
})