driftdetect-core 0.4.6 → 0.6.0

package/dist/test-topology/extractors/csharp-test-extractor.js

@@ -0,0 +1,367 @@
+/**
+ * C# Test Extractor
+ *
+ * Extracts test information from xUnit, NUnit, MSTest, and Moq.
+ */
+import { BaseTestExtractor } from './base-test-extractor.js';
+// ============================================================================
+// Framework Detection
+// ============================================================================
+const FRAMEWORK_NAMESPACES = {
+    'Xunit': 'xunit',
+    'xUnit': 'xunit',
+    'NUnit': 'nunit',
+    'NUnit.Framework': 'nunit',
+    'Microsoft.VisualStudio.TestTools.UnitTesting': 'mstest',
+};
+const FRAMEWORK_ATTRIBUTES = {
+    'Fact': 'xunit',
+    'Theory': 'xunit',
+    'Test': 'nunit',
+    'TestCase': 'nunit',
+    'TestMethod': 'mstest',
+    'DataTestMethod': 'mstest',
+};
+// ============================================================================
+// Extractor Implementation
+// ============================================================================
+export class CSharpTestExtractor extends BaseTestExtractor {
+    constructor(parser) {
+        super(parser, 'csharp');
+    }
+    extract(content, filePath) {
+        const tree = this.parser.parse(content);
+        const root = tree.rootNode;
+        const framework = this.detectFramework(root);
+        const testCases = this.extractTestCases(root);
+        const mocks = this.extractMocks(root, framework);
+        const setupBlocks = this.extractSetupBlocks(root);
+        // Enrich test cases with quality
+        for (const test of testCases) {
+            const testMocks = mocks.filter(m => m.line >= test.line && m.line <= test.line + 100);
+            test.quality = this.calculateQuality(test.assertions, testMocks, test.directCalls);
+        }
+        return {
+            file: filePath,
+            framework,
+            language: 'csharp',
+            testCases,
+            mocks,
+            setupBlocks,
+        };
+    }
+    detectFramework(root) {
+        const imports = this.findImports(root);
+        // Check using statements
+        for (const imp of imports) {
+            for (const [pattern, framework] of Object.entries(FRAMEWORK_NAMESPACES)) {
+                if (imp.includes(pattern)) {
+                    return framework;
+                }
+            }
+        }
+        // Check for test attributes in the code
+        const text = root.text;
+        for (const [attr, framework] of Object.entries(FRAMEWORK_ATTRIBUTES)) {
+            if (text.includes(`[${attr}]`) || text.includes(`[${attr}(`)) {
+                return framework;
+            }
+        }
+        return 'unknown';
+    }
+    extractTestCases(root) {
+        const testCases = [];
+        let currentClass;
+        this.walkNode(root, (node) => {
+            // Track class context
+            if (node.type === 'class_declaration') {
+                const nameNode = node.childForFieldName('name');
+                if (nameNode) {
+                    currentClass = nameNode.text;
+                }
+            }
+            // Find methods with test attributes
+            if (node.type === 'method_declaration') {
+                const attributeList = this.findAttributeList(node);
+                const nameNode = node.childForFieldName('name');
+                const bodyNode = node.childForFieldName('body');
+                if (nameNode && bodyNode && this.hasTestAttribute(attributeList)) {
+                    const name = nameNode.text;
+                    const qualifiedName = currentClass
+                        ? `${currentClass}.${name}`
+                        : name;
+                    const directCalls = this.extractFunctionCalls(bodyNode);
+                    const assertions = this.extractAssertions(bodyNode);
+                    testCases.push({
+                        id: this.generateTestId('', name, node.startPosition.row),
+                        name,
+                        parentBlock: currentClass,
+                        qualifiedName,
+                        file: '',
+                        line: node.startPosition.row + 1,
+                        directCalls,
+                        transitiveCalls: [],
+                        assertions,
+                        quality: {
+                            assertionCount: assertions.length,
+                            hasErrorCases: false,
+                            hasEdgeCases: false,
+                            mockRatio: 0,
+                            setupRatio: 0,
+                            score: 50,
+                        },
+                    });
+                }
+            }
+        });
+        return testCases;
+    }
+    extractMocks(root, _framework) {
+        const mocks = [];
+        this.walkNode(root, (node) => {
+            // new Mock<T>() pattern (Moq)
+            if (node.type === 'object_creation_expression') {
+                const typeNode = node.childForFieldName('type');
+                if (typeNode) {
+                    const typeText = typeNode.text;
+                    // Mock<IService>
+                    if (typeText.startsWith('Mock<')) {
+                        const mockTarget = typeText.slice(5, -1); // Extract type from Mock<Type>
+                        mocks.push({
+                            target: mockTarget,
+                            mockType: 'Mock<T>',
+                            line: node.startPosition.row + 1,
+                            isExternal: this.isExternalCSharpType(mockTarget),
+                        });
+                    }
+                    // Substitute.For<IService>() (NSubstitute)
+                    if (typeText.includes('Substitute')) {
+                        mocks.push({
+                            target: typeText,
+                            mockType: 'NSubstitute',
+                            line: node.startPosition.row + 1,
+                            isExternal: false,
+                        });
+                    }
+                }
+            }
+            // Substitute.For<T>() pattern (NSubstitute)
+            if (node.type === 'invocation_expression') {
+                const fnNode = node.childForFieldName('function');
+                if (fnNode?.type === 'member_access_expression') {
+                    const text = fnNode.text;
+                    if (text.includes('Substitute.For')) {
+                        const typeMatch = text.match(/For<([^>]+)>/);
+                        mocks.push({
+                            target: typeMatch?.[1] ?? 'unknown',
+                            mockType: 'Substitute.For<T>',
+                            line: node.startPosition.row + 1,
+                            isExternal: false,
+                        });
+                    }
+                    // A.Fake<T>() (FakeItEasy)
+                    if (text.includes('A.Fake')) {
+                        const typeMatch = text.match(/Fake<([^>]+)>/);
+                        mocks.push({
+                            target: typeMatch?.[1] ?? 'unknown',
+                            mockType: 'A.Fake<T>',
+                            line: node.startPosition.row + 1,
+                            isExternal: false,
+                        });
+                    }
+                }
+            }
+            // .Setup() calls on mocks (Moq)
+            if (node.type === 'invocation_expression') {
+                const fnNode = node.childForFieldName('function');
+                if (fnNode?.type === 'member_access_expression') {
+                    const nameNode = fnNode.childForFieldName('name');
+                    if (nameNode?.text === 'Setup' || nameNode?.text === 'SetupGet' ||
+                        nameNode?.text === 'SetupSet' || nameNode?.text === 'SetupSequence') {
+                        mocks.push({
+                            target: 'mock_setup',
+                            mockType: `Moq.${nameNode.text}`,
+                            line: node.startPosition.row + 1,
+                            isExternal: false,
+                            hasImplementation: true,
+                        });
+                    }
+                }
+            }
+        });
+        return mocks;
+    }
+    extractSetupBlocks(root) {
+        const blocks = [];
+        this.walkNode(root, (node) => {
+            if (node.type === 'method_declaration') {
+                const attributeList = this.findAttributeList(node);
+                const bodyNode = node.childForFieldName('body');
+                if (bodyNode) {
+                    let setupType = null;
+                    // Check for setup/teardown attributes
+                    for (const attr of attributeList) {
+                        const attrText = attr.text;
+                        // xUnit: constructor is setup, IDisposable.Dispose is teardown
+                        // NUnit
+                        if (attrText.includes('[SetUp]'))
+                            setupType = 'beforeEach';
+                        else if (attrText.includes('[TearDown]'))
+                            setupType = 'afterEach';
+                        else if (attrText.includes('[OneTimeSetUp]'))
+                            setupType = 'beforeAll';
+                        else if (attrText.includes('[OneTimeTearDown]'))
+                            setupType = 'afterAll';
+                        // MSTest
+                        else if (attrText.includes('[TestInitialize]'))
+                            setupType = 'beforeEach';
+                        else if (attrText.includes('[TestCleanup]'))
+                            setupType = 'afterEach';
+                        else if (attrText.includes('[ClassInitialize]'))
+                            setupType = 'beforeAll';
+                        else if (attrText.includes('[ClassCleanup]'))
+                            setupType = 'afterAll';
+                    }
+                    if (setupType) {
+                        const calls = this.extractFunctionCalls(bodyNode);
+                        blocks.push({
+                            type: setupType,
+                            line: node.startPosition.row + 1,
+                            calls,
+                        });
+                    }
+                }
+            }
+            // xUnit: constructor is setup
+            if (node.type === 'constructor_declaration') {
+                const bodyNode = node.childForFieldName('body');
+                if (bodyNode) {
+                    const calls = this.extractFunctionCalls(bodyNode);
+                    blocks.push({
+                        type: 'beforeEach',
+                        line: node.startPosition.row + 1,
+                        calls,
+                    });
+                }
+            }
+        });
+        return blocks;
+    }
+    findImports(root) {
+        const imports = [];
+        this.walkNode(root, (node) => {
+            if (node.type === 'using_directive') {
+                const nameNode = node.childForFieldName('name');
+                if (nameNode) {
+                    imports.push(nameNode.text);
+                }
+            }
+        });
+        return imports;
+    }
+    findAttributeList(node) {
+        const attributes = [];
+        // Look for attribute_list siblings before the method
+        let sibling = node.previousNamedSibling;
+        while (sibling && sibling.type === 'attribute_list') {
+            attributes.push(sibling);
+            sibling = sibling.previousNamedSibling;
+        }
+        // Also check children
+        this.walkNode(node, (child) => {
+            if (child.type === 'attribute_list') {
+                attributes.push(child);
+            }
+        });
+        return attributes;
+    }
+    hasTestAttribute(attributeList) {
+        const testAttributes = ['Fact', 'Theory', 'Test', 'TestCase', 'TestMethod', 'DataTestMethod'];
+        for (const attrList of attributeList) {
+            const text = attrList.text;
+            for (const attr of testAttributes) {
+                if (text.includes(`[${attr}]`) || text.includes(`[${attr}(`)) {
+                    return true;
+                }
+            }
+        }
+        return false;
+    }
+    extractAssertions(node) {
+        const assertions = [];
+        this.walkNode(node, (child) => {
+            if (child.type === 'invocation_expression') {
+                const fnNode = child.childForFieldName('function');
+                if (fnNode?.type === 'member_access_expression') {
+                    const objNode = fnNode.childForFieldName('expression');
+                    const nameNode = fnNode.childForFieldName('name');
+                    if (objNode && nameNode) {
+                        const obj = objNode.text;
+                        const method = nameNode.text;
+                        // xUnit Assert
+                        if (obj === 'Assert') {
+                            assertions.push({
+                                matcher: `Assert.${method}`,
+                                line: child.startPosition.row + 1,
+                                isErrorAssertion: method === 'Throws' || method === 'ThrowsAsync' ||
+                                    method === 'ThrowsAny' || method === 'ThrowsAnyAsync',
+                                isEdgeCaseAssertion: method === 'Null' || method === 'NotNull' ||
+                                    method === 'Empty' || method === 'NotEmpty',
+                            });
+                        }
+                        // NUnit Assert
+                        if (obj === 'Assert' || obj.endsWith('.Assert')) {
+                            assertions.push({
+                                matcher: method,
+                                line: child.startPosition.row + 1,
+                                isErrorAssertion: method === 'Throws' || method === 'ThrowsAsync' ||
+                                    method === 'Catch' || method === 'CatchAsync',
+                                isEdgeCaseAssertion: method === 'IsNull' || method === 'IsNotNull' ||
+                                    method === 'IsEmpty' || method === 'IsNotEmpty',
+                            });
+                        }
+                        // FluentAssertions: .Should()
+                        if (method === 'Should') {
+                            assertions.push({
+                                matcher: 'Should',
+                                line: child.startPosition.row + 1,
+                                isErrorAssertion: false,
+                                isEdgeCaseAssertion: false,
+                            });
+                        }
+                        // Moq Verify
+                        if (method === 'Verify' || method === 'VerifyAll' || method === 'VerifyNoOtherCalls') {
+                            assertions.push({
+                                matcher: `Mock.${method}`,
+                                line: child.startPosition.row + 1,
+                                isErrorAssertion: false,
+                                isEdgeCaseAssertion: false,
+                            });
+                        }
+                    }
+                }
+            }
+        });
+        return assertions;
+    }
+    isExternalCSharpType(typeName) {
+        // Interface naming convention - typically internal
+        if (typeName.startsWith('I') && typeName.length > 1 &&
+            typeName[1] === typeName[1]?.toUpperCase()) {
+            return false;
+        }
+        // Common external types
+        const externalPrefixes = [
+            'System.', 'Microsoft.', 'Newtonsoft.', 'AutoMapper.',
+            'Serilog.', 'NLog.', 'log4net.',
+        ];
+        return externalPrefixes.some(prefix => typeName.startsWith(prefix));
+    }
+}
+// ============================================================================
+// Factory
+// ============================================================================
+export function createCSharpTestExtractor(parser) {
+    return new CSharpTestExtractor(parser);
+}
+//# sourceMappingURL=csharp-test-extractor.js.map

package/dist/test-topology/extractors/csharp-test-extractor.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"csharp-test-extractor.js","sourceRoot":"","sources":["../../../src/test-topology/extractors/csharp-test-extractor.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAGH,OAAO,EAAE,iBAAiB,EAAE,MAAM,0BAA0B,CAAC;AAU7D,+EAA+E;AAC/E,sBAAsB;AACtB,+EAA+E;AAE/E,MAAM,oBAAoB,GAAkC;IAC1D,OAAO,EAAE,OAAO;IAChB,OAAO,EAAE,OAAO;IAChB,OAAO,EAAE,OAAO;IAChB,iBAAiB,EAAE,OAAO;IAC1B,8CAA8C,EAAE,QAAQ;CACzD,CAAC;AAEF,MAAM,oBAAoB,GAAkC;IAC1D,MAAM,EAAE,OAAO;IACf,QAAQ,EAAE,OAAO;IACjB,MAAM,EAAE,OAAO;IACf,UAAU,EAAE,OAAO;IACnB,YAAY,EAAE,QAAQ;IACtB,gBAAgB,EAAE,QAAQ;CAC3B,CAAC;AAEF,+EAA+E;AAC/E,2BAA2B;AAC3B,+EAA+E;AAE/E,MAAM,OAAO,mBAAoB,SAAQ,iBAAiB;IACxD,YAAY,MAAc;QACxB,KAAK,CAAC,MAAM,EAAE,QAAQ,CAAC,CAAC;IAC1B,CAAC;IAED,OAAO,CAAC,OAAe,EAAE,QAAgB;QACvC,MAAM,IAAI,GAAG,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;QACxC,MAAM,IAAI,GAAG,IAAI,CAAC,QAAQ,CAAC;QAE3B,MAAM,SAAS,GAAG,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;QAC7C,MAAM,SAAS,GAAG,IAAI,CAAC,gBAAgB,CAAC,IAAI,CAAC,CAAC;QAC9C,MAAM,KAAK,GAAG,IAAI,CAAC,YAAY,CAAC,IAAI,EAAE,SAAS,CAAC,CAAC;QACjD,MAAM,WAAW,GAAG,IAAI,CAAC,kBAAkB,CAAC,IAAI,CAAC,CAAC;QAElD,iCAAiC;QACjC,KAAK,MAAM,IAAI,IAAI,SAAS,EAAE,CAAC;YAC7B,MAAM,SAAS,GAAG,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CACjC,CAAC,CAAC,IAAI,IAAI,IAAI,CAAC,IAAI,IAAI,CAAC,CAAC,IAAI,IAAI,IAAI,CAAC,IAAI,GAAG,GAAG,CACjD,CAAC;YACF,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC,gBAAgB,CAAC,IAAI,CAAC,UAAU,EAAE,SAAS,EAAE,IAAI,CAAC,WAAW,CAAC,CAAC;QACrF,CAAC;QAED,OAAO;YACL,IAAI,EAAE,QAAQ;YACd,SAAS;YACT,QAAQ,EAAE,QAAQ;YAClB,SAAS;YACT,KAAK;YACL,WAAW;SACZ,CAAC;IACJ,CAAC;IAED,eAAe,CAAC,IAAuB;QACrC,MAAM,OAAO,GAAG,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC;QAEvC,yBAAyB;QACzB,KAAK,MAAM,GAAG,IAAI,OAAO,EAAE,CAAC;YAC1B,KAAK,MAAM,CAAC,OAAO,EAAE,SAAS,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,oBAAoB,CAAC,EAAE,CAAC;gBACxE,IAAI,GAAG,CAAC,QAAQ,CAAC,OAAO,CAAC,EAAE,CAAC;oBAC1B,OAAO,SAAS,CAAC;gBACnB,CAAC;YACH,CAAC;QACH,CAAC;QAED,wCAAwC;QACxC,MAAM,IAAI,GAAG,IAAI,CAAC,IAAI,CAAC;QACvB,KAAK,MAAM,CAAC,IAAI,EAAE,SAAS,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,oBAAoB,CAAC,EAAE,CAAC;YACrE,IAAI,IAAI,CAAC,QAAQ,CAAC,IAAI,IAAI,GAAG,CAAC,IAAI,IAAI,CAAC,QAAQ,CAAC,IAAI,IAAI,GAAG,CAAC,EAAE,CAAC;gBAC7D,OAAO,SAAS,CAAC;YACnB,CAAC;QACH,CAAC;QAED,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,gBAAgB,CAAC,IAAuB;QACtC,MAAM,SAAS,GAAe,EAAE,CAAC;QACjC,IAAI,YAAgC,CAAC;QAErC,IAAI,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC,IAAI,EAAE,EAAE;YAC3B,sBAAsB;YACtB,IAAI,IAAI,CAAC,IAAI,KAAK,mBAAmB,EAAE,CAAC;gBACtC,MAAM,QAAQ,GAAG,IAAI,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC;gBAChD,IAAI,QAAQ,EAAE,CAAC;oBACb,YAAY,GAAG,QAAQ,CAAC,IAAI,CAAC;gBAC/B,CAAC;YACH,CAAC;YAED,oCAAoC;YACpC,IAAI,IAAI,CAAC,IAAI,KAAK,oBAAoB,EAAE,CAAC;gBACvC,MAAM,aAAa,GAAG,IAAI,CAAC,iBAAiB,CAAC,IAAI,CAAC,CAAC;gBACnD,MAAM,QAAQ,GAAG,IAAI,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC;gBAChD,MAAM,QAAQ,GAAG,IAAI,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC;gBAEhD,IAAI,QAAQ,IAAI,QAAQ,IAAI,IAAI,CAAC,gBAAgB,CAAC,aAAa,CAAC,EAAE,CAAC;oBACjE,MAAM,IAAI,GAAG,QAAQ,CAAC,IAAI,CAAC;oBAC3B,MAAM,aAAa,GAAG,YAAY;wBAChC,CAAC,CAAC,GAAG,YAAY,IAAI,IAAI,EAAE;wBAC3B,CAAC,CAAC,IAAI,CAAC;oBAET,MAAM,WAAW,GAAG,IAAI,CAAC,oBAAoB,CAAC,QAAQ,CAAC,CAAC;oBACxD,MAAM,UAAU,GAAG,IAAI,CAAC,iBAAiB,CAAC,QAAQ,CAAC,CAAC;oBAEpD,SAAS,CAAC,IAAI,CAAC;wBACb,EAAE,EAAE,IAAI,CAAC,cAAc,CAAC,EAAE,EAAE,IAAI,EAAE,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC;wBACzD,IAAI;wBACJ,WAAW,EAAE,YAAY;wBACzB,aAAa;wBACb,IAAI,EAAE,EAAE;wBACR,IAAI,EAAE,IAAI,CAAC,aAAa,CAAC,GAAG,GAAG,CAAC;wBAChC,WAAW;wBACX,eAAe,EAAE,EAAE;wBACnB,UAAU;wBACV,OAAO,EAAE;4BACP,cAAc,EAAE,UAAU,CAAC,MAAM;4BACjC,aAAa,EAAE,KAAK;4BACpB,YAAY,EAAE,KAAK;4BACnB,SAAS,EAAE,CAAC;4BACZ,UAAU,EAAE,CAAC;4BACb,KAAK,EAAE,EAAE;yBACV;qBACF,CAAC,CAAC;gBACL,CAAC;YACH,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,YAAY,CAAC,IAAuB,EAAE,UAAyB;QAC7D,MAAM,KAAK,GAAoB,EAAE,CAAC;QAElC,IAAI,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC,IAAI,EAAE,EAAE;YAC3B,8BAA8B;YAC9B,IAAI,IAAI,CAAC,IAAI,KAAK,4BAA4B,EAAE,CAAC;gBAC/C,MAAM,QAAQ,GAAG,IAAI,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC;gBAEhD,IAAI,QAAQ,EAAE,CAAC;oBACb,MAAM,QAAQ,GAAG,QAAQ,CAAC,IAAI,CAAC;oBAE/B,iBAAiB;oBACjB,IAAI,QAAQ,CAAC,UAAU,CAAC,OAAO,CAAC,EAAE,CAAC;wBACjC,MAAM,UAAU,GAAG,QAAQ,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,+BAA+B;wBACzE,KAAK,CAAC,IAAI,CAAC;4BACT,MAAM,EAAE,UAAU;4BAClB,QAAQ,EAAE,SAAS;4BACnB,IAAI,EAAE,IAAI,CAAC,aAAa,CAAC,GAAG,GAAG,CAAC;4BAChC,UAAU,EAAE,IAAI,CAAC,oBAAoB,CAAC,UAAU,CAAC;yBAClD,CAAC,CAAC;oBACL,CAAC;oBAED,2CAA2C;oBAC3C,IAAI,QAAQ,CAAC,QAAQ,CAAC,YAAY,CAAC,EAAE,CAAC;wBACpC,KAAK,CAAC,IAAI,CAAC;4BACT,MAAM,EAAE,QAAQ;4BAChB,QAAQ,EAAE,aAAa;4BACvB,IAAI,EAAE,IAAI,CAAC,aAAa,CAAC,GAAG,GAAG,CAAC;4BAChC,UAAU,EAAE,KAAK;yBAClB,CAAC,CAAC;oBACL,CAAC;gBACH,CAAC;YACH,CAAC;YAED,4CAA4C;YAC5C,IAAI,IAAI,CAAC,IAAI,KAAK,uBAAuB,EAAE,CAAC;gBAC1C,MAAM,MAAM,GAAG,IAAI,CAAC,iBAAiB,CAAC,UAAU,CAAC,CAAC;gBAElD,IAAI,MAAM,EAAE,IAAI,KAAK,0BAA0B,EAAE,CAAC;oBAChD,MAAM,IAAI,GAAG,MAAM,CAAC,IAAI,CAAC;oBAEzB,IAAI,IAAI,CAAC,QAAQ,CAAC,gBAAgB,CAAC,EAAE,CAAC;wBACpC,MAAM,SAAS,GAAG,IAAI,CAAC,KAAK,CAAC,cAAc,CAAC,CAAC;wBAC7C,KAAK,CAAC,IAAI,CAAC;4BACT,MAAM,EAAE,SAAS,EAAE,CAAC,CAAC,CAAC,IAAI,SAAS;4BACnC,QAAQ,EAAE,mBAAmB;4BAC7B,IAAI,EAAE,IAAI,CAAC,aAAa,CAAC,GAAG,GAAG,CAAC;4BAChC,UAAU,EAAE,KAAK;yBAClB,CAAC,CAAC;oBACL,CAAC;oBAED,2BAA2B;oBAC3B,IAAI,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,CAAC;wBAC5B,MAAM,SAAS,GAAG,IAAI,CAAC,KAAK,CAAC,eAAe,CAAC,CAAC;wBAC9C,KAAK,CAAC,IAAI,CAAC;4BACT,MAAM,EAAE,SAAS,EAAE,CAAC,CAAC,CAAC,IAAI,SAAS;4BACnC,QAAQ,EAAE,WAAW;4BACrB,IAAI,EAAE,IAAI,CAAC,aAAa,CAAC,GAAG,GAAG,CAAC;4BAChC,UAAU,EAAE,KAAK;yBAClB,CAAC,CAAC;oBACL,CAAC;gBACH,CAAC;YACH,CAAC;YAED,gCAAgC;YAChC,IAAI,IAAI,CAAC,IAAI,KAAK,uBAAuB,EAAE,CAAC;gBAC1C,MAAM,MAAM,GAAG,IAAI,CAAC,iBAAiB,CAAC,UAAU,CAAC,CAAC;gBAElD,IAAI,MAAM,EAAE,IAAI,KAAK,0BAA0B,EAAE,CAAC;oBAChD,MAAM,QAAQ,GAAG,MAAM,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC;oBAElD,IAAI,QAAQ,EAAE,IAAI,KAAK,OAAO,IAAI,QAAQ,EAAE,IAAI,KAAK,UAAU;wBAC3D,QAAQ,EAAE,IAAI,KAAK,UAAU,IAAI,QAAQ,EAAE,IAAI,KAAK,eAAe,EAAE,CAAC;wBACxE,KAAK,CAAC,IAAI,CAAC;4BACT,MAAM,EAAE,YAAY;4BACpB,QAAQ,EAAE,OAAO,QAAQ,CAAC,IAAI,EAAE;4BAChC,IAAI,EAAE,IAAI,CAAC,aAAa,CAAC,GAAG,GAAG,CAAC;4BAChC,UAAU,EAAE,KAAK;4BACjB,iBAAiB,EAAE,IAAI;yBACxB,CAAC,CAAC;oBACL,CAAC;gBACH,CAAC;YACH,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,OAAO,KAAK,CAAC;IACf,CAAC;IAED,kBAAkB,CAAC,IAAuB;QACxC,MAAM,MAAM,GAAiB,EAAE,CAAC;QAEhC,IAAI,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC,IAAI,EAAE,EAAE;YAC3B,IAAI,IAAI,CAAC,IAAI,KAAK,oBAAoB,EAAE,CAAC;gBACvC,MAAM,aAAa,GAAG,IAAI,CAAC,iBAAiB,CAAC,IAAI,CAAC,CAAC;gBACnD,MAAM,QAAQ,GAAG,IAAI,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC;gBAEhD,IAAI,QAAQ,EAAE,CAAC;oBACb,IAAI,SAAS,GAA8B,IAAI,CAAC;oBAEhD,sCAAsC;oBACtC,KAAK,MAAM,IAAI,IAAI,aAAa,EAAE,CAAC;wBACjC,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC;wBAE3B,+DAA+D;wBAC/D,QAAQ;wBACR,IAAI,QAAQ,CAAC,QAAQ,CAAC,SAAS,CAAC;4BAAE,SAAS,GAAG,YAAY,CAAC;6BACtD,IAAI,QAAQ,CAAC,QAAQ,CAAC,YAAY,CAAC;4BAAE,SAAS,GAAG,WAAW,CAAC;6BAC7D,IAAI,QAAQ,CAAC,QAAQ,CAAC,gBAAgB,CAAC;4BAAE,SAAS,GAAG,WAAW,CAAC;6BACjE,IAAI,QAAQ,CAAC,QAAQ,CAAC,mBAAmB,CAAC;4BAAE,SAAS,GAAG,UAAU,CAAC;wBACxE,SAAS;6BACJ,IAAI,QAAQ,CAAC,QAAQ,CAAC,kBAAkB,CAAC;4BAAE,SAAS,GAAG,YAAY,CAAC;6BACpE,IAAI,QAAQ,CAAC,QAAQ,CAAC,eAAe,CAAC;4BAAE,SAAS,GAAG,WAAW,CAAC;6BAChE,IAAI,QAAQ,CAAC,QAAQ,CAAC,mBAAmB,CAAC;4BAAE,SAAS,GAAG,WAAW,CAAC;6BACpE,IAAI,QAAQ,CAAC,QAAQ,CAAC,gBAAgB,CAAC;4BAAE,SAAS,GAAG,UAAU,CAAC;oBACvE,CAAC;oBAED,IAAI,SAAS,EAAE,CAAC;wBACd,MAAM,KAAK,GAAG,IAAI,CAAC,oBAAoB,CAAC,QAAQ,CAAC,CAAC;wBAClD,MAAM,CAAC,IAAI,CAAC;4BACV,IAAI,EAAE,SAAS;4BACf,IAAI,EAAE,IAAI,CAAC,aAAa,CAAC,GAAG,GAAG,CAAC;4BAChC,KAAK;yBACN,CAAC,CAAC;oBACL,CAAC;gBACH,CAAC;YACH,CAAC;YAED,8BAA8B;YAC9B,IAAI,IAAI,CAAC,IAAI,KAAK,yBAAyB,EAAE,CAAC;gBAC5C,MAAM,QAAQ,GAAG,IAAI,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC;gBAChD,IAAI,QAAQ,EAAE,CAAC;oBACb,MAAM,KAAK,GAAG,IAAI,CAAC,oBAAoB,CAAC,QAAQ,CAAC,CAAC;oBAClD,MAAM,CAAC,IAAI,CAAC;wBACV,IAAI,EAAE,YAAY;wBAClB,IAAI,EAAE,IAAI,CAAC,aAAa,CAAC,GAAG,GAAG,CAAC;wBAChC,KAAK;qBACN,CAAC,CAAC;gBACL,CAAC;YACH,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,OAAO,MAAM,CAAC;IAChB,CAAC;IAES,WAAW,CAAC,IAAuB;QAC3C,MAAM,OAAO,GAAa,EAAE,CAAC;QAE7B,IAAI,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC,IAAI,EAAE,EAAE;YAC3B,IAAI,IAAI,CAAC,IAAI,KAAK,iBAAiB,EAAE,CAAC;gBACpC,MAAM,QAAQ,GAAG,IAAI,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC;gBAChD,IAAI,QAAQ,EAAE,CAAC;oBACb,OAAO,CAAC,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;gBAC9B,CAAC;YACH,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,OAAO,OAAO,CAAC;IACjB,CAAC;IAEO,iBAAiB,CAAC,IAAuB;QAC/C,MAAM,UAAU,GAAwB,EAAE,CAAC;QAE3C,qDAAqD;QACrD,IAAI,OAAO,GAAG,IAAI,CAAC,oBAAoB,CAAC;QACxC,OAAO,OAAO,IAAI,OAAO,CAAC,IAAI,KAAK,gBAAgB,EAAE,CAAC;YACpD,UAAU,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YACzB,OAAO,GAAG,OAAO,CAAC,oBAAoB,CAAC;QACzC,CAAC;QAED,sBAAsB;QACtB,IAAI,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC,KAAK,EAAE,EAAE;YAC5B,IAAI,KAAK,CAAC,IAAI,KAAK,gBAAgB,EAAE,CAAC;gBACpC,UAAU,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACzB,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,OAAO,UAAU,CAAC;IACpB,CAAC;IAEO,gBAAgB,CAAC,aAAkC;QACzD,MAAM,cAAc,GAAG,CAAC,MAAM,EAAE,QAAQ,EAAE,MAAM,EAAE,UAAU,EAAE,YAAY,EAAE,gBAAgB,CAAC,CAAC;QAE9F,KAAK,MAAM,QAAQ,IAAI,aAAa,EAAE,CAAC;YACrC,MAAM,IAAI,GAAG,QAAQ,CAAC,IAAI,CAAC;YAC3B,KAAK,MAAM,IAAI,IAAI,cAAc,EAAE,CAAC;gBAClC,IAAI,IAAI,CAAC,QAAQ,CAAC,IAAI,IAAI,GAAG,CAAC,IAAI,IAAI,CAAC,QAAQ,CAAC,IAAI,IAAI,GAAG,CAAC,EAAE,CAAC;oBAC7D,OAAO,IAAI,CAAC;gBACd,CAAC;YACH,CAAC;QACH,CAAC;QAED,OAAO,KAAK,CAAC;IACf,CAAC;IAEO,iBAAiB,CAAC,IAAuB;QAC/C,MAAM,UAAU,GAAoB,EAAE,CAAC;QAEvC,IAAI,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC,KAAK,EAAE,EAAE;YAC5B,IAAI,KAAK,CAAC,IAAI,KAAK,uBAAuB,EAAE,CAAC;gBAC3C,MAAM,MAAM,GAAG,KAAK,CAAC,iBAAiB,CAAC,UAAU,CAAC,CAAC;gBAEnD,IAAI,MAAM,EAAE,IAAI,KAAK,0BAA0B,EAAE,CAAC;oBAChD,MAAM,OAAO,GAAG,MAAM,CAAC,iBAAiB,CAAC,YAAY,CAAC,CAAC;oBACvD,MAAM,QAAQ,GAAG,MAAM,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC;oBAElD,IAAI,OAAO,IAAI,QAAQ,EAAE,CAAC;wBACxB,MAAM,GAAG,GAAG,OAAO,CAAC,IAAI,CAAC;wBACzB,MAAM,MAAM,GAAG,QAAQ,CAAC,IAAI,CAAC;wBAE7B,eAAe;wBACf,IAAI,GAAG,KAAK,QAAQ,EAAE,CAAC;4BACrB,UAAU,CAAC,IAAI,CAAC;gCACd,OAAO,EAAE,UAAU,MAAM,EAAE;gCAC3B,IAAI,EAAE,KAAK,CAAC,aAAa,CAAC,GAAG,GAAG,CAAC;gCACjC,gBAAgB,EAAE,MAAM,KAAK,QAAQ,IAAI,MAAM,KAAK,aAAa;oCAChD,MAAM,KAAK,WAAW,IAAI,MAAM,KAAK,gBAAgB;gCACtE,mBAAmB,EAAE,MAAM,KAAK,MAAM,IAAI,MAAM,KAAK,SAAS;oCAC1C,MAAM,KAAK,OAAO,IAAI,MAAM,KAAK,UAAU;6BAChE,CAAC,CAAC;wBACL,CAAC;wBAED,eAAe;wBACf,IAAI,GAAG,KAAK,QAAQ,IAAI,GAAG,CAAC,QAAQ,CAAC,SAAS,CAAC,EAAE,CAAC;4BAChD,UAAU,CAAC,IAAI,CAAC;gCACd,OAAO,EAAE,MAAM;gCACf,IAAI,EAAE,KAAK,CAAC,aAAa,CAAC,GAAG,GAAG,CAAC;gCACjC,gBAAgB,EAAE,MAAM,KAAK,QAAQ,IAAI,MAAM,KAAK,aAAa;oCAChD,MAAM,KAAK,OAAO,IAAI,MAAM,KAAK,YAAY;gCAC9D,mBAAmB,EAAE,MAAM,KAAK,QAAQ,IAAI,MAAM,KAAK,WAAW;oCAC9C,MAAM,KAAK,SAAS,IAAI,MAAM,KAAK,YAAY;6BACpE,CAAC,CAAC;wBACL,CAAC;wBAED,8BAA8B;wBAC9B,IAAI,MAAM,KAAK,QAAQ,EAAE,CAAC;4BACxB,UAAU,CAAC,IAAI,CAAC;gCACd,OAAO,EAAE,QAAQ;gCACjB,IAAI,EAAE,KAAK,CAAC,aAAa,CAAC,GAAG,GAAG,CAAC;gCACjC,gBAAgB,EAAE,KAAK;gCACvB,mBAAmB,EAAE,KAAK;6BAC3B,CAAC,CAAC;wBACL,CAAC;wBAED,aAAa;wBACb,IAAI,MAAM,KAAK,QAAQ,IAAI,MAAM,KAAK,WAAW,IAAI,MAAM,KAAK,oBAAoB,EAAE,CAAC;4BACrF,UAAU,CAAC,IAAI,CAAC;gCACd,OAAO,EAAE,QAAQ,MAAM,EAAE;gCACzB,IAAI,EAAE,KAAK,CAAC,aAAa,CAAC,GAAG,GAAG,CAAC;gCACjC,gBAAgB,EAAE,KAAK;gCACvB,mBAAmB,EAAE,KAAK;6BAC3B,CAAC,CAAC;wBACL,CAAC;oBACH,CAAC;gBACH,CAAC;YACH,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,OAAO,UAAU,CAAC;IACpB,CAAC;IAEO,oBAAoB,CAAC,QAAgB;QAC3C,mDAAmD;QACnD,IAAI,QAAQ,CAAC,UAAU,CAAC,GAAG,CAAC,IAAI,QAAQ,CAAC,MAAM,GAAG,CAAC;YAC/C,QAAQ,CAAC,CAAC,CAAC,KAAK,QAAQ,CAAC,CAAC,CAAC,EAAE,WAAW,EAAE,EAAE,CAAC;YAC/C,OAAO,KAAK,CAAC;QACf,CAAC;QAED,wBAAwB;QACxB,MAAM,gBAAgB,GAAG;YACvB,SAAS,EAAE,YAAY,EAAE,aAAa,EAAE,aAAa;YACrD,UAAU,EAAE,OAAO,EAAE,UAAU;SAChC,CAAC;QAEF,OAAO,gBAAgB,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,QAAQ,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,CAAC;IACtE,CAAC;CACF;AAED,+EAA+E;AAC/E,UAAU;AACV,+EAA+E;AAE/E,MAAM,UAAU,yBAAyB,CAAC,MAAc;IACtD,OAAO,IAAI,mBAAmB,CAAC,MAAM,CAAC,CAAC;AACzC,CAAC"}

package/dist/test-topology/extractors/go-test-extractor.d.ts

@@ -0,0 +1,33 @@
+/**
+ * Go Test Extractor
+ *
+ * Extracts test information from Go testing frameworks:
+ * - Standard library testing package
+ * - testify (assert, require, suite)
+ * - ginkgo/gomega
+ * - gomock
+ */
+import type Parser from 'tree-sitter';
+import { BaseTestExtractor } from './base-test-extractor.js';
+import type { TestExtraction, TestCase, MockStatement, SetupBlock, TestFramework } from '../types.js';
+export declare class GoTestExtractor extends BaseTestExtractor {
+    constructor(parser: Parser);
+    extract(content: string, filePath: string): TestExtraction;
+    detectFramework(root: Parser.SyntaxNode): TestFramework;
+    extractTestCases(root: Parser.SyntaxNode): TestCase[];
+    private hasTestingTParam;
+    private hasTestingBParam;
+    extractMocks(root: Parser.SyntaxNode, _framework: TestFramework): MockStatement[];
+    extractSetupBlocks(root: Parser.SyntaxNode): SetupBlock[];
+    protected findImports(root: Parser.SyntaxNode): string[];
+    private extractAssertions;
+    /**
+     * Override to add Go-specific test framework calls
+     */
+    protected isTestFrameworkCall(name: string): boolean;
+}
+/**
+ * Create a Go test extractor instance
+ */
+export declare function createGoTestExtractor(parser: Parser): GoTestExtractor;
+//# sourceMappingURL=go-test-extractor.d.ts.map

package/dist/test-topology/extractors/go-test-extractor.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"go-test-extractor.d.ts","sourceRoot":"","sources":["../../../src/test-topology/extractors/go-test-extractor.ts"],"names":[],"mappings":"AAAA;;;;;;;;GAQG;AAEH,OAAO,KAAK,MAAM,MAAM,aAAa,CAAC;AACtC,OAAO,EAAE,iBAAiB,EAAE,MAAM,0BAA0B,CAAC;AAC7D,OAAO,KAAK,EACV,cAAc,EACd,QAAQ,EACR,aAAa,EACb,UAAU,EAEV,aAAa,EACd,MAAM,aAAa,CAAC;AAqBrB,qBAAa,eAAgB,SAAQ,iBAAiB;gBACxC,MAAM,EAAE,MAAM;IAI1B,OAAO,CAAC,OAAO,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,GAAG,cAAc;IA0B1D,eAAe,CAAC,IAAI,EAAE,MAAM,CAAC,UAAU,GAAG,aAAa;IAoBvD,gBAAgB,CAAC,IAAI,EAAE,MAAM,CAAC,UAAU,GAAG,QAAQ,EAAE;IAsHrD,OAAO,CAAC,gBAAgB;IAkBxB,OAAO,CAAC,gBAAgB;IAiBxB,YAAY,CAAC,IAAI,EAAE,MAAM,CAAC,UAAU,EAAE,UAAU,EAAE,aAAa,GAAG,aAAa,EAAE;IAyDjF,kBAAkB,CAAC,IAAI,EAAE,MAAM,CAAC,UAAU,GAAG,UAAU,EAAE;IAmEzD,SAAS,CAAC,WAAW,CAAC,IAAI,EAAE,MAAM,CAAC,UAAU,GAAG,MAAM,EAAE;IAexD,OAAO,CAAC,iBAAiB;IAoEzB;;OAEG;cACgB,mBAAmB,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO;CAiD9D;AAED;;GAEG;AACH,wBAAgB,qBAAqB,CAAC,MAAM,EAAE,MAAM,GAAG,eAAe,CAErE"}