feat: Add ExpressionedNode.setExpression

dsherret · Sep 22, 2018 · 12e0ca7 · 12e0ca7
1 parent 532131f
commit 12e0ca7
Show file tree

Hide file tree

Showing 2 changed files with 24 additions and 1 deletion.
diff --git a/src/compiler/expression/expressioned/ExpressionedNode.ts b/src/compiler/expression/expressioned/ExpressionedNode.ts
@@ -1,5 +1,6 @@
 import { Constructor } from "../../../types";
 import { ts } from "../../../typescript";
+import { WriterFunction } from "../../../types";
 import { Node } from "../../common";
 import { Expression } from "../Expression";
 
@@ -10,12 +11,22 @@ export interface ExpressionedNode {
      * Gets the expression.
      */
     getExpression(): Expression;
+    /**
+     * Sets the expression.
+     * @param textOrWriterFunction - Text to set the expression with.
+     */
+    setExpression(textOrWriterFunction: string | WriterFunction): this;
 }
 
 export function ExpressionedNode<T extends Constructor<ExpressionedNodeExtensionType>>(Base: T): Constructor<ExpressionedNode> & T {
     return class extends Base implements ExpressionedNode {
         getExpression() {
             return this.getNodeFromCompilerNode(this.compilerNode.expression);
         }
+
+        setExpression(textOrWriterFunction: string | WriterFunction) {
+            this.getExpression().replaceWithText(textOrWriterFunction);
+            return this;
+        }
     };
 }
diff --git a/src/tests/compiler/expression/expressioned/expressionedNodeTests.ts b/src/tests/compiler/expression/expressioned/expressionedNodeTests.ts
@@ -6,12 +6,24 @@ import { getInfoFromTextWithDescendant } from "../../testHelpers";
 describe(nameof(ExpressionedNode), () => {
     describe(nameof<ExpressionedNode>(n => n.getExpression), () => {
         function doTest(text: string, expectedText: string) {
-            const {descendant} = getInfoFromTextWithDescendant<ParenthesizedExpression>(text, SyntaxKind.ParenthesizedExpression);
+            const { descendant } = getInfoFromTextWithDescendant<ParenthesizedExpression>(text, SyntaxKind.ParenthesizedExpression);
             expect(descendant.getExpression().getText()).to.equal(expectedText);
         }
 
         it("should get the correct expression", () => {
             doTest("(x + 1)", "x + 1");
         });
     });
+
+    describe(nameof<ExpressionedNode>(n => n.setExpression), () => {
+        function doTest(text: string, newText: string, expected: string) {
+            const { descendant, sourceFile } = getInfoFromTextWithDescendant<ParenthesizedExpression>(text, SyntaxKind.ParenthesizedExpression);
+            descendant.setExpression(newText);
+            expect(sourceFile.getFullText()).to.equal(expected);
+        }
+
+        it("should set", () => {
+            doTest("const v = (x + 1)", "y + 2", "const v = (y + 2)");
+        });
+    });
 });