brighterscript 0.41.2 → 1.0.0-alpha.12

package/dist/files/BrsFile.Class.spec.js

@@ -11,6 +11,7 @@ const util_1 = require("../util");
 const fsExtra = require("fs-extra");
 const BrsTranspileState_1 = require("../parser/BrsTranspileState");
 const assert_1 = require("assert");
+const reflection_1 = require("../astUtils/reflection");
 let sinon = sinonImport.createSandbox();
 describe('BrsFile BrighterScript classes', () => {
     let tmpPath = (0, util_1.standardizePath) `${process.cwd()}/.tmp`;
@@ -29,10 +30,10 @@ describe('BrsFile BrighterScript classes', () => {
         fsExtra.emptyDirSync(tmpPath);
     });
     function addFile(relativePath, text) {
-        return program.addOrReplaceFile({ src: `${rootDir}/${relativePath}`, dest: relativePath }, text);
+        return program.setFile({ src: `${rootDir}/${relativePath}`, dest: relativePath }, text);
     }
     it('detects all classes after parse', () => {
-        let file = program.addOrReplaceFile({ src: `${rootDir}/source/main.brs`, dest: 'source/main.brs' }, `
+        let file = program.setFile({ src: `${rootDir}/source/main.brs`, dest: 'source/main.brs' }, `
             class Animal
             end class
             class Duck
@@ -43,7 +44,7 @@ describe('BrsFile BrighterScript classes', () => {
         (0, chai_1.expect)(file.parser.references.classStatements.map(x => x.getName(Parser_1.ParseMode.BrighterScript)).sort()).to.eql(['Animal', 'Duck']);
     });
     it('does not cause errors with incomplete class statement', () => {
-        program.addOrReplaceFile({ src: `${rootDir}/source/main.bs`, dest: 'source/main.bs' }, `
+        program.setFile({ src: `${rootDir}/source/main.bs`, dest: 'source/main.bs' }, `
             class
         `);
         program.validate();
@@ -51,7 +52,7 @@ describe('BrsFile BrighterScript classes', () => {
     });
     it('catches child class missing super call in constructor', () => {
         var _a;
-        program.addOrReplaceFile({ src: `${rootDir}/source/main.bs`, dest: 'source/main.bs' }, `
+        program.setFile({ src: `${rootDir}/source/main.bs`, dest: 'source/main.bs' }, `
             class Person
                 sub new()
                 end sub
@@ -66,7 +67,7 @@ describe('BrsFile BrighterScript classes', () => {
     });
     it('access modifier is option for override', () => {
         var _a;
-        let file = program.addOrReplaceFile({ src: `${rootDir}/source/main.bs`, dest: 'source/main.bs' }, `
+        let file = program.setFile({ src: `${rootDir}/source/main.bs`, dest: 'source/main.bs' }, `
             class Animal
                 sub move()
                 end sub
@@ -85,7 +86,7 @@ describe('BrsFile BrighterScript classes', () => {
     });
     it('supports various namespace configurations', () => {
         var _a;
-        program.addOrReplaceFile({ src: `${rootDir}/source/main.bs`, dest: 'source/main.bs' }, `
+        program.setFile({ src: `${rootDir}/source/main.bs`, dest: 'source/main.bs' }, `
             class Animal
                 sub new()
                     bigBird = new Birds.Bird()
@@ -110,7 +111,7 @@ describe('BrsFile BrighterScript classes', () => {
     describe('super', () => {
         it('always requires super call in child constructor', () => {
             var _a;
-            program.addOrReplaceFile('source/main.bs', `
+            program.setFile('source/main.bs', `
                 class Bird
                 end class
                 class Duck extends Bird
@@ -123,7 +124,7 @@ describe('BrsFile BrighterScript classes', () => {
         });
         it('requires super call in child when parent has own `new` method', () => {
             var _a;
-            program.addOrReplaceFile('source/main.bs', `
+            program.setFile('source/main.bs', `
                 class Bird
                     sub new()
                     end sub
@@ -138,7 +139,7 @@ describe('BrsFile BrighterScript classes', () => {
         });
         it('allows non-`m` expressions and statements before the super call', () => {
             var _a;
-            program.addOrReplaceFile('source/main.bs', `
+            program.setFile('source/main.bs', `
                 class Bird
                     sub new(name)
                     end sub
@@ -156,7 +157,7 @@ describe('BrsFile BrighterScript classes', () => {
             (0, chai_1.expect)((_a = program.getDiagnostics()[0]) === null || _a === void 0 ? void 0 : _a.message).to.be.undefined;
         });
         it('allows non-`m` expressions and statements before the super call', () => {
-            program.addOrReplaceFile('source/main.bs', `
+            program.setFile('source/main.bs', `
                 class Bird
                     sub new(name)
                     end sub
@@ -164,7 +165,7 @@ describe('BrsFile BrighterScript classes', () => {
                 class Duck extends Bird
                     sub new()
                         m.name = m.name + "Duck"
-                        super()
+                        super(m.name)
                     end sub
                 end class
             `);
@@ -357,6 +358,9 @@ describe('BrsFile BrighterScript classes', () => {
                 class Animal
                     sub new(name as string)
                     end sub
+
+                    sub DoSomething()
+                    end sub
                 end class
 
                 class Duck extends Animal
@@ -370,6 +374,8 @@ describe('BrsFile BrighterScript classes', () => {
                     instance = {}
                     instance.new = sub(name as string)
                     end sub
+                    instance.DoSomething = sub()
+                    end sub
                     return instance
                 end function
                 function Animal(name as string)
@@ -661,7 +667,7 @@ describe('BrsFile BrighterScript classes', () => {
     });
     it('detects using `new` keyword on non-classes', () => {
         var _a;
-        program.addOrReplaceFile({ src: `${rootDir}/source/main.brs`, dest: 'source/main.brs' }, `
+        program.setFile({ src: `${rootDir}/source/main.brs`, dest: 'source/main.brs' }, `
             sub quack()
             end sub
             sub main()
@@ -673,7 +679,7 @@ describe('BrsFile BrighterScript classes', () => {
     });
     it('detects missing call to super', () => {
         var _a;
-        program.addOrReplaceFile({ src: `${rootDir}/source/main.brs`, dest: 'source/main.brs' }, `
+        program.setFile({ src: `${rootDir}/source/main.brs`, dest: 'source/main.brs' }, `
             class Animal
                 sub new()
                 end sub
@@ -688,7 +694,7 @@ describe('BrsFile BrighterScript classes', () => {
     });
     it.skip('detects calls to unknown m methods', () => {
         var _a;
-        program.addOrReplaceFile({ src: `${rootDir}/source/main.brs`, dest: 'source/main.brs' }, `
+        program.setFile({ src: `${rootDir}/source/main.brs`, dest: 'source/main.brs' }, `
             class Animal
                 sub new()
                     m.methodThatDoesNotExist()
@@ -699,7 +705,7 @@ describe('BrsFile BrighterScript classes', () => {
         (0, chai_1.expect)((_a = program.getDiagnostics()[0]) === null || _a === void 0 ? void 0 : _a.message).to.equal(DiagnosticMessages_1.DiagnosticMessages.methodDoesNotExistOnType('methodThatDoesNotExist', 'Animal'));
     });
     it('detects duplicate member names', () => {
-        program.addOrReplaceFile({ src: `${rootDir}/source/main.bs`, dest: 'source/main.bs' }, `
+        program.setFile({ src: `${rootDir}/source/main.bs`, dest: 'source/main.bs' }, `
             class Animal
                 public name
                 public name
@@ -724,7 +730,7 @@ describe('BrsFile BrighterScript classes', () => {
         (0, chai_1.expect)(diagnostics).to.eql([Object.assign(Object.assign({}, DiagnosticMessages_1.DiagnosticMessages.duplicateIdentifier('name')), { range: vscode_languageserver_1.Range.create(3, 23, 3, 27) }), Object.assign(Object.assign({}, DiagnosticMessages_1.DiagnosticMessages.duplicateIdentifier('name')), { range: vscode_languageserver_1.Range.create(4, 27, 4, 31) }), Object.assign(Object.assign({}, DiagnosticMessages_1.DiagnosticMessages.duplicateIdentifier('age')), { range: vscode_languageserver_1.Range.create(8, 27, 8, 30) }), Object.assign(Object.assign({}, DiagnosticMessages_1.DiagnosticMessages.duplicateIdentifier('age')), { range: vscode_languageserver_1.Range.create(10, 23, 10, 26) })]);
     });
     it('detects mismatched member type in child class', () => {
-        program.addOrReplaceFile({ src: `${rootDir}/source/main.bs`, dest: 'source/main.bs' }, `
+        program.setFile({ src: `${rootDir}/source/main.bs`, dest: 'source/main.bs' }, `
             class Animal
                 public name
             end class
@@ -738,7 +744,7 @@ describe('BrsFile BrighterScript classes', () => {
         (0, chai_1.expect)(program.getDiagnostics().map(x => x.message).sort()[0]).to.eql(DiagnosticMessages_1.DiagnosticMessages.classChildMemberDifferentMemberTypeThanAncestor('method', 'field', 'Animal').message);
     });
     it('allows untyped overridden field in child class', () => {
-        program.addOrReplaceFile({ src: `${rootDir}/source/main.bs`, dest: 'source/main.bs' }, `
+        program.setFile({ src: `${rootDir}/source/main.bs`, dest: 'source/main.bs' }, `
             class Animal
                 public name
             end class
@@ -750,7 +756,7 @@ describe('BrsFile BrighterScript classes', () => {
         (0, testHelpers_spec_1.expectZeroDiagnostics)(program);
     });
     it('allows overridden property name in child class', () => {
-        program.addOrReplaceFile('source/main.bs', `
+        program.setFile('source/main.bs', `
             class Bird
                 public name = "bird"
             end class
@@ -762,7 +768,7 @@ describe('BrsFile BrighterScript classes', () => {
         (0, testHelpers_spec_1.expectZeroDiagnostics)(program);
     });
     it('flags incompatible child field type changes', () => {
-        program.addOrReplaceFile('source/main.bs', `
+        program.setFile('source/main.bs', `
             class Bird
                 public age = 12
                 public name = "bird"
@@ -783,7 +789,7 @@ describe('BrsFile BrighterScript classes', () => {
         ]);
     });
     it('detects overridden methods without override keyword', () => {
-        program.addOrReplaceFile({ src: `${rootDir}/source/main.brs`, dest: 'source/main.brs' }, `
+        program.setFile({ src: `${rootDir}/source/main.brs`, dest: 'source/main.brs' }, `
             class Animal
                 sub speak()
                 end sub
@@ -797,7 +803,7 @@ describe('BrsFile BrighterScript classes', () => {
         (0, chai_1.expect)(program.getDiagnostics()[0]).to.exist.and.to.include(Object.assign({}, DiagnosticMessages_1.DiagnosticMessages.missingOverrideKeyword('Animal')));
     });
     it('detects overridden methods with different visibility', () => {
-        program.addOrReplaceFile({ src: `${rootDir}/source/main.bs`, dest: 'source/main.bs' }, `
+        program.setFile({ src: `${rootDir}/source/main.bs`, dest: 'source/main.bs' }, `
             class Animal
                 sub speakInPublic()
                 end sub
@@ -821,7 +827,7 @@ describe('BrsFile BrighterScript classes', () => {
         (0, chai_1.expect)(program.getDiagnostics()[2]).to.exist.and.to.include(Object.assign({}, DiagnosticMessages_1.DiagnosticMessages.mismatchedOverriddenMemberVisibility('Duck', 'speakWithFamily', 'public', 'private', 'Animal')));
     });
     it('allows overridden methods with matching visibility', () => {
-        program.addOrReplaceFile({ src: `${rootDir}/source/main.bs`, dest: 'source/main.bs' }, `
+        program.setFile({ src: `${rootDir}/source/main.bs`, dest: 'source/main.bs' }, `
             class Animal
                 sub speakInPublic()
                 end sub
@@ -843,7 +849,7 @@ describe('BrsFile BrighterScript classes', () => {
         (0, chai_1.expect)(program.getDiagnostics()).to.be.empty;
     });
     it('detects extending unknown parent class', () => {
-        program.addOrReplaceFile('source/main.brs', `
+        program.setFile('source/main.brs', `
             class Duck extends Animal
                 sub speak()
                 end sub
@@ -854,7 +860,7 @@ describe('BrsFile BrighterScript classes', () => {
     });
     it('catches newable class without namespace name', () => {
         var _a;
-        program.addOrReplaceFile('source/main.bs', `
+        program.setFile('source/main.bs', `
             namespace NameA.NameB
                 class Duck
                 end class
@@ -869,7 +875,7 @@ describe('BrsFile BrighterScript classes', () => {
     });
     it('supports newable class namespace inference', () => {
         var _a;
-        program.addOrReplaceFile({ src: `${rootDir}/source/main.bs`, dest: 'source/main.bs' }, `
+        program.setFile({ src: `${rootDir}/source/main.bs`, dest: 'source/main.bs' }, `
             namespace NameA.NameB
                 class Duck
                 end class
@@ -883,7 +889,7 @@ describe('BrsFile BrighterScript classes', () => {
     });
     it('catches extending unknown namespaced class', () => {
         var _a;
-        program.addOrReplaceFile({ src: `${rootDir}/source/main.bs`, dest: 'source/main.bs' }, `
+        program.setFile({ src: `${rootDir}/source/main.bs`, dest: 'source/main.bs' }, `
             namespace NameA.NameB
                 class Animal
                 end class
@@ -896,7 +902,7 @@ describe('BrsFile BrighterScript classes', () => {
     });
     it('supports omitting namespace prefix for items in same namespace', () => {
         var _a;
-        program.addOrReplaceFile({ src: `${rootDir}/source/main.bs`, dest: 'source/main.bs' }, `
+        program.setFile({ src: `${rootDir}/source/main.bs`, dest: 'source/main.bs' }, `
             namespace NameA.NameB
                 class Animal
                 end class
@@ -909,7 +915,7 @@ describe('BrsFile BrighterScript classes', () => {
     });
     it('catches duplicate root-level class declarations', () => {
         var _a;
-        program.addOrReplaceFile({ src: `${rootDir}/source/main.bs`, dest: 'source/main.bs' }, `
+        program.setFile({ src: `${rootDir}/source/main.bs`, dest: 'source/main.bs' }, `
             class Animal
             end class
             class Animal
@@ -919,7 +925,7 @@ describe('BrsFile BrighterScript classes', () => {
     });
     it('catches duplicate namespace-level class declarations', () => {
         var _a;
-        program.addOrReplaceFile({ src: `${rootDir}/source/main.bs`, dest: 'source/main.bs' }, `
+        program.setFile({ src: `${rootDir}/source/main.bs`, dest: 'source/main.bs' }, `
             namespace NameA.NameB
                 class Animal
                 end class
@@ -931,7 +937,7 @@ describe('BrsFile BrighterScript classes', () => {
     });
     it('catches namespaced class name which is the same as a global class', () => {
         var _a;
-        program.addOrReplaceFile({ src: `${rootDir}/source/main.bs`, dest: 'source/main.bs' }, `
+        program.setFile({ src: `${rootDir}/source/main.bs`, dest: 'source/main.bs' }, `
             namespace NameA.NameB
                 class Animal
                 end class
@@ -944,7 +950,7 @@ describe('BrsFile BrighterScript classes', () => {
     });
     it('catches class with same name as function', () => {
         var _a;
-        program.addOrReplaceFile('source/main.bs', `
+        program.setFile('source/main.bs', `
             class Animal
             end class
             sub Animal()
@@ -955,7 +961,7 @@ describe('BrsFile BrighterScript classes', () => {
     });
     it('catches class with same name (but different case) as function', () => {
         var _a;
-        program.addOrReplaceFile('source/main.bs', `
+        program.setFile('source/main.bs', `
             class ANIMAL
             end class
             sub animal()
@@ -966,7 +972,7 @@ describe('BrsFile BrighterScript classes', () => {
     });
     it('catches variable with same name as class', () => {
         var _a;
-        program.addOrReplaceFile('source/main.bs', `
+        program.setFile('source/main.bs', `
             class Animal
             end class
             sub main()
@@ -977,7 +983,7 @@ describe('BrsFile BrighterScript classes', () => {
         (0, chai_1.expect)((_a = program.getDiagnostics()[0]) === null || _a === void 0 ? void 0 : _a.message).to.equal(DiagnosticMessages_1.DiagnosticMessages.localVarSameNameAsClass('Animal').message);
     });
     it('allows extending classes with more than one dot in the filename', () => {
-        program.addOrReplaceFile('source/testclass.bs', `
+        program.setFile('source/testclass.bs', `
             class Foo
             end class
 
@@ -987,14 +993,14 @@ describe('BrsFile BrighterScript classes', () => {
                 end sub
             end class
         `);
-        program.addOrReplaceFile('source/testclass_no_testdot.bs', `
+        program.setFile('source/testclass_no_testdot.bs', `
             class BarNoDot extends Foo
                 sub new()
                     super()
                 end sub
             end class
         `);
-        program.addOrReplaceFile('source/testclass.dot.bs', `
+        program.setFile('source/testclass.dot.bs', `
             class BarDot extends Foo
                 sub new()
                 super()
@@ -1005,7 +1011,7 @@ describe('BrsFile BrighterScript classes', () => {
         (0, testHelpers_spec_1.expectZeroDiagnostics)(program);
     });
     it('computes correct super index for grandchild class', () => {
-        program.addOrReplaceFile('source/main.bs', `
+        program.setFile('source/main.bs', `
             sub Main()
                 c = new App.ClassC()
             end sub
@@ -1043,7 +1049,7 @@ describe('BrsFile BrighterScript classes', () => {
         `, 'trim', 'source/App.ClassC.bs');
     });
     it('computes correct super index for namespaced child class and global parent class', () => {
-        program.addOrReplaceFile('source/ClassA.bs', `
+        program.setFile('source/ClassA.bs', `
             class ClassA
             end class
         `);
@@ -1069,7 +1075,7 @@ describe('BrsFile BrighterScript classes', () => {
         `, 'trim', 'source/App.ClassB.bs');
     });
     it('does not crash when parent class is missing', () => {
-        const file = program.addOrReplaceFile('source/ClassB.bs', `
+        const file = program.setFile('source/ClassB.bs', `
             class ClassB extends ClassA
             end class
         `);
@@ -1077,8 +1083,317 @@ describe('BrsFile BrighterScript classes', () => {
             file.parser.references.classStatements[0].getParentClassIndex(new BrsTranspileState_1.BrsTranspileState(file));
         });
     });
+    describe('getHover', () => {
+        const animalClassCode = `
+        class Animal
+            kind as string
+            isHungry as boolean
+
+            sub new(kind as string)
+                m.kind = kind
+                m.isHungry = true ' born hungry
+            end sub
+
+            sub eat(foodAmount as integer)
+                if foodAmount > 100
+                  m.isHungry = false
+                end if
+            end sub
+
+            sub sleep()
+              m.isHungry = false
+            end sub
+
+        end class
+
+        class Dog extends Animal
+            breed as string
+            sub new(breed as string)
+                super("Dog")
+                m.breed = breed
+            end sub
+
+            sub snooze()
+                m.sleep()
+            end sub
+        end class
+
+        class DogHouse
+            puppy as Dog
+            sub new(pup as Dog)
+                m.puppy = pup
+            end sub
+        end class
+
+        sub main()
+            snoopy = new Dog("Beagle")
+            biplane = new DogHouse(snoopy)
+            print snoopy.kind ' Dog
+            print snoopy.breed ' Beagle
+            print snoopy.isHungry ' true
+            feedAnimal(biplane.puppy)
+            print snoopy.isHungry ' false
+        end sub
+
+        sub feedAnimal(beast as Animal)
+            beast.eat(100)
+        end sub
+        `;
+        it('correctly parses the file', () => {
+            program.setFile('source/animal.bs', animalClassCode);
+            program.validate();
+            (0, chai_1.expect)(program.getDiagnostics().length).to.equal(0);
+        });
+        it('gets the correct text for m', () => {
+            let animalCode = program.setFile('source/animal.bs', animalClassCode);
+            program.validate();
+            let hover = animalCode.getHover(vscode_languageserver_1.Position.create(6, 17));
+            (0, chai_1.expect)(hover).to.exist;
+            (0, chai_1.expect)(hover.contents).to.equal('m as Animal');
+            hover = animalCode.getHover(vscode_languageserver_1.Position.create(26, 17));
+            (0, chai_1.expect)(hover).to.exist;
+            (0, chai_1.expect)(hover.contents).to.equal('m as Dog');
+        });
+        it('gets the correct text for m.field', () => {
+            let animalCode = program.setFile('source/animal.bs', animalClassCode);
+            program.validate();
+            let hover = animalCode.getHover(vscode_languageserver_1.Position.create(6, 19));
+            (0, chai_1.expect)(hover).to.exist;
+            (0, chai_1.expect)(hover.contents).to.equal('Animal.kind as string');
+            hover = animalCode.getHover(vscode_languageserver_1.Position.create(26, 19));
+            (0, chai_1.expect)(hover).to.exist;
+            (0, chai_1.expect)(hover.contents).to.equal('Dog.breed as string');
+        });
+        it('gets the correct text for m.method', () => {
+            let animalCode = program.setFile('source/animal.bs', animalClassCode);
+            program.validate();
+            let hover = animalCode.getHover(vscode_languageserver_1.Position.create(30, 21));
+            (0, chai_1.expect)(hover).to.exist;
+            (0, chai_1.expect)(hover.contents).to.equal('sub Animal.sleep() as void');
+        });
+        it('gets the correct text for obj.field', () => {
+            let animalCode = program.setFile('source/animal.brs', animalClassCode);
+            program.validate();
+            let hover = animalCode.getHover(vscode_languageserver_1.Position.create(46, 29));
+            (0, chai_1.expect)(hover).to.exist;
+            //TODO TYPES: This should probably say 'Animal.isHungry ...' because that field is defined in Animal
+            (0, chai_1.expect)(hover.contents).to.equal('Dog.isHungry as boolean');
+        });
+        it('gets the correct text for obj.method', () => {
+            let animalCode = program.setFile('source/animal.bs', animalClassCode);
+            program.validate();
+            let hover = animalCode.getHover(vscode_languageserver_1.Position.create(52, 21));
+            (0, chai_1.expect)(hover).to.exist;
+            (0, chai_1.expect)(hover.contents).to.equal('sub Animal.eat(foodAmount as integer) as void');
+        });
+    });
+    describe('getHover class members', () => {
+        const testCode = `
+            class Foo
+                sub new(name as string)
+                end sub
+            end class
+
+            class Bar
+                sub doSomething()
+                    myFoo = new Foo()
+                end sub
+
+                function getInt() as integer
+                    return 1
+                end function
+            end class
+
+            class Buz
+                myInt as integer
+                sub new(i as integer)
+                    myInt = i
+                end sub
+            end class
+
+            class Bee extends Buz
+                sub new(i as integer)
+                    super(i)
+                end sub
+
+                sub varSameNameAsMember()
+                   myInt = 4567
+                   print myInt ' should not print m.myInt
+                end sub
+            end class`;
+        it('correctly parses the file', () => {
+            program.setFile('source/fooBar.bs', testCode);
+            program.validate();
+            (0, chai_1.expect)(program.getDiagnostics().length).to.equal(0);
+        });
+        it('gets the correct text for new Class()', () => {
+            let file = program.setFile('source/fooBar.bs', testCode);
+            program.validate();
+            let hover = file.getHover(vscode_languageserver_1.Position.create(8, 34)); // new Foo("hello")
+            (0, chai_1.expect)(hover).to.exist;
+            (0, chai_1.expect)(hover.contents).to.equal('new Foo(name as string)');
+        });
+        it('gets the correct text for created object', () => {
+            let file = program.setFile('source/fooBar.bs', testCode);
+            program.validate();
+            let hover = file.getHover(vscode_languageserver_1.Position.create(8, 23)); // myFoo
+            (0, chai_1.expect)(hover).to.exist;
+            (0, chai_1.expect)(hover.contents).to.equal('myFoo as Foo');
+        });
+        it('gets the correct text for class declaration', () => {
+            let file = program.setFile('source/fooBar.bs', testCode);
+            program.validate();
+            let hover = file.getHover(vscode_languageserver_1.Position.create(6, 21)); // class Bar
+            (0, chai_1.expect)(hover).to.exist;
+            (0, chai_1.expect)(hover.contents).to.equal('class Bar');
+        });
+        it('gets the correct text for class method declaration', () => {
+            let file = program.setFile('source/fooBar.bs', testCode);
+            program.validate();
+            let hover = file.getHover(vscode_languageserver_1.Position.create(11, 29)); // getInt
+            (0, chai_1.expect)(hover).to.exist;
+            (0, chai_1.expect)(hover.contents).to.equal('function Bar.getInt() as integer');
+        });
+        it('gets the correct text for class field declaration', () => {
+            let file = program.setFile('source/fooBar.bs', testCode);
+            program.validate();
+            let hover = file.getHover(vscode_languageserver_1.Position.create(17, 20)); // myInt
+            (0, chai_1.expect)(hover).to.exist;
+            (0, chai_1.expect)(hover.contents).to.equal('Buz.myInt as integer');
+        });
+        it('gets the correct text for super() call', () => {
+            let file = program.setFile('source/fooBar.bs', testCode);
+            program.validate();
+            let hover = file.getHover(vscode_languageserver_1.Position.create(25, 23)); // super() in Bee.new
+            (0, chai_1.expect)(hover).to.exist;
+            (0, chai_1.expect)(hover.contents).to.equal('new Buz(i as integer)');
+        });
+        it('gets the correct text for variable with same name as a member', () => {
+            let file = program.setFile('source/fooBar.bs', testCode);
+            program.validate();
+            let hover = file.getHover(vscode_languageserver_1.Position.create(29, 23)); // myInt in Bee.varSameNameAsMember
+            (0, chai_1.expect)(hover).to.exist;
+            (0, chai_1.expect)(hover.contents).to.equal('myInt as integer');
+        });
+    });
+    describe('getSymbolTypeFromToken', () => {
+        const testClassCode = `
+        class KlassA
+            function getInt() as integer
+                return 1
+            end function
+
+            sub printInt(i as integer)
+                print i
+            end sub
+        end class
+
+        class KlassB
+
+            function getA() as KlassA
+                return new KlassA()
+            end function
+        end class
+
+        class KlassC
+            public b as KlassB
+
+            sub new()
+                m.b = new KlassB()
+            end sub
+        end class
+
+        sub main()
+            c = new KlassC()
+            a = c.b.getA()
+            num = c.b.getA().getInt()
+            c.b.getA().printInt(num)
+        end sub
+        `;
+        it('correctly parses the file', () => {
+            program.setFile('source/klassTest.bs', testClassCode);
+            program.validate();
+            (0, chai_1.expect)(program.getDiagnostics().length).to.equal(0);
+        });
+        describe('finding tokens', () => {
+            let klassCode;
+            let mainScope;
+            beforeEach(() => {
+                klassCode = program.setFile('source/klassTest.bs', testClassCode);
+                const scopes = program.getScopesForFile(klassCode);
+                (0, chai_1.expect)(scopes.length).to.eql(1);
+                mainScope = scopes[0];
+                mainScope.linkSymbolTable();
+            });
+            afterEach(() => {
+                mainScope.unlinkSymbolTable();
+            });
+            it('gets correct class for m', () => {
+                var _a;
+                const position = vscode_languageserver_1.Position.create(22, 17); // 'm' from m.b = new KlassB()
+                const token = klassCode.parser.getTokenAt(position);
+                (0, chai_1.expect)(token.text).to.equal('m');
+                const func = klassCode.getFunctionExpressionAtPosition(position);
+                let klass = (_a = klassCode.getClassFromToken(token, func, mainScope)) === null || _a === void 0 ? void 0 : _a.item;
+                (0, chai_1.expect)(klass).to.exist;
+                (0, chai_1.expect)(klass.getName(Parser_1.ParseMode.BrighterScript)).to.equal('KlassC');
+            });
+            it('gets correct class for fieldMember', () => {
+                var _a;
+                const position = vscode_languageserver_1.Position.create(22, 19); // 'b' from m.b = new KlassB()
+                const token = klassCode.parser.getTokenAt(position);
+                (0, chai_1.expect)(token.text).to.equal('b');
+                const func = klassCode.getFunctionExpressionAtPosition(position);
+                let klass = (_a = klassCode.getClassFromToken(token, func, mainScope)) === null || _a === void 0 ? void 0 : _a.item;
+                (0, chai_1.expect)(klass).to.exist;
+                (0, chai_1.expect)(klass.getName(Parser_1.ParseMode.BrighterScript)).to.equal('KlassB');
+            });
+            it('gets correct class for variable', () => {
+                var _a;
+                const position = vscode_languageserver_1.Position.create(28, 17); // 'c' from c.b.getA().getInt()
+                const token = klassCode.parser.getTokenAt(position);
+                (0, chai_1.expect)(token.text).to.equal('c');
+                const func = klassCode.getFunctionExpressionAtPosition(position);
+                let klass = (_a = klassCode.getClassFromToken(token, func, mainScope)) === null || _a === void 0 ? void 0 : _a.item;
+                (0, chai_1.expect)(klass).to.exist;
+                (0, chai_1.expect)(klass.getName(Parser_1.ParseMode.BrighterScript)).to.equal('KlassC');
+            });
+            it('gets correct class for variable field', () => {
+                var _a;
+                const position = vscode_languageserver_1.Position.create(28, 19); // 'b' from c.b.getA()
+                const token = klassCode.parser.getTokenAt(position);
+                (0, chai_1.expect)(token.text).to.equal('b');
+                const func = klassCode.getFunctionExpressionAtPosition(position);
+                let klass = (_a = klassCode.getClassFromToken(token, func, mainScope)) === null || _a === void 0 ? void 0 : _a.item;
+                (0, chai_1.expect)(klass).to.exist;
+                (0, chai_1.expect)(klass.getName(Parser_1.ParseMode.BrighterScript)).to.equal('KlassB');
+            });
+            it('gets type and return class for variable function field', () => {
+                const position = vscode_languageserver_1.Position.create(28, 23); // 'getA' from c.b.getA()
+                const token = klassCode.parser.getTokenAt(position);
+                (0, chai_1.expect)(token.text).to.equal('getA');
+                const func = klassCode.getFunctionExpressionAtPosition(position);
+                let { type, symbolContainer } = klassCode.getSymbolTypeFromToken(token, func, mainScope);
+                (0, chai_1.expect)(type).to.exist;
+                (0, chai_1.expect)((0, reflection_1.isFunctionType)(type)).to.be.true;
+                (0, chai_1.expect)(symbolContainer).to.exist;
+                (0, chai_1.expect)((0, reflection_1.isCustomType)(symbolContainer)).to.be.true;
+                (0, chai_1.expect)(symbolContainer.name).to.equal('KlassA');
+            });
+            it('gets type and class for field from return value of function', () => {
+                const position = vscode_languageserver_1.Position.create(29, 30); // 'getInt' from c.b.getA().getInt()
+                const token = klassCode.parser.getTokenAt(position);
+                (0, chai_1.expect)(token.text).to.equal('getInt');
+                const func = klassCode.getFunctionExpressionAtPosition(position);
+                let { type, symbolContainer } = klassCode.getSymbolTypeFromToken(token, func, mainScope);
+                (0, chai_1.expect)(type).to.exist;
+                (0, chai_1.expect)((0, reflection_1.isFunctionType)(type)).to.be.true;
+                (0, chai_1.expect)(symbolContainer).to.be.undefined; // getInt() returns integer - no class reference at this point
+            });
+        });
+    });
     it('does not crash when child has field with same name as sub in parent', () => {
-        program.addOrReplaceFile('source/main.bs', `
+        program.setFile('source/main.bs', `
             class Parent
                 public function helloWorld()
                 end function
@@ -1090,7 +1405,7 @@ describe('BrsFile BrighterScript classes', () => {
         program.validate();
     });
     it('does not crash when child has method with same name as field in parent', () => {
-        program.addOrReplaceFile('source/main.bs', `
+        program.setFile('source/main.bs', `
             class Parent
                 public helloWorld as string
             end class