ts-simple-ast/dist/tests/compiler/base/bodyableNodeTests.js

TypeScript compiler wrapper for AST navigation and code generation.

"use strict";
Object.defineProperty(exports, "__esModule", { value: true });
const chai_1 = require("chai");
const testHelpers_1 = require("./../testHelpers");
describe("BodyableNode", () => {
    describe("setBodyText", () => {
        describe("using a writer", () => {
            function doTest(startCode, writerFunc, expectedCode) {
                const { firstChild, sourceFile } = testHelpers_1.getInfoFromText(startCode);
                firstChild.setBodyText(writerFunc);
                chai_1.expect(sourceFile.getFullText()).to.equal(expectedCode);
            }
            it("should set the body text with only one ending newline when writing a line at the end", () => {
                doTest("function myFunction() {}", writer => {
                    writer.writeLine("test;").writeLine("test2;");
                }, "function myFunction() {\n    test;\n    test2;\n}");
            });
            it("should set the body text with only one ending newline when not writing a line at the end", () => {
                doTest("function myFunction() {}", writer => {
                    writer.writeLine("test;").write("test2;");
                }, "function myFunction() {\n    test;\n    test2;\n}");
            });
        });
        describe("class method", () => {
            function doTest(startCode, newText, expectedCode) {
                const { firstChild, sourceFile } = testHelpers_1.getInfoFromText(startCode);
                firstChild.getInstanceMethods()[0].setBodyText(newText);
                chai_1.expect(sourceFile.getFullText()).to.equal(expectedCode);
            }
            it("should set the body text with the correct indentation", () => {
                doTest("class C {\n    myMethod() {\n        const test = 5;\n    }\n}\n", "var myVar;\nfunction myInnerFunction() {\n}", "class C {\n    myMethod() {\n        var myVar;\n        function myInnerFunction() {\n        }\n    }\n}\n");
            });
            it("should throw if no body exists", () => {
                const { firstChild, sourceFile } = testHelpers_1.getInfoFromText("declare class C { myMethod(); }");
                chai_1.expect(() => {
                    firstChild.getInstanceMethods()[0].setBodyText("'some text';");
                }).to.throw();
            });
        });
        describe("function", () => {
            function doTest(startCode, newText, expectedCode) {
                const { firstChild, sourceFile } = testHelpers_1.getInfoFromText(startCode);
                firstChild.setBodyText(newText);
                chai_1.expect(sourceFile.getFullText()).to.equal(expectedCode);
            }
            it("should set the body text when empty", () => {
                doTest("function myFunction() {}", "var myVar;", "function myFunction() {\n    var myVar;\n}");
            });
            it("should not indent an empty line", () => {
                doTest("function myFunction() {}", "var myVar;\n\n'test';", "function myFunction() {\n    var myVar;\n\n    'test';\n}");
            });
            it("should set the body text to empty when providing an empty string", () => {
                doTest("function myFunction() { /* some text */ }", "", "function myFunction() {\n}");
            });
            it("should set the body text when not empty", () => {
                doTest("function myFunction() {\n    function myInnerFunction() {}\n}", "var myVar;", "function myFunction() {\n    var myVar;\n}");
            });
            it("should insert multiple lines on the correct indentation", () => {
                doTest("function myFunction() {\n    \n}", "var myVar;\nlet mySecond;", "function myFunction() {\n    var myVar;\n    let mySecond;\n}");
            });
        });
        describe("nested function", () => {
            function doTest(startCode, newText, expectedCode) {
                const { firstChild, sourceFile } = testHelpers_1.getInfoFromText(startCode);
                firstChild.getFunctions()[0].setBodyText(newText);
                chai_1.expect(sourceFile.getFullText()).to.equal(expectedCode);
            }
            it("should insert into a function within a function", () => {
                doTest("function myFunction() {\n    function myInnerFunction() {}\n}", "var myVar;\nlet mySecond;", "function myFunction() {\n    function myInnerFunction() {\n        var myVar;\n        let mySecond;\n    }\n}");
            });
        });
    });
    describe("fill", () => {
        function doTest(startCode, structure, expectedCode) {
            const { firstChild, sourceFile } = testHelpers_1.getInfoFromText(startCode);
            firstChild.fill(structure);
            chai_1.expect(firstChild.getText()).to.equal(expectedCode);
        }
        it("should set the text of a function when using a string", () => {
            doTest("function myFunction() {\n}", { bodyText: "var myVar;" }, "function myFunction() {\n    var myVar;\n}");
        });
        it("should set the text of a function when using a writer", () => {
            doTest("function myFunction() {\n}", { bodyText: writer => writer.writeLine("var myVar;") }, "function myFunction() {\n    var myVar;\n}");
        });
    });
});

//# sourceMappingURL=bodyableNodeTests.js.map