Add test coverage

src/api/transform/__tests__/stream.test.ts

@@ -0,0 +1,114 @@
+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);
+        });
+    });
+});