feat: Add JsxSpreadAttribute.setExpression

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

Hide file tree

Showing 2 changed files with 22 additions and 0 deletions.
diff --git a/src/compiler/jsx/JsxSpreadAttribute.ts b/src/compiler/jsx/JsxSpreadAttribute.ts
@@ -1,5 +1,6 @@
 import { removeChildren } from "../../manipulation";
 import { JsxSpreadAttributeStructure } from "../../structures";
+import { WriterFunction } from "../../types";
 import { ts } from "../../typescript";
 import { callBaseGetStructure } from "../callBaseGetStructure";
 import { Node } from "../common";
@@ -13,6 +14,15 @@ export class JsxSpreadAttribute extends JsxSpreadAttributeBase<ts.JsxSpreadAttri
         return this.getNodeFromCompilerNode(this.compilerNode.expression);
     }
 
+    /**
+     * Sets the expression.
+     * @param textOrWriterFunction - Text to set the expression with.
+     */
+    setExpression(textOrWriterFunction: string | WriterFunction) {
+        this.getExpression().replaceWithText(textOrWriterFunction);
+        return this;
+    }
+
     /**
      * Removes the JSX spread attribute.
      */

diff --git a/src/tests/compiler/jsx/jsxSpreadAttributeTests.ts b/src/tests/compiler/jsx/jsxSpreadAttributeTests.ts
@@ -24,6 +24,18 @@ describe(nameof(JsxSpreadAttribute), () => {
         });
     });
 
+    describe(nameof<JsxSpreadAttribute>(n => n.setExpression), () => {
+        function doTest(text: string, expression: string, expected: string) {
+            const { descendant, sourceFile } = getInfo(text);
+            descendant.setExpression(expression);
+            expect(sourceFile.getFullText()).to.equal(expected);
+        }
+
+        it("should set", () => {
+            doTest(`var t = (<jsx {...test} />);`, "newExpr", `var t = (<jsx {...newExpr} />);`);
+        });
+    });
+
     describe(nameof<JsxSpreadAttribute>(n => n.getStructure), () => {
         function doTest(text: string, expectedStructure: MakeRequired<JsxSpreadAttributeStructure>) {
             const { descendant } = getInfo(text);