tlc-claude-code 2.2.1 → 2.4.0

package/server/lib/test-selector.test.js

@@ -0,0 +1,172 @@
+// @depends server/lib/test-selector.js
+import { describe, it, expect } from 'vitest';
+import {
+  parseDependsComment,
+  buildTestMap,
+  getAffectedTests,
+  formatSelection,
+} from './test-selector.js';
+
+describe('test-selector', () => {
+  describe('parseDependsComment', () => {
+    it('extracts a single @depends path', () => {
+      const result = parseDependsComment('// @depends src/lib/auth.js');
+      expect(result).toEqual(['src/lib/auth.js']);
+    });
+
+    it('extracts multiple @depends from separate lines', () => {
+      const content = '// @depends src/a.js\n// @depends src/b.js';
+      const result = parseDependsComment(content);
+      expect(result).toEqual(['src/a.js', 'src/b.js']);
+    });
+
+    it('returns empty array when no @depends present', () => {
+      const result = parseDependsComment('no depends here');
+      expect(result).toEqual([]);
+    });
+
+    it('handles comma-separated dependencies on one line', () => {
+      const result = parseDependsComment('// @depends src/a.js, src/b.js');
+      expect(result).toEqual(['src/a.js', 'src/b.js']);
+    });
+
+    it('handles empty string', () => {
+      const result = parseDependsComment('');
+      expect(result).toEqual([]);
+    });
+
+    it('trims whitespace from paths', () => {
+      const result = parseDependsComment('// @depends   src/foo.js  ');
+      expect(result).toEqual(['src/foo.js']);
+    });
+
+    it('ignores @depends inside non-comment lines', () => {
+      const content = 'const x = "// @depends fake.js";\n// @depends real.js';
+      const result = parseDependsComment(content);
+      // Both lines start with // so the string literal line won't match
+      // Actually the first line starts with 'const' so it won't match
+      expect(result).toEqual(['real.js']);
+    });
+  });
+
+  describe('buildTestMap', () => {
+    it('builds mapping from test files to their dependencies', async () => {
+      const mockReadFile = async (filePath) => {
+        const files = {
+          'tests/auth.test.js': '// @depends src/auth.js\nimport { auth } from "../src/auth.js";',
+          'tests/db.test.js': '// @depends src/db.js, src/models.js\nimport { db } from "../src/db.js";',
+          'tests/utils.test.js': 'import { utils } from "../src/utils.js";',
+        };
+        if (files[filePath] === undefined) {
+          throw new Error(`ENOENT: no such file: ${filePath}`);
+        }
+        return files[filePath];
+      };
+
+      const testFiles = ['tests/auth.test.js', 'tests/db.test.js', 'tests/utils.test.js'];
+      const result = await buildTestMap(testFiles, { readFile: mockReadFile });
+
+      expect(result).toBeInstanceOf(Map);
+      expect(result.get('tests/auth.test.js')).toEqual(['src/auth.js']);
+      expect(result.get('tests/db.test.js')).toEqual(['src/db.js', 'src/models.js']);
+      expect(result.get('tests/utils.test.js')).toEqual([]);
+    });
+
+    it('handles readFile errors gracefully by treating as no-depends', async () => {
+      const mockReadFile = async () => {
+        throw new Error('ENOENT: file not found');
+      };
+
+      const result = await buildTestMap(['missing.test.js'], { readFile: mockReadFile });
+
+      expect(result.get('missing.test.js')).toEqual([]);
+    });
+
+    it('returns empty map for empty test file list', async () => {
+      const mockReadFile = async () => '';
+      const result = await buildTestMap([], { readFile: mockReadFile });
+
+      expect(result).toBeInstanceOf(Map);
+      expect(result.size).toBe(0);
+    });
+  });
+
+  describe('getAffectedTests', () => {
+    const testMap = new Map([
+      ['tests/auth.test.js', ['src/auth.js']],
+      ['tests/db.test.js', ['src/db.js', 'src/models.js']],
+      ['tests/utils.test.js', []],  // no @depends — always included
+      ['tests/api.test.js', ['src/api.js', 'src/auth.js']],
+    ]);
+
+    it('returns tests depending on a changed file', () => {
+      const result = getAffectedTests(['src/auth.js'], testMap);
+
+      expect(result).toContain('tests/auth.test.js');
+      expect(result).toContain('tests/api.test.js');
+      // no-depends tests are always included
+      expect(result).toContain('tests/utils.test.js');
+    });
+
+    it('returns only no-depends tests when changed file is unrelated', () => {
+      const result = getAffectedTests(['src/unrelated.js'], testMap);
+
+      expect(result).toEqual(['tests/utils.test.js']);
+    });
+
+    it('always includes tests with no @depends annotation', () => {
+      const result = getAffectedTests(['src/db.js'], testMap);
+
+      expect(result).toContain('tests/utils.test.js');
+      expect(result).toContain('tests/db.test.js');
+      expect(result).not.toContain('tests/auth.test.js');
+    });
+
+    it('returns only no-depends tests when changedFiles is empty', () => {
+      const result = getAffectedTests([], testMap);
+
+      expect(result).toEqual(['tests/utils.test.js']);
+    });
+
+    it('handles empty test map', () => {
+      const result = getAffectedTests(['src/auth.js'], new Map());
+
+      expect(result).toEqual([]);
+    });
+
+    it('handles multiple changed files', () => {
+      const result = getAffectedTests(['src/auth.js', 'src/db.js'], testMap);
+
+      expect(result).toContain('tests/auth.test.js');
+      expect(result).toContain('tests/db.test.js');
+      expect(result).toContain('tests/api.test.js');
+      expect(result).toContain('tests/utils.test.js');
+    });
+  });
+
+  describe('formatSelection', () => {
+    it('returns summary with skip count when not all tests selected', () => {
+      const result = formatSelection(3, 10);
+
+      expect(result).toBe('Running 3 of 10 tests (7 skipped — dependencies unchanged)');
+    });
+
+    it('returns "all tests" message when all selected', () => {
+      const result = formatSelection(10, 10);
+
+      expect(result).toBe('Running all 10 tests');
+    });
+
+    it('handles 1 of 1', () => {
+      const result = formatSelection(1, 1);
+
+      expect(result).toBe('Running all 1 tests');
+    });
+
+    it('handles 0 of N', () => {
+      const result = formatSelection(0, 5);
+
+      expect(result).toBe('Running 0 of 5 tests (5 skipped — dependencies unchanged)');
+    });
+  });
+});