ts-simple-ast
Version:
TypeScript compiler wrapper for AST navigation and code generation.
110 lines (108 loc) • 5.92 kB
JavaScript
"use strict";
Object.defineProperty(exports, "__esModule", { value: true });
const chai_1 = require("chai");
const compiler_1 = require("./../../../../compiler");
const testHelpers_1 = require("./../../testHelpers");
describe("StatementedNode", () => {
describe("insertClasses", () => {
function doTest(startCode, index, structures, expectedText) {
const { sourceFile } = testHelpers_1.getInfoFromText(startCode);
const result = sourceFile.insertClasses(index, structures);
chai_1.expect(sourceFile.getFullText()).to.equal(expectedText);
chai_1.expect(result.length).to.equal(structures.length);
}
it("should insert to an empty file", () => {
doTest("", 0, [{
name: "Identifier"
}], "class Identifier {\n}\n");
});
it("should insert at the start of a file", () => {
doTest("enum Enum {\n}\n", 0, [{ name: "Identifier" }], "class Identifier {\n}\n\nenum Enum {\n}\n");
});
it("should insert at the end of a file", () => {
doTest("enum Enum {\n}\n", 1, [{ name: "Identifier" }], "enum Enum {\n}\n\nclass Identifier {\n}\n");
});
it("should insert in the middle of children", () => {
doTest("class Identifier1 {\n}\n\nclass Identifier3 {\n}\n", 1, [{ name: "Identifier2" }], "class Identifier1 {\n}\n\nclass Identifier2 {\n}\n\nclass Identifier3 {\n}\n");
});
it("should insert multiple", () => {
doTest("class Identifier1 {\n}\n", 1, [{ name: "Identifier2" }, { name: "Identifier3" }], "class Identifier1 {\n}\n\nclass Identifier2 {\n}\n\nclass Identifier3 {\n}\n");
});
it("should have the expected text adding to non-source file", () => {
const { sourceFile } = testHelpers_1.getInfoFromText("namespace Namespace {\n}\n");
const namespaceDec = sourceFile.getNamespaces()[0];
namespaceDec.insertClasses(0, [{
name: "Identifier"
}]);
chai_1.expect(sourceFile.getFullText()).to.equal("namespace Namespace {\n class Identifier {\n }\n}\n");
});
});
describe("insertClass", () => {
function doTest(startCode, index, structure, expectedText) {
const { sourceFile } = testHelpers_1.getInfoFromText(startCode);
const result = sourceFile.insertClass(index, structure);
chai_1.expect(sourceFile.getFullText()).to.equal(expectedText);
chai_1.expect(result).to.be.instanceOf(compiler_1.ClassDeclaration);
}
it("should insert", () => {
doTest("class Identifier2 {\n}\n", 0, { name: "Identifier1" }, "class Identifier1 {\n}\n\nclass Identifier2 {\n}\n");
});
});
describe("addClasses", () => {
function doTest(startCode, structures, expectedText) {
const { sourceFile } = testHelpers_1.getInfoFromText(startCode);
const result = sourceFile.addClasses(structures);
chai_1.expect(sourceFile.getFullText()).to.equal(expectedText);
chai_1.expect(result.length).to.equal(structures.length);
}
it("should add multiple", () => {
doTest("class Identifier1 {\n}\n", [{ name: "Identifier2" }, { name: "Identifier3" }], "class Identifier1 {\n}\n\nclass Identifier2 {\n}\n\nclass Identifier3 {\n}\n");
});
});
describe("addClass", () => {
function doTest(startCode, structure, expectedText) {
const { sourceFile } = testHelpers_1.getInfoFromText(startCode);
const result = sourceFile.addClass(structure);
chai_1.expect(sourceFile.getFullText()).to.equal(expectedText);
chai_1.expect(result).to.be.instanceOf(compiler_1.ClassDeclaration);
}
it("should add one", () => {
doTest("class Identifier1 {\n}\n", { name: "Identifier2" }, "class Identifier1 {\n}\n\nclass Identifier2 {\n}\n");
});
});
describe("getClasses", () => {
const { sourceFile } = testHelpers_1.getInfoFromText("class Identifier1 {}\nclass Identifier2 { prop: string; }");
const classes = sourceFile.getClasses();
it("should have the expected number of classes", () => {
chai_1.expect(classes.length).to.equal(2);
});
it("should have correct type", () => {
chai_1.expect(classes[0]).to.be.instanceOf(compiler_1.ClassDeclaration);
});
});
describe("getClass", () => {
const { sourceFile } = testHelpers_1.getInfoFromText("class Identifier1 {}\nclass Identifier2 { prop: string; }");
it("should get a class by a name", () => {
chai_1.expect(sourceFile.getClass("Identifier2").getName()).to.equal("Identifier2");
});
it("should get a class by a search function", () => {
chai_1.expect(sourceFile.getClass(c => c.getName() === "Identifier1").getName()).to.equal("Identifier1");
});
it("should return undefined when the class doesn't exist", () => {
chai_1.expect(sourceFile.getClass("asdf")).to.be.undefined;
});
});
describe("getClassOrThrow", () => {
const { sourceFile } = testHelpers_1.getInfoFromText("class Identifier1 {}\nclass Identifier2 { prop: string; }");
it("should get a class by a name", () => {
chai_1.expect(sourceFile.getClassOrThrow("Identifier2").getName()).to.equal("Identifier2");
});
it("should get a class by a search function", () => {
chai_1.expect(sourceFile.getClassOrThrow(c => c.getName() === "Identifier1").getName()).to.equal("Identifier1");
});
it("should throw when the class doesn't exist", () => {
chai_1.expect(() => sourceFile.getClassOrThrow("asdf")).to.throw();
});
});
});
//# sourceMappingURL=classTests.js.map