brighterscript 1.0.0-alpha.14 → 1.0.0-alpha.17

package/dist/files/BrsFile.spec.js

@@ -2,7 +2,6 @@
 Object.defineProperty(exports, "__esModule", { value: true });
 const chai_1 = require("chai");
 const sinonImport = require("sinon");
-const path = require("path");
 const vscode_languageserver_1 = require("vscode-languageserver");
 const Program_1 = require("../Program");
 const BooleanType_1 = require("../types/BooleanType");
@@ -20,19 +19,24 @@ const PluginInterface_1 = require("../PluginInterface");
 const testHelpers_spec_1 = require("../testHelpers.spec");
 const Parser_1 = require("../parser/Parser");
 const Logger_1 = require("../Logger");
-const VoidType_1 = require("../types/VoidType");
+const Statement_1 = require("../parser/Statement");
+const creators_1 = require("../astUtils/creators");
+const fsExtra = require("fs-extra");
+const ArrayType_1 = require("../types/ArrayType");
 const FloatType_1 = require("../types/FloatType");
 const ObjectType_1 = require("../types/ObjectType");
-const ArrayType_1 = require("../types/ArrayType");
+const VoidType_1 = require("../types/VoidType");
 let sinon = sinonImport.createSandbox();
 describe('BrsFile', () => {
-    let rootDir = (0, util_1.standardizePath) `${process.cwd()}/.tmp/rootDir`;
+    let tempDir = (0, util_1.standardizePath) `${process.cwd()}/.tmp`;
+    let rootDir = (0, util_1.standardizePath) `${tempDir}/rootDir`;
     let program;
     let srcPath = (0, util_1.standardizePath) `${rootDir}/source/main.brs`;
     let destPath = 'source/main.brs';
     let file;
     let testTranspile = (0, testHelpers_spec_1.getTestTranspile)(() => [program, rootDir]);
     beforeEach(() => {
+        fsExtra.emptyDirSync(tempDir);
         program = new Program_1.Program({ rootDir: rootDir, sourceMap: true });
         file = new BrsFile_1.BrsFile(srcPath, destPath, program);
     });
@@ -64,6 +68,14 @@ describe('BrsFile', () => {
         //BrighterScript
         (0, chai_1.expect)(new BrsFile_1.BrsFile(`${rootDir}/source/main.bs`, 'source/main.bs', program).needsTranspiled).to.be.true;
     });
+    it('computes new import statements after clearing parser references', () => {
+        const file = program.setFile('source/main.bs', ``);
+        (0, chai_1.expect)(file.ownScriptImports).to.be.empty;
+        file.parser.ast.statements.push(new Statement_1.ImportStatement((0, creators_1.createToken)(TokenKind_1.TokenKind.Import), (0, creators_1.createToken)(TokenKind_1.TokenKind.StringLiteral, 'pkg:/source/lib.brs')));
+        (0, chai_1.expect)(file.ownScriptImports).to.be.empty;
+        file.parser.invalidateReferences();
+        (0, chai_1.expect)(file.ownScriptImports.map(x => x.text)).to.eql(['pkg:/source/lib.brs']);
+    });
     it('allows adding diagnostics', () => {
         const expected = [{
                 message: 'message',
@@ -158,6 +170,76 @@ describe('BrsFile', () => {
             (0, chai_1.expect)(names).to.includes('Main');
             (0, chai_1.expect)(names).to.includes('SayHello');
         });
+        it('includes every type of item at base level', () => {
+            program.setFile('source/main.bs', `
+                sub main()
+                    print
+                end sub
+                sub speak()
+                end sub
+                namespace stuff
+                end namespace
+                class Person
+                end class
+                enum Direction
+                end enum
+            `);
+            (0, testHelpers_spec_1.expectCompletionsIncludes)(program.getCompletions('source/main.bs', util_1.default.createPosition(2, 26)), [{
+                    label: 'main',
+                    kind: vscode_languageserver_1.CompletionItemKind.Function
+                }, {
+                    label: 'speak',
+                    kind: vscode_languageserver_1.CompletionItemKind.Function
+                }, {
+                    label: 'stuff',
+                    kind: vscode_languageserver_1.CompletionItemKind.Module
+                }, {
+                    label: 'Person',
+                    kind: vscode_languageserver_1.CompletionItemKind.Class
+                }, {
+                    label: 'Direction',
+                    kind: vscode_languageserver_1.CompletionItemKind.Enum
+                }]);
+        });
+        describe('namespaces', () => {
+            it('gets full namespace completions at any point through the leading identifier', () => {
+                program.setFile('source/main.bs', `
+                    sub main()
+                        foo.bar
+                    end sub
+
+                    namespace foo.bar
+                    end namespace
+
+                    class Person
+                    end class
+                `);
+                const result = program.getCompletions(`${rootDir}/source/main.bs`, vscode_languageserver_1.Position.create(2, 24)).map(x => x.label);
+                (0, chai_1.expect)(result).includes('main');
+                (0, chai_1.expect)(result).includes('foo');
+                (0, chai_1.expect)(result).includes('Person');
+            });
+            it('gets namespace completions', () => {
+                program.setFile('source/main.bs', `
+                    namespace foo.bar
+                        function sayHello()
+                        end function
+                    end namespace
+
+                    sub Main()
+                        print "hello"
+                        foo.ba
+                        foo.bar.
+                    end sub
+                `);
+                let result = program.getCompletions(`${rootDir}/source/main.bs`, vscode_languageserver_1.Position.create(8, 30));
+                let names = result.map(x => x.label);
+                (0, chai_1.expect)(names).to.includes('bar');
+                result = program.getCompletions(`${rootDir}/source/main.bs`, vscode_languageserver_1.Position.create(9, 32));
+                names = result.map(x => x.label);
+                (0, chai_1.expect)(names).to.includes('sayHello');
+            });
+        });
         it('always includes `m`', () => {
             //eslint-disable-next-line @typescript-eslint/no-floating-promises
             program.setFile({ src: `${rootDir}/source/main.brs`, dest: 'source/main.brs' }, `
@@ -354,7 +436,7 @@ describe('BrsFile', () => {
                         'bs:disable-next-line: LINT9999
                         name = "bob
                     end sub
-                `);
+                    `);
                 (0, chai_1.expect)(file.commentFlags[0]).to.exist;
                 (0, testHelpers_spec_1.expectHasDiagnostics)(program);
             });
@@ -364,7 +446,7 @@ describe('BrsFile', () => {
                     sub main()
                         something = true 'bs:disable-line: LINT1005
                     end sub
-                `);
+                    `);
                 file.addDiagnostics([{
                         code: 'LINT1005',
                         file: file,
@@ -375,18 +457,18 @@ describe('BrsFile', () => {
                 (0, testHelpers_spec_1.expectZeroDiagnostics)(scope);
             });
             it('adds diagnostics for unknown numeric diagnostic codes', () => {
-                program.setFile({ src: `${rootDir}/source/main.brs`, dest: 'source/main.brs' }, `
+                program.setFile({ src: `${rootDir} / source / main.brs`, dest: 'source/main.brs' }, `
                     sub main()
                         print "hi" 'bs:disable-line: 123456 999999   aaaab
                     end sub
-                `);
+                    `);
                 program.validate();
                 (0, testHelpers_spec_1.expectDiagnostics)(program, [Object.assign(Object.assign({}, DiagnosticMessages_1.DiagnosticMessages.unknownDiagnosticCode(123456)), { range: vscode_languageserver_1.Range.create(2, 53, 2, 59) }), Object.assign(Object.assign({}, DiagnosticMessages_1.DiagnosticMessages.unknownDiagnosticCode(999999)), { range: vscode_languageserver_1.Range.create(2, 60, 2, 66) })]);
             });
         });
         describe('bs:disable-line', () => {
             it('works for all', () => {
-                let file = program.setFile({ src: `${rootDir}/source/main.brs`, dest: 'source/main.brs' }, `
+                let file = program.setFile({ src: `${rootDir} / source / main.brs`, dest: 'source/main.brs' }, `
                     sub Main()
                         z::;;%%%%%% 'bs:disable-line
                     end sub
@@ -402,7 +484,7 @@ describe('BrsFile', () => {
                 (0, testHelpers_spec_1.expectZeroDiagnostics)(program);
             });
             it('works for specific codes', () => {
-                program.setFile({ src: `${rootDir}/source/main.brs`, dest: 'source/main.brs' }, `
+                program.setFile({ src: `${rootDir} /source/main.brs`, dest: 'source/main.brs' }, `
                     sub main()
                         'should not have any errors
                         DoSomething(1) 'bs:disable-line:1002
@@ -421,12 +503,12 @@ describe('BrsFile', () => {
                 //the current version of BRS causes parse errors after the `parse` keyword, showing error in comments
                 //the program should ignore all diagnostics found in brs:* comment lines EXCEPT
                 //for the diagnostics about using unknown error codes
-                program.setFile({ src: `${rootDir}/source/main.brs`, dest: 'source/main.brs' }, `
+                program.setFile({ src: `${rootDir} /source/main.brs`, dest: 'source/main.brs' }, `
                     sub main()
                         stop 'bs:disable-line
                         print "need a valid line to fix stop error"
                     end sub
-                `);
+                    `);
                 program.validate();
                 (0, testHelpers_spec_1.expectZeroDiagnostics)(program);
             });
@@ -436,9 +518,9 @@ describe('BrsFile', () => {
         it('supports iife in assignment', () => {
             program.setFile('source/main.brs', `
                 sub main()
-                    result = sub()
+                result = sub()
                     end sub()
-                    result = (sub()
+                result = (sub()
                     end sub)()
                 end sub
             `);
@@ -457,26 +539,26 @@ describe('BrsFile', () => {
             testParseMode('source/main.spec.bs', Parser_1.ParseMode.BrighterScript);
         });
         it('supports labels and goto statements', () => {
-            let file = program.setFile({ src: `${rootDir}/source/main.brs`, dest: 'source/main.brs' }, `
+            let file = program.setFile({ src: `${rootDir} /source/main.brs`, dest: 'source/main.brs' }, `
                 sub Main()
                     'multiple goto statements on one line
-                    goto myLabel : goto myLabel
+                    goto myLabel: goto myLabel
                     myLabel:
                 end sub
             `);
             (0, testHelpers_spec_1.expectZeroDiagnostics)(file);
         });
         it('supports empty print statements', () => {
-            let file = program.setFile({ src: `${rootDir}/source/main.brs`, dest: 'source/main.brs' }, `
+            let file = program.setFile({ src: `${rootDir} /source/main.brs`, dest: 'source/main.brs' }, `
                 sub main()
-                   print
+                    print
                 end sub
             `);
             (0, testHelpers_spec_1.expectZeroDiagnostics)(file);
         });
         describe('conditional compile', () => {
             it('works for upper case keywords', () => {
-                let file = program.setFile({ src: `${rootDir}/source/main.brs`, dest: 'source/main.brs' }, `
+                let file = program.setFile({ src: `${rootDir} /source/main.brs`, dest: 'source/main.brs' }, `
                     sub main()
                         #CONST someFlag = true
                         #IF someFlag
@@ -491,7 +573,7 @@ describe('BrsFile', () => {
                 (0, testHelpers_spec_1.expectZeroDiagnostics)(file);
             });
             it('supports single-word #elseif and #endif', () => {
-                let file = program.setFile({ src: `${rootDir}/source/main.brs`, dest: 'source/main.brs' }, `
+                let file = program.setFile({ src: `${rootDir} /source/main.brs`, dest: 'source/main.brs' }, `
                     sub main()
                         #const someFlag = true
                         #if someFlag
@@ -504,7 +586,7 @@ describe('BrsFile', () => {
                 (0, testHelpers_spec_1.expectZeroDiagnostics)(file);
             });
             it('supports multi-word #else if and #end if', () => {
-                let file = program.setFile({ src: `${rootDir}/source/main.brs`, dest: 'source/main.brs' }, `
+                let file = program.setFile({ src: `${rootDir} /source/main.brs`, dest: 'source/main.brs' }, `
                     sub main()
                         #const someFlag = true
                         #if someFlag
@@ -517,17 +599,17 @@ describe('BrsFile', () => {
                 (0, testHelpers_spec_1.expectZeroDiagnostics)(file);
             });
             it('does not choke on invalid code inside a false conditional compile', () => {
-                let file = program.setFile({ src: `${rootDir}/source/main.brs`, dest: 'source/main.brs' }, `
+                let file = program.setFile({ src: `${rootDir} /source/main.brs`, dest: 'source/main.brs' }, `
                     sub main()
                         #if false
-                            non-commented code here should not cause parse errors
+                            non - commented code here should not cause parse errors
                         #end if
                     end sub
                 `);
                 (0, testHelpers_spec_1.expectZeroDiagnostics)(file);
             });
             it('detects syntax error in #if', () => {
-                let file = program.setFile({ src: `${rootDir}/source/main.brs`, dest: 'source/main.brs' }, `
+                let file = program.setFile({ src: `${rootDir} /source/main.brs`, dest: 'source/main.brs' }, `
                     sub main()
                         #if true1
                             print "true"
@@ -539,7 +621,7 @@ describe('BrsFile', () => {
                 ]);
             });
             it('detects syntax error in #const', () => {
-                let file = program.setFile({ src: `${rootDir}/source/main.brs`, dest: 'source/main.brs' }, `
+                let file = program.setFile({ src: `${rootDir} /source/main.brs`, dest: 'source/main.brs' }, `
                     sub main()
                         #if %
                             print "true"
@@ -552,7 +634,7 @@ describe('BrsFile', () => {
                 ]);
             });
             it('detects #const name using reserved word', () => {
-                let file = program.setFile({ src: `${rootDir}/source/main.brs`, dest: 'source/main.brs' }, `
+                let file = program.setFile({ src: `${rootDir} /source/main.brs`, dest: 'source/main.brs' }, `
                     sub main()
                         #const function = true
                     end sub
@@ -563,7 +645,7 @@ describe('BrsFile', () => {
                 ]);
             });
             it('detects syntax error in #const', () => {
-                let file = program.setFile({ src: `${rootDir}/source/main.brs`, dest: 'source/main.brs' }, `
+                let file = program.setFile({ src: `${rootDir} /source/main.brs`, dest: 'source/main.brs' }, `
                     sub main()
                         #const someConst = 123
                     end sub
@@ -574,22 +656,22 @@ describe('BrsFile', () => {
             });
         });
         it('supports stop statement', () => {
-            let file = program.setFile({ src: `${rootDir}/source/main.brs`, dest: 'source/main.brs' }, `
+            let file = program.setFile({ src: `${rootDir} /source/main.brs`, dest: 'source/main.brs' }, `
                 sub main()
-                   stop
+                    stop
                 end sub
             `);
             (0, testHelpers_spec_1.expectZeroDiagnostics)(file);
         });
         it('supports single-line if statements', () => {
-            let file = program.setFile({ src: `${rootDir}/source/main.brs`, dest: 'source/main.brs' }, `
+            let file = program.setFile({ src: `${rootDir} /source/main.brs`, dest: 'source/main.brs' }, `
                 sub main()
                     if 1 < 2: return true: end if
                     if 1 < 2: return true
                     end if
                     if false : print "true" : end if
                     if true: print "8 worked": else if true: print "not run": else: print "not run": end if
-                    if true then : test = sub() : print "yes" : end sub : end if
+                    if true then: test = sub() : print "yes" : end sub: end if
                 end sub
             `);
             (0, testHelpers_spec_1.expectZeroDiagnostics)(file);
@@ -674,7 +756,7 @@ describe('BrsFile', () => {
                     print &he2
                     print 1.2E+2
                     print 2!
-                    print 12D-12
+                    print 12D - 12
                     print 2.3#
                     print &hFEDCBA9876543210&
                     print 9876543210&
@@ -741,11 +823,11 @@ describe('BrsFile', () => {
         it('supports variable names ending with type designators', () => {
             file.parse(`
                 sub main()
-                  name$ = "bob"
-                  age% = 1
-                  height! = 5.5
-                  salary# = 9.87654321
-                  someHex& = 13
+                    name$ = "bob"
+                    age% = 1
+                    height! = 5.5
+                        salary# = 9.87654321
+                    someHex& = 13
                 end sub
             `);
             (0, testHelpers_spec_1.expectZeroDiagnostics)(file);
@@ -755,7 +837,7 @@ describe('BrsFile', () => {
                 sub main()
                     if true then
                         print "true"
-                    else    if true then
+                    else if true then
                         print "also true"
                     end if
                 end sub
@@ -766,9 +848,9 @@ describe('BrsFile', () => {
             file.parse(`
                 function GetObject()
                     obj = {
-                        stop: function() as void
+                    stop: function () as void
 
-                        end function
+                            end function
                     }
                     return obj
                 end function
@@ -779,10 +861,10 @@ describe('BrsFile', () => {
             file.parse(`
                 function GetObject()
                     obj = {
-                        run: function() as void
+                        run: function () as void
 
                         end function
-                    }
+                        }
                     return obj
                 end function
             `);
@@ -810,7 +892,7 @@ describe('BrsFile', () => {
                 function Main()
                     promise = {
                         then: sub()
-                        end sub
+                                end sub
                     }
                     promise.then()
                 end function
@@ -820,7 +902,7 @@ describe('BrsFile', () => {
         it('supports function as parameter type', () => {
             file.parse(`
                 sub Main()
-                    doWork = function(callback as function)
+                    doWork = function (callback as function)
                     end function
                 end sub
             `);
@@ -936,7 +1018,7 @@ describe('BrsFile', () => {
                 sub main()
                 end sub
                 import "file.brs"
-            `);
+        `);
             program.validate();
             (0, testHelpers_spec_1.expectDiagnostics)(program, [
                 DiagnosticMessages_1.DiagnosticMessages.importStatementMustBeDeclaredAtTopOfFile()
@@ -973,7 +1055,7 @@ describe('BrsFile', () => {
         it('supports colons as separators in associative array properties', () => {
             file.parse(`
                 sub Main()
-                    obj = {x:0 : y: 1}
+                    obj = { x: 0 : y: 1 }
                 end sub
             `);
             (0, testHelpers_spec_1.expectZeroDiagnostics)(file);
@@ -991,8 +1073,8 @@ describe('BrsFile', () => {
         });
         it('succeeds when finding variables with the word "function" in them', () => {
             file.parse(`
-                function Test()
-                    typeCheckFunction = RBS_CMN_GetFunction(invalid, methodName)
+        function Test()
+        typeCheckFunction = RBS_CMN_GetFunction(invalid, methodName)
                 end function
             `);
         });
@@ -1003,14 +1085,14 @@ describe('BrsFile', () => {
                     print "A"
                 end function
 
-                 function DoB()
-                     print "B"
-                 end function
+                function DoB()
+                    print "B"
+                end function
             `);
             (0, chai_1.expect)(file.callables[0].name).to.equal('DoA');
             (0, chai_1.expect)(file.callables[0].nameRange).to.eql(vscode_languageserver_1.Range.create(1, 25, 1, 28));
             (0, chai_1.expect)(file.callables[1].name).to.equal('DoB');
-            (0, chai_1.expect)(file.callables[1].nameRange).to.eql(vscode_languageserver_1.Range.create(5, 26, 5, 29));
+            (0, chai_1.expect)(file.callables[1].nameRange).to.eql(vscode_languageserver_1.Range.create(5, 25, 5, 28));
         });
         it('throws an error if the file has already been parsed', () => {
             let file = new BrsFile_1.BrsFile('abspath', 'relpath', program);
@@ -1341,13 +1423,13 @@ describe('BrsFile', () => {
         it('finds variable type from other variable', () => {
             file.parse(`
                 sub Main()
-                   name = "bob"
-                   nameCopy = name
+                    name = "bob"
+                    nameCopy = name
                 end sub
             `);
             (0, testHelpers_spec_1.expectSymbolTableEquals)(file.parser.references.functionExpressions[0].symbolTable, [
-                ['name', new StringType_1.StringType(), util_1.default.createRange(2, 19, 2, 23)],
-                ['nameCopy', new StringType_1.StringType(), util_1.default.createRange(3, 19, 3, 27)]
+                ['name', new StringType_1.StringType(), util_1.default.createRange(2, 20, 2, 24)],
+                ['nameCopy', new StringType_1.StringType(), util_1.default.createRange(3, 20, 3, 28)]
             ]);
         });
         it('sets proper range for functions', () => {
@@ -1358,7 +1440,6 @@ describe('BrsFile', () => {
                     end function
                 end sub
             `);
-            (0, chai_1.expect)(file.parser.references.functionExpressions).to.be.length(2);
             (0, chai_1.expect)(file.parser.references.functionExpressions.map(x => x.range)).to.eql([
                 util_1.default.createRange(1, 16, 5, 23),
                 util_1.default.createRange(2, 30, 4, 32)
@@ -1367,10 +1448,10 @@ describe('BrsFile', () => {
     });
     describe('getHover', () => {
         it('works for param types', () => {
-            let file = program.setFile({ src: `${rootDir}/source/main.brs`, dest: 'source/main.brs' }, `
+            let file = program.setFile({ src: `${rootDir} /source/main.brs`, dest: 'source/main.brs' }, `
                 sub DoSomething(name as string)
                     name = 1
-                    sayMyName = function(name as string)
+                    sayMyName = function (name as string)
                     end function
                 end sub
             `);
@@ -1385,19 +1466,19 @@ describe('BrsFile', () => {
         });
         //ignore this for now...it's not a huge deal
         it('does not match on keywords or data types', () => {
-            let file = program.setFile({ src: `${rootDir}/source/main.brs`, dest: 'source/main.brs' }, `
+            let file = program.setFile({ src: `${rootDir} /source/main.brs`, dest: 'source/main.brs' }, `
                 sub Main(name as string)
                 end sub
-                sub as()
+                sub as ()
                 end sub
             `);
-            //hover over the `as`
+            //hover over the `as `
             (0, chai_1.expect)(file.getHover(vscode_languageserver_1.Position.create(1, 31))).not.to.exist;
             //hover over the `string`
             (0, chai_1.expect)(file.getHover(vscode_languageserver_1.Position.create(1, 36))).not.to.exist;
         });
         it('finds declared function', () => {
-            let file = program.setFile({ src: `${rootDir}/source/main.brs`, dest: 'source/main.brs' }, `
+            let file = program.setFile({ src: `${rootDir} /source/main.brs`, dest: 'source/main.brs' }, `
                 function Main(count = 1)
                     firstName = "bob"
                     age = 21
@@ -1666,6 +1747,106 @@ describe('BrsFile', () => {
         //this test will throw an exception if something went wrong
     });
     describe('transpile', () => {
+        it('transpiles if statement keywords as provided', () => {
+            const code = `
+                If True Then
+                    Print True
+                Else If True Then
+                    print True
+                Else If False Then
+                    Print False
+                Else
+                    Print False
+                End If
+            `;
+            testTranspile(code);
+            testTranspile(code.toLowerCase());
+            testTranspile(code.toUpperCase());
+        });
+        it('does not transpile `then` tokens', () => {
+            const code = `
+                if true
+                    print true
+                else if true
+                    print false
+                end if
+            `;
+            testTranspile(code);
+        });
+        it('honors spacing between multi-word tokens', () => {
+            testTranspile(`
+                if true
+                    print true
+                elseif true
+                    print false
+                endif
+            `);
+        });
+        it('handles when only some of the statements have `then`', () => {
+            testTranspile(`
+                if true
+                else if true then
+                else if true
+                else if true then
+                    if true then
+                        return true
+                    end if
+                end if
+            `);
+        });
+        it('retains casing of parameter types', () => {
+            function test(type) {
+                testTranspile(`
+                    sub one(a as ${type}, b as ${type.toUpperCase()}, c as ${type.toLowerCase()})
+                    end sub
+                `);
+            }
+            test('Boolean');
+            test('Double');
+            test('Dynamic');
+            test('Float');
+            test('Integer');
+            test('LongInteger');
+            test('Object');
+            test('String');
+        });
+        it('retains casing of return types', () => {
+            function test(type) {
+                testTranspile(`
+                    sub one() as ${type}
+                    end sub
+
+                    sub two() as ${type.toLowerCase()}
+                    end sub
+
+                    sub three() as ${type.toUpperCase()}
+                    end sub
+                `);
+            }
+            test('Boolean');
+            test('Double');
+            test('Dynamic');
+            test('Float');
+            test('Integer');
+            test('LongInteger');
+            test('Object');
+            test('String');
+            test('Void');
+        });
+        it('retains casing of literal types', () => {
+            function test(type) {
+                testTranspile(`
+                    sub main()
+                        thing = ${type}
+                        thing = ${type.toLowerCase()}
+                        thing = ${type.toUpperCase()}
+                    end sub
+                `);
+            }
+            test('Invalid');
+            test('True');
+            test('False');
+        });
         describe('throwStatement', () => {
             it('transpiles properly', () => {
                 testTranspile(`
@@ -1914,29 +2095,6 @@ describe('BrsFile', () => {
                 end sub
         `, null, 'trim');
         });
-        it('adds `then` when missing', () => {
-            testTranspile(`
-                sub a()
-                    if true
-                        print "true"
-                    else if true
-                        print "true"
-                    else
-                        print "true"
-                    end if
-                end sub
-            `, `
-                sub a()
-                    if true then
-                        print "true"
-                    else if true then
-                        print "true"
-                    else
-                        print "true"
-                    end if
-                end sub
-            `, 'trim');
-        });
         it('does not add leading or trailing newlines', () => {
             testTranspile(`function abc()\nend function`, undefined, 'none');
         });
@@ -2386,11 +2544,16 @@ describe('BrsFile', () => {
                     end class
                     class Duck extends Bird
                     end class
-                end namespace`, (0, testHelpers_spec_1.trim) `
+                end namespace
+            `, (0, testHelpers_spec_1.trim) `
                 namespace AnimalKingdom
                     class Bird
+                        sub new()
+                        end sub
                     end class
                     class Duck extends AnimalKingdom.Bird
+                        sub new()
+                        end sub
                     end class
                 end namespace
             `);
@@ -2430,6 +2593,8 @@ describe('BrsFile', () => {
                     function getDuck()
                     end function
                     class Duck
+                        sub new()
+                        end sub
                         @anMember
                         @anMember("field")
                         private thing as dynamic
@@ -2489,6 +2654,8 @@ describe('BrsFile', () => {
                 end namespace
             `, (0, testHelpers_spec_1.trim) `
                 class Person
+                    sub new()
+                    end sub
                     public name as string
                     public age as integer
                     public sub getAge() as integer
@@ -2496,6 +2663,8 @@ describe('BrsFile', () => {
                 end class
                 namespace NameA.NameB
                     class Person
+                        sub new()
+                        end sub
                         public name as string
                         public age as integer
                         public sub getAge() as integer
@@ -2504,6 +2673,17 @@ describe('BrsFile', () => {
                 end namespace
             `);
         });
+        it('creates constructor properly', () => {
+            testTypedef(`
+                class Parent
+                end class
+            `, (0, testHelpers_spec_1.trim) `
+                class Parent
+                    sub new()
+                    end sub
+                end class
+            `);
+        });
         it('sets properties to dynamic when initialized to invalid', () => {
             testTypedef(`
                 class Human
@@ -2512,6 +2692,8 @@ describe('BrsFile', () => {
                 end class
             `, (0, testHelpers_spec_1.trim) `
                 class Human
+                    sub new()
+                    end sub
                     public firstName as dynamic
                     public lastName as string
                 end class
@@ -2558,6 +2740,8 @@ describe('BrsFile', () => {
                 end class
             `, (0, testHelpers_spec_1.trim) `
                 class Human
+                    sub new()
+                    end sub
                     public firstName as string
                     protected middleName as string
                     private lastName as string
@@ -2584,10 +2768,14 @@ describe('BrsFile', () => {
                 end class
             `, (0, testHelpers_spec_1.trim) `
                 class Animal
+                    sub new()
+                    end sub
                     public sub speak()
                     end sub
                 end class
                 class Dog extends Animal
+                    sub new()
+                    end sub
                     public override sub speak()
                     end sub
                 end class
@@ -2659,47 +2847,35 @@ describe('BrsFile', () => {
         });
     });
     describe('Plugins', () => {
-        function testPluginTranspile() {
-            testTranspile(`
-                sub main()
-                    sayHello(sub()
-                        print "sub hello"
-                    end sub)
-                    print "something"
-                end sub
-
-                sub sayHello(fn)
-                    fn()
-                    print "hello"
-                end sub
-            `, `
-                sub main()
-                    sayHello(sub()
-                        \n                    end sub)
-                    \n                end sub
-
-                sub sayHello(fn)
-                    fn()
-                    \n                end sub
+        let pluginFileName;
+        let idx = 1;
+        beforeEach(() => {
+            pluginFileName = `plugin-${idx++}.js`;
+            fsExtra.outputFileSync((0, util_1.standardizePath) `${tempDir}/plugins/${pluginFileName}`, `
+                function plugin() {
+                    return {
+                        name: 'lower-file-name',
+                        afterFileParse: (evt) => {
+                            evt.file._customProp = true;
+                        }
+                    };
+                }
+                exports.default = plugin;
             `);
-        }
-        it('can use a plugin object which transforms the AST', () => {
-            program.plugins = new PluginInterface_1.default(util_1.default.loadPlugins('', [
-                require.resolve('../examples/plugins/removePrint')
-            ]), new Logger_1.Logger());
-            testPluginTranspile();
         });
-        it('can load an absolute plugin which transforms the AST', () => {
-            program.plugins = new PluginInterface_1.default(util_1.default.loadPlugins('', [
-                path.resolve(process.cwd(), './dist/examples/plugins/removePrint.js')
+        it('can load an absolute plugin which receives callbacks', () => {
+            program.plugins = new PluginInterface_1.default(util_1.default.loadPlugins(tempDir, [
+                (0, util_1.standardizePath) `${tempDir}/plugins/${pluginFileName}`
             ]), new Logger_1.Logger());
-            testPluginTranspile();
+            const file = program.setFile('source/MAIN.brs', '');
+            (0, chai_1.expect)(file._customProp).to.exist;
         });
-        it('can load a relative plugin which transforms the AST', () => {
-            program.plugins = new PluginInterface_1.default(util_1.default.loadPlugins(process.cwd(), [
-                './dist/examples/plugins/removePrint.js'
+        it('can load a relative plugin which receives callbacks', () => {
+            program.plugins = new PluginInterface_1.default(util_1.default.loadPlugins(tempDir, [
+                `./plugins/${pluginFileName}`
             ]), new Logger_1.Logger());
-            testPluginTranspile();
+            const file = program.setFile('source/MAIN.brs', '');
+            (0, chai_1.expect)(file._customProp).to.exist;
         });
     });
     describe('getSymbolTypeFromToken', () => {