From c4ce80353ef1be399c6167dac43101e684c362bb Mon Sep 17 00:00:00 2001
From: Christopher Dwyer-Perkins <chris@inverted-solutions.com>
Date: Tue, 26 Mar 2024 11:46:09 -0300
Subject: [PATCH 1/3] Fixed a bug where scope validations where incomplete for
 node tests

---
 bsc-plugin/src/lib/rooibos/FileFactory.ts    | 10 ++++--
 bsc-plugin/src/lib/rooibos/RooibosSession.ts | 14 +++------
 bsc-plugin/src/lib/rooibos/TestSuite.ts      |  7 ++++-
 bsc-plugin/src/plugin.spec.ts                | 33 +++++++++++++++++++-
 bsc-plugin/src/plugin.ts                     | 31 +++++++++++++-----
 5 files changed, 74 insertions(+), 21 deletions(-)

diff --git a/bsc-plugin/src/lib/rooibos/FileFactory.ts b/bsc-plugin/src/lib/rooibos/FileFactory.ts
index 82ff914d..24b16c59 100644
--- a/bsc-plugin/src/lib/rooibos/FileFactory.ts
+++ b/bsc-plugin/src/lib/rooibos/FileFactory.ts
@@ -3,6 +3,7 @@ import { standardizePath as s } from 'brighterscript';
 import * as path from 'path';
 import * as fs from 'fs';
 import * as fse from 'fs-extra';
+import type { TestSuite } from './TestSuite';
 
 export class FileFactory {
     private coverageComponentXmlTemplate;
@@ -71,10 +72,15 @@ export class FileFactory {
         );
     }
 
-    public createTestXML(name: string, baseName: string, useBs = true): string {
+    public createTestXML(name: string, baseName: string, suite?: TestSuite): string {
         let scriptImports = [];
         for (let fileName of this.frameworkFileNames) {
-            scriptImports.push(`<script type="text/bright${useBs ? 'er' : ''}script" uri="pkg:/${this.targetPath}${fileName}.${useBs ? 'bs' : 'brs'}" />`);
+            scriptImports.push(`<script type="text/brighterscript" uri="pkg:/${this.targetPath}${fileName}.bs" />`);
+        }
+
+        // Add the test spec file rather then relying on auto imports
+        if (suite) {
+            scriptImports.push(`<script type="text/brighterscript" uri="pkg:/${suite.file.pkgPath.replace(/\\/g, '/')}" />`);
         }
 
         let contents = `<?xml version="1.0" encoding="UTF-8" ?>
diff --git a/bsc-plugin/src/lib/rooibos/RooibosSession.ts b/bsc-plugin/src/lib/rooibos/RooibosSession.ts
index bbbac930..1b44dd86 100644
--- a/bsc-plugin/src/lib/rooibos/RooibosSession.ts
+++ b/bsc-plugin/src/lib/rooibos/RooibosSession.ts
@@ -199,18 +199,14 @@ export class RooibosSession {
     }
 
     createNodeFile(program: Program, suite: TestSuite) {
-        let p = path.join('components', 'rooibos', 'generated');
-
         let xmlText = this.getNodeTestXmlText(suite);
-        let bsPath = path.join(p, `${suite.generatedNodeName}.bs`);
-        this.fileFactory.addFile(program, path.join(p, `${suite.generatedNodeName}.xml`), xmlText);
-        let bsFile = program.getFile(bsPath);
+        this.fileFactory.addFile(program, suite.xmlPkgPath, xmlText);
+        let bsFile = program.getFile(suite.bsPkgPath);
         if (bsFile) {
             (bsFile as BrsFile).parser.statements.push();
             bsFile.needsTranspiled = true;
         }
-        let brsFile = this.fileFactory.addFile(program, bsPath, undent`
-            import "pkg:/${suite.file.pkgPath}"
+        let brsFile = this.fileFactory.addFile(program, suite.bsPkgPath, undent`
             function init()
                 nodeRunner = Rooibos_TestRunner(m.top.getScene(), m)
                 m.top.rooibosTestResult = nodeRunner.runInNodeMode("${suite.name}")
@@ -219,8 +215,8 @@ export class RooibosSession {
         brsFile.parser.invalidateReferences();
     }
 
-    private getNodeTestXmlText(suite: TestSuite) {
-        return this.fileFactory.createTestXML(suite.generatedNodeName, suite.nodeName);
+    public getNodeTestXmlText(suite: TestSuite) {
+        return this.fileFactory.createTestXML(suite.generatedNodeName, suite.nodeName, suite);
     }
 
     private getNamespaceLookup(scope: Scope): Map<string, NamespaceContainer> {
diff --git a/bsc-plugin/src/lib/rooibos/TestSuite.ts b/bsc-plugin/src/lib/rooibos/TestSuite.ts
index 724fdd51..2838434f 100644
--- a/bsc-plugin/src/lib/rooibos/TestSuite.ts
+++ b/bsc-plugin/src/lib/rooibos/TestSuite.ts
@@ -1,3 +1,4 @@
+import * as  path from 'path';
 import type { AstEditor, BrsFile, ClassStatement } from 'brighterscript';
 
 import { diagnosticNodeTestIllegalNode, diagnosticNodeTestRequiresNode } from '../utils/Diagnostics';
@@ -58,6 +59,8 @@ export class TestBlock {
     public tearDownFunctionName: string;
     public beforeEachFunctionName: string;
     public afterEachFunctionName: string;
+    public xmlPkgPath: string;
+    public bsPkgPath: string;
 
 }
 
@@ -71,7 +74,9 @@ export class TestSuite extends TestBlock {
             this.annotation.name = classStatement.name.text;
         }
         this.generatedNodeName = (this.name || 'ERROR').replace(/[^a-zA-Z0-9]/g, '_');
-
+        let pathBase = path.join('components', 'rooibos', 'generated');
+        this.xmlPkgPath = path.join(pathBase, this.generatedNodeName + '.xml');
+        this.bsPkgPath = path.join(pathBase, this.generatedNodeName + '.bs');
     }
 
     //state
diff --git a/bsc-plugin/src/plugin.spec.ts b/bsc-plugin/src/plugin.spec.ts
index c4d66044..fa69977a 100644
--- a/bsc-plugin/src/plugin.spec.ts
+++ b/bsc-plugin/src/plugin.spec.ts
@@ -26,9 +26,9 @@ describe('RooibosPlugin', () => {
                 isGlobalMethodMockingEnabled: true
             }
         };
+        fsExtra.emptyDirSync(tmpPath);
         fsExtra.ensureDirSync(_stagingFolderPath);
         fsExtra.ensureDirSync(_rootDir);
-        fsExtra.ensureDirSync(tmpPath);
 
         builder = new ProgramBuilder();
         // eslint-disable-next-line @typescript-eslint/no-unsafe-argument
@@ -2013,6 +2013,37 @@ describe('RooibosPlugin', () => {
         });
     });
 
+    describe('does not break class extending', () => {
+        it('does not prevent valid scope based diagnostics for node tests', () => {
+            program.setFile('components/customComponent.xml', `
+                <component name="CustomComponent" extends="Group" />
+            `);
+            program.setFile('source/baseTestClass.spec.bs', `
+                class BaseTestClass extends rooibos.BaseTestSuite
+                    public function customHelperFunction() as boolean
+                        return true
+                    end function
+                end class
+            `);
+            program.setFile('components/test2.spec.bs', `
+                @suite
+                @SGNode("CustomComponent")
+                class ATest2 extends BaseTestClass
+                    @describe("groupA")
+                    @it("test1")
+                    function _()
+                        item = {id: "item"}
+                        m.expectNotCalled(item.getFunction())
+                        m.expectNotCalled(item.getFunction())
+                    end function
+                end class
+            `);
+            program.validate();
+            let files = [...Object.values(program.files)].map(x => ({ src: x.srcPath, dest: x.pkgPath }));
+            expect(program.getDiagnostics().map(x=> x.message)).to.eql([`Cannot find name 'BaseTestClass'`]);
+        });
+    });
+
     describe('addTestRunnerMetadata', () => {
         it('does not permanently modify the AST', async () => {
             program.setFile('source/test.spec.bs', `
diff --git a/bsc-plugin/src/plugin.ts b/bsc-plugin/src/plugin.ts
index cdac2fbe..027bebd0 100644
--- a/bsc-plugin/src/plugin.ts
+++ b/bsc-plugin/src/plugin.ts
@@ -6,12 +6,7 @@ import type {
     TranspileObj,
     AstEditor,
     BeforeFileTranspileEvent,
-    PluginHandler,
-    ClassStatement,
-    FunctionStatement,
-    Statement,
-    Scope,
-    BrsFile
+    XmlFile
 } from 'brighterscript';
 import {
     isBrsFile
@@ -21,6 +16,7 @@ import { CodeCoverageProcessor } from './lib/rooibos/CodeCoverageProcessor';
 import { FileFactory } from './lib/rooibos/FileFactory';
 import type { RooibosConfig } from './lib/rooibos/RooibosConfig';
 import * as minimatch from 'minimatch';
+import * as path from 'path';
 import { MockUtil } from './lib/rooibos/MockUtil';
 import { getScopeForSuite } from './lib/rooibos/Utils';
 
@@ -119,6 +115,15 @@ export class RooibosPlugin implements CompilerPlugin {
         this.fileFactory.addFrameworkFiles(program);
     }
 
+    afterFileDispose(file: BscFile) {
+        // eslint-disable-next-line @typescript-eslint/dot-notation
+        const xmlFile = file['rooibosXmlFile'] as XmlFile;
+        if (xmlFile) {
+            // Remove the old generated xml files
+            this._builder.program.removeFile(xmlFile.srcPath);
+        }
+    }
+
     afterFileParse(file: BscFile): void {
         // console.log('afp', file.pkgPath);
         if (file.pathAbsolute.includes('/rooibos/bsc-plugin/dist/framework')) {
@@ -129,10 +134,20 @@ export class RooibosPlugin implements CompilerPlugin {
         if (this.fileFactory.isIgnoredFile(file) || !this.shouldSearchInFileForTests(file)) {
             return;
         }
-
         // console.log('processing ', file.pkgPath);
+
         if (isBrsFile(file)) {
-            this.session.processFile(file);
+            // Add the node test component so brighter script can validate the test files
+            let suites = this.session.processFile(file);
+            let nodeSuites = suites.filter((ts) => ts.isNodeTest);
+            for (const suite of nodeSuites) {
+                const xmlFile = this._builder.program.setFile({
+                    src: path.resolve(suite.xmlPkgPath),
+                    dest: suite.xmlPkgPath
+                }, this.session.getNodeTestXmlText(suite));
+                // eslint-disable-next-line @typescript-eslint/dot-notation
+                file['rooibosXmlFile'] = xmlFile;
+            }
         }
     }
 

From 685377c9624760dbae0f59ac25ff25ab89117a84 Mon Sep 17 00:00:00 2001
From: Christopher Dwyer-Perkins <chris@inverted-solutions.com>
Date: Tue, 11 Jun 2024 10:30:50 -0300
Subject: [PATCH 2/3] Update bsc-plugin/src/plugin.spec.ts

---
 bsc-plugin/src/plugin.spec.ts | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/bsc-plugin/src/plugin.spec.ts b/bsc-plugin/src/plugin.spec.ts
index fa69977a..b504dde4 100644
--- a/bsc-plugin/src/plugin.spec.ts
+++ b/bsc-plugin/src/plugin.spec.ts
@@ -2013,7 +2013,7 @@ describe('RooibosPlugin', () => {
         });
     });
 
-    describe('does not break class extending', () => {
+    describe('does not prevent component scope validation of node tests', () => {
         it('does not prevent valid scope based diagnostics for node tests', () => {
             program.setFile('components/customComponent.xml', `
                 <component name="CustomComponent" extends="Group" />

From 2e54661dff7986f2c277a7924c7734a73a5e5cc9 Mon Sep 17 00:00:00 2001
From: Bronley Plumb <bronley@gmail.com>
Date: Tue, 2 Jul 2024 12:30:57 -0400
Subject: [PATCH 3/3] Fix lint issues.

---
 bsc-plugin/src/lib/rooibos/TestSuite.ts | 2 +-
 bsc-plugin/src/plugin.spec.ts           | 3 +--
 2 files changed, 2 insertions(+), 3 deletions(-)

diff --git a/bsc-plugin/src/lib/rooibos/TestSuite.ts b/bsc-plugin/src/lib/rooibos/TestSuite.ts
index 2838434f..076dc6b5 100644
--- a/bsc-plugin/src/lib/rooibos/TestSuite.ts
+++ b/bsc-plugin/src/lib/rooibos/TestSuite.ts
@@ -1,4 +1,4 @@
-import * as  path from 'path';
+import * as path from 'path';
 import type { AstEditor, BrsFile, ClassStatement } from 'brighterscript';
 
 import { diagnosticNodeTestIllegalNode, diagnosticNodeTestRequiresNode } from '../utils/Diagnostics';
diff --git a/bsc-plugin/src/plugin.spec.ts b/bsc-plugin/src/plugin.spec.ts
index e079c25c..dbe5bcd4 100644
--- a/bsc-plugin/src/plugin.spec.ts
+++ b/bsc-plugin/src/plugin.spec.ts
@@ -2019,7 +2019,7 @@ describe('RooibosPlugin', () => {
             `);
             program.validate();
             let files = [...Object.values(program.files)].map(x => ({ src: x.srcPath, dest: x.pkgPath }));
-            expect(program.getDiagnostics().map(x=> x.message)).to.eql([`Cannot find name 'BaseTestClass'`]);
+            expect(program.getDiagnostics().map(x => x.message)).to.eql([`Cannot find name 'BaseTestClass'`]);
         });
     });
 
@@ -2366,4 +2366,3 @@ function getTestSubContents() {
     );
     return result;
 }
-