fixed instanceof on exported classes (#394)

tomblind · web-flow · commit b51e8261f1b3 · 2019-02-12T05:42:57.000-07:00
diff --git a/src/LuaTransformer.ts b/src/LuaTransformer.ts
@@ -531,7 +531,7 @@ export class LuaTransformer {
         );
         const assignClassPrototypeConstructor = tstl.createAssignmentStatement(
             classPrototypeConstructor,
-            className,
+            createClassNameWithExport(),
             statement
         );
         result.push(assignClassPrototypeConstructor);
diff --git a/test/translation/lua/modulesClassExport.lua b/test/translation/lua/modulesClassExport.lua
@@ -3,7 +3,7 @@ exports.TestClass = exports.TestClass or {};
 exports.TestClass.__index = exports.TestClass;
 exports.TestClass.prototype = exports.TestClass.prototype or {};
 exports.TestClass.prototype.__index = exports.TestClass.prototype;
-exports.TestClass.prototype.constructor = TestClass;
+exports.TestClass.prototype.constructor = exports.TestClass;
 exports.TestClass.new = function(...)
     local self = setmetatable({}, exports.TestClass.prototype);
     self:____constructor(...);
diff --git a/test/translation/lua/modulesClassWithMemberExport.lua b/test/translation/lua/modulesClassWithMemberExport.lua
@@ -3,7 +3,7 @@ exports.TestClass = exports.TestClass or {};
 exports.TestClass.__index = exports.TestClass;
 exports.TestClass.prototype = exports.TestClass.prototype or {};
 exports.TestClass.prototype.__index = exports.TestClass.prototype;
-exports.TestClass.prototype.constructor = TestClass;
+exports.TestClass.prototype.constructor = exports.TestClass;
 exports.TestClass.new = function(...)
     local self = setmetatable({}, exports.TestClass.prototype);
     self:____constructor(...);
diff --git a/test/unit/typechecking.spec.ts b/test/unit/typechecking.spec.ts
@@ -133,4 +133,17 @@ export class TypeCheckingTests {
             "Cannot use instanceof on classes with decorator '@extension' or '@metaExtension'."
         );
     }
+
+    @Test("instanceof export")
+    public instanceOfExport(): void
+    {
+        const result = util.transpileExecuteAndReturnExport(
+            `export class myClass {}
+            let inst = new myClass();
+            export const result = inst instanceof myClass;`,
+            "result"
+        );
+
+        Expect(result).toBeTruthy();
+    }
 }
