test: add tests for unified cost tracking and chart types

Co-authored-by: hotlong <50353452+hotlong@users.noreply.github.com>

Author: Copilot

packages/spec/src/ai/cost.test.ts

@@ -1,5 +1,7 @@
 import { describe, it, expect } from 'vitest';
 import {
+  TokenUsageSchema,
+  AIOperationCostSchema,
   CostMetricTypeSchema,
   BillingPeriodSchema,
   CostEntrySchema,
@@ -14,12 +16,130 @@ import {
   CostOptimizationRecommendationSchema,
   CostReportSchema,
   CostQueryFiltersSchema,
+  type TokenUsage,
+  type AIOperationCost,
   type CostEntry,
   type BudgetLimit,
   type CostAlert,
   type CostReport,
 } from './cost.zod';
 
+describe('TokenUsageSchema', () => {
+  it('should accept valid token usage', () => {
+    const usage: TokenUsage = {
+      prompt: 1500,
+      completion: 800,
+      total: 2300,
+    };
+    expect(() => TokenUsageSchema.parse(usage)).not.toThrow();
+  });
+
+  it('should require all fields', () => {
+    expect(() => TokenUsageSchema.parse({
+      prompt: 1500,
+      completion: 800,
+      // missing total
+    })).toThrow();
+  });
+
+  it('should reject negative values', () => {
+    expect(() => TokenUsageSchema.parse({
+      prompt: -100,
+      completion: 800,
+      total: 700,
+    })).toThrow();
+  });
+});
+
+describe('AIOperationCostSchema', () => {
+  it('should accept minimal AI operation cost', () => {
+    const cost: AIOperationCost = {
+      operationId: 'op-123',
+      operationType: 'conversation',
+      modelId: 'gpt-4-turbo',
+      tokens: {
+        prompt: 1500,
+        completion: 800,
+        total: 2300,
+      },
+      cost: 0.05,
+      timestamp: '2024-01-15T10:00:00Z',
+    };
+    expect(() => AIOperationCostSchema.parse(cost)).not.toThrow();
+  });
+
+  it('should accept all operation types', () => {
+    const types = ['conversation', 'orchestration', 'prediction', 'rag', 'nlq'] as const;
+    
+    types.forEach(type => {
+      const cost: AIOperationCost = {
+        operationId: 'op-123',
+        operationType: type,
+        modelId: 'gpt-4-turbo',
+        tokens: {
+          prompt: 1500,
+          completion: 800,
+          total: 2300,
+        },
+        cost: 0.05,
+        timestamp: '2024-01-15T10:00:00Z',
+      };
+      expect(() => AIOperationCostSchema.parse(cost)).not.toThrow();
+    });
+  });
+
+  it('should accept AI operation cost with agent name and metadata', () => {
+    const cost: AIOperationCost = {
+      operationId: 'op-456',
+      operationType: 'orchestration',
+      agentName: 'support_agent',
+      modelId: 'gpt-4-turbo',
+      tokens: {
+        prompt: 2000,
+        completion: 1200,
+        total: 3200,
+      },
+      cost: 0.08,
+      timestamp: '2024-01-15T11:00:00Z',
+      metadata: {
+        workflowId: 'wf-789',
+        recordId: 'case-123',
+      },
+    };
+    expect(() => AIOperationCostSchema.parse(cost)).not.toThrow();
+  });
+
+  it('should reject invalid timestamp', () => {
+    expect(() => AIOperationCostSchema.parse({
+      operationId: 'op-123',
+      operationType: 'conversation',
+      modelId: 'gpt-4-turbo',
+      tokens: {
+        prompt: 1500,
+        completion: 800,
+        total: 2300,
+      },
+      cost: 0.05,
+      timestamp: 'invalid-timestamp',
+    })).toThrow();
+  });
+
+  it('should reject negative cost', () => {
+    expect(() => AIOperationCostSchema.parse({
+      operationId: 'op-123',
+      operationType: 'conversation',
+      modelId: 'gpt-4-turbo',
+      tokens: {
+        prompt: 1500,
+        completion: 800,
+        total: 2300,
+      },
+      cost: -0.05,
+      timestamp: '2024-01-15T10:00:00Z',
+    })).toThrow();
+  });
+});
+
 describe('CostMetricTypeSchema', () => {
   it('should accept all valid metric types', () => {
     const types = ['token', 'request', 'character', 'second', 'image', 'embedding'] as const;

packages/spec/src/ui/chart.test.ts

@@ -0,0 +1,210 @@
+import { describe, it, expect } from 'vitest';
+import {
+  ChartTypeSchema,
+  ChartConfigSchema,
+  type ChartType,
+  type ChartConfig,
+} from './chart.zod';
+
+describe('ChartTypeSchema', () => {
+  it('should accept all comparison chart types', () => {
+    const types = ['bar', 'horizontal-bar', 'column', 'grouped-bar', 'stacked-bar'] as const;
+    
+    types.forEach(type => {
+      expect(() => ChartTypeSchema.parse(type)).not.toThrow();
+    });
+  });
+
+  it('should accept all trend chart types', () => {
+    const types = ['line', 'area', 'stacked-area', 'step-line'] as const;
+    
+    types.forEach(type => {
+      expect(() => ChartTypeSchema.parse(type)).not.toThrow();
+    });
+  });
+
+  it('should accept all distribution chart types', () => {
+    const types = ['pie', 'donut', 'funnel'] as const;
+    
+    types.forEach(type => {
+      expect(() => ChartTypeSchema.parse(type)).not.toThrow();
+    });
+  });
+
+  it('should accept all relationship chart types', () => {
+    const types = ['scatter', 'bubble'] as const;
+    
+    types.forEach(type => {
+      expect(() => ChartTypeSchema.parse(type)).not.toThrow();
+    });
+  });
+
+  it('should accept all composition chart types', () => {
+    const types = ['treemap', 'sunburst', 'sankey'] as const;
+    
+    types.forEach(type => {
+      expect(() => ChartTypeSchema.parse(type)).not.toThrow();
+    });
+  });
+
+  it('should accept all performance chart types', () => {
+    const types = ['gauge', 'metric', 'kpi'] as const;
+    
+    types.forEach(type => {
+      expect(() => ChartTypeSchema.parse(type)).not.toThrow();
+    });
+  });
+
+  it('should accept all geo chart types', () => {
+    const types = ['choropleth', 'bubble-map'] as const;
+    
+    types.forEach(type => {
+      expect(() => ChartTypeSchema.parse(type)).not.toThrow();
+    });
+  });
+
+  it('should accept all advanced chart types', () => {
+    const types = ['heatmap', 'radar', 'waterfall', 'box-plot', 'violin'] as const;
+    
+    types.forEach(type => {
+      expect(() => ChartTypeSchema.parse(type)).not.toThrow();
+    });
+  });
+
+  it('should accept all tabular chart types', () => {
+    const types = ['table', 'pivot'] as const;
+    
+    types.forEach(type => {
+      expect(() => ChartTypeSchema.parse(type)).not.toThrow();
+    });
+  });
+
+  it('should reject invalid chart type', () => {
+    expect(() => ChartTypeSchema.parse('invalid-chart')).toThrow();
+  });
+});
+
+describe('ChartConfigSchema', () => {
+  it('should accept minimal chart config', () => {
+    const config: ChartConfig = {
+      type: 'bar',
+    };
+    const result = ChartConfigSchema.parse(config);
+    expect(result.type).toBe('bar');
+    expect(result.showLegend).toBe(true);
+    expect(result.showDataLabels).toBe(false);
+  });
+
+  it('should accept full chart config', () => {
+    const config: ChartConfig = {
+      type: 'line',
+      title: 'Sales Trend',
+      description: 'Monthly sales performance',
+      showLegend: true,
+      showDataLabels: true,
+      colors: ['#FF6384', '#36A2EB', '#FFCE56'],
+    };
+    expect(() => ChartConfigSchema.parse(config)).not.toThrow();
+  });
+
+  it('should apply default values', () => {
+    const config: ChartConfig = {
+      type: 'pie',
+      title: 'Revenue by Region',
+    };
+    const result = ChartConfigSchema.parse(config);
+    expect(result.showLegend).toBe(true);
+    expect(result.showDataLabels).toBe(false);
+  });
+
+  it('should allow custom colors', () => {
+    const config: ChartConfig = {
+      type: 'donut',
+      colors: ['#1f77b4', '#ff7f0e', '#2ca02c', '#d62728'],
+    };
+    const result = ChartConfigSchema.parse(config);
+    expect(result.colors).toHaveLength(4);
+  });
+});
+
+describe('Real-World Chart Configuration Examples', () => {
+  it('should accept bar chart for comparison', () => {
+    const config: ChartConfig = {
+      type: 'bar',
+      title: 'Sales by Product Category',
+      description: 'Comparison of sales across different product categories',
+      showLegend: true,
+      showDataLabels: true,
+      colors: ['#4e79a7', '#f28e2c', '#e15759'],
+    };
+    expect(() => ChartConfigSchema.parse(config)).not.toThrow();
+  });
+
+  it('should accept line chart for trends', () => {
+    const config: ChartConfig = {
+      type: 'line',
+      title: 'Revenue Trend',
+      description: 'Monthly revenue over the past year',
+      showLegend: true,
+      showDataLabels: false,
+    };
+    expect(() => ChartConfigSchema.parse(config)).not.toThrow();
+  });
+
+  it('should accept pie chart for distribution', () => {
+    const config: ChartConfig = {
+      type: 'pie',
+      title: 'Market Share',
+      description: 'Market share by competitor',
+      showLegend: true,
+      showDataLabels: true,
+    };
+    expect(() => ChartConfigSchema.parse(config)).not.toThrow();
+  });
+
+  it('should accept gauge for performance metrics', () => {
+    const config: ChartConfig = {
+      type: 'gauge',
+      title: 'Customer Satisfaction Score',
+      description: 'Current satisfaction rating (0-100)',
+      showLegend: false,
+      colors: ['#22c55e', '#eab308', '#ef4444'],
+    };
+    expect(() => ChartConfigSchema.parse(config)).not.toThrow();
+  });
+
+  it('should accept heatmap for correlation analysis', () => {
+    const config: ChartConfig = {
+      type: 'heatmap',
+      title: 'User Activity Heatmap',
+      description: 'Hourly user activity by day of week',
+      showLegend: true,
+      showDataLabels: false,
+      colors: ['#440154', '#31688e', '#35b779', '#fde724'],
+    };
+    expect(() => ChartConfigSchema.parse(config)).not.toThrow();
+  });
+
+  it('should accept funnel chart for conversion tracking', () => {
+    const config: ChartConfig = {
+      type: 'funnel',
+      title: 'Sales Funnel',
+      description: 'Conversion rates at each stage',
+      showLegend: false,
+      showDataLabels: true,
+    };
+    expect(() => ChartConfigSchema.parse(config)).not.toThrow();
+  });
+
+  it('should accept waterfall chart for financial analysis', () => {
+    const config: ChartConfig = {
+      type: 'waterfall',
+      title: 'Profit & Loss Breakdown',
+      description: 'Revenue, costs, and profit components',
+      showLegend: false,
+      showDataLabels: true,
+      colors: ['#22c55e', '#ef4444', '#6366f1'],
+    };
+    expect(() => ChartConfigSchema.parse(config)).not.toThrow();
+  });
+});