Achieve 100% test coverage across all modules

Author: edycutjong

tests/lib/mock.test.ts

@@ -67,10 +67,13 @@ describe('mock', () => {
     });
 
     it('should return smart money labels for SM addresses', () => {
-      const result = getMockData('profiler labels', ['--address', MOCK_WALLETS.smartMoney1]) as any[];
-      expect(result).toBeInstanceOf(Array);
-      expect(result[0].label).toBe('Smart Money');
-      expect(result[0].category).toBe('Smart Money');
+      const result1 = getMockData('profiler labels', ['--address', MOCK_WALLETS.smartMoney1]) as any[];
+      expect(result1).toBeInstanceOf(Array);
+      expect(result1[0].label).toBe('Smart Money');
+      
+      const result2 = getMockData('profiler labels', ['--address', MOCK_WALLETS.smartMoney2]) as any[];
+      expect(result2).toBeInstanceOf(Array);
+      expect(result2[0].label).toBe('Smart Money');
     });
 
     it('should return exchange labels for labeled addresses', () => {
@@ -180,6 +183,12 @@ describe('mock', () => {
       expect(result).toBeNull();
     });
 
+    it('should handle missing arg values gracefully for profiler labels', () => {
+      const result = getMockData('profiler labels', []) as any[];
+      expect(result).toBeInstanceOf(Array);
+      expect(result.length).toBe(0);
+    });
+
     it('should handle missing arg values gracefully', () => {
       const result = getMockData('profiler trace', []) as any;
       expect(result).toBeDefined();

tests/lib/nansen.test.ts

@@ -78,6 +78,14 @@ describe('nansen', () => {
     const r = await execNansen('cmd', []); expect(r.error).toBe('rate limit'); expect(r.code).toBe('RL');
   });
 
+  it('handles JSON error in stderr with no error field', async () => {
+    mockMode = false;
+    mockExecFile.mockImplementation((_c: any, _a: any, _o: any, cb: any) => cb(new Error('fallback msg'), '', '{"code":"NO_ERR_FIELD"}'));
+    const r = await execNansen('cmd', []); 
+    expect(r.error).toBe('fallback msg'); 
+    expect(r.code).toBe('NO_ERR_FIELD');
+  });
+
   it('handles plain error text', async () => {
     mockMode = false;
     mockExecFile.mockImplementation((_c: any, _a: any, _o: any, cb: any) => cb(new Error('e'), '', 'plain error'));

tests/lib/telemetry.test.ts

@@ -167,12 +167,21 @@ describe('telemetry', () => {
       recordCall(makeEntry({ endpoint: 'profiler trace', status: '200', cache: 'MISS', latency_ms: 200 }));
       recordCall(makeEntry({ endpoint: 'profiler labels', status: 'SUCCESS', cache: 'HIT', latency_ms: 5 }));
       recordCall(makeEntry({ endpoint: 'long-endpoint-name-that-exceeds-forty-characters-limit', status: 'ERROR', latency_ms: 600 }));
+      recordCall(makeEntry({ endpoint: 'unknown-status', status: 'UNKNOWN_STATUS', cache: 'N/A', latency_ms: 10 }));
 
       const spy = vi.spyOn(console, 'log').mockImplementation(() => {});
       printTelemetryReceipt();
       expect(spy).toHaveBeenCalled();
       spy.mockRestore();
     });
+
+    it('should print receipt with zero error count', () => {
+      recordCall(makeEntry({ endpoint: 'profiler trace', status: '200', cache: 'MISS', latency_ms: 200 }));
+      const spy = vi.spyOn(console, 'log').mockImplementation(() => {});
+      printTelemetryReceipt();
+      expect(spy).toHaveBeenCalled();
+      spy.mockRestore();
+    });
   });
 
   describe('aliases', () => {

tests/lib/terminal-report.test.ts

@@ -15,12 +15,12 @@ function createMockGraph(): GraphData {
   return {
     nodes: [
       { id: '0xseed', label: '0xseed...dead', type: 'seed', balance_usd: 1000000, pnl_30d: 50000, labels: [], sm_labels: [], depth: 0, defi_protocols: 2 },
-      { id: '0xfund', label: 'Fund Alpha', type: 'smart-money', balance_usd: 5000000, pnl_30d: 200000, labels: ['Fund'], sm_labels: ['Smart Money'], depth: 1, defi_protocols: 5 },
+      { id: '0xfund', label: 'Fund Alpha', type: 'smart-money', balance_usd: 5000000, pnl_30d: 200000, labels: ['Fund'], sm_labels: [], depth: 1, defi_protocols: 5 },
       { id: '0xlabel', label: 'Binance', type: 'labeled', balance_usd: 250000, pnl_30d: -10000, labels: ['Binance'], sm_labels: [], depth: 1, defi_protocols: 0 },
       // Use empty label to trigger `truncate` coverage
       { id: '0xunknown_long_address', label: '', type: 'unknown', balance_usd: 500, pnl_30d: 0, labels: [], sm_labels: [], depth: 2, defi_protocols: 0 },
       { id: '0xshort', label: '', type: 'unknown', balance_usd: 100, pnl_30d: 0, labels: [], sm_labels: [], depth: 2, defi_protocols: 0 },
-      { id: '0xcontract', label: 'Uniswap Router', type: 'contract', balance_usd: 0, pnl_30d: 0, labels: ['Contract'], sm_labels: [], depth: 1, defi_protocols: 0 },
+      { id: '0xcontract', label: 'Uniswap Router', type: 'contract', balance_usd: 100000, pnl_30d: 0, labels: ['Contract'], sm_labels: [], depth: 1, defi_protocols: 0 },
     ],
     links: [
       { source: '0xseed', target: '0xfund', volume_usd: 500000, tx_count: 20, direction: 'outflow' },