98 lines
2.9 KiB
JavaScript
98 lines
2.9 KiB
JavaScript
import traverse from "../lib";
|
|
import assert from "assert";
|
|
import { parse } from "babylon";
|
|
import * as t from "babel-types";
|
|
import { expect } from "chai";
|
|
|
|
describe("path/replacement", function () {
|
|
describe("replaceWith", function () {
|
|
it("replaces declaration in ExportDefaultDeclaration node", function() {
|
|
const ast = parse("export default function() {};", { sourceType: "module" });
|
|
traverse(ast, {
|
|
FunctionDeclaration(path) {
|
|
path.replaceWith(t.arrayExpression([
|
|
t.functionExpression(
|
|
path.node.id,
|
|
path.node.params,
|
|
path.node.body,
|
|
path.node.generator,
|
|
path.node.async
|
|
),
|
|
]));
|
|
},
|
|
});
|
|
|
|
assert(ast.program.body[0].declaration.type == "ArrayExpression");
|
|
});
|
|
|
|
it("throws error when trying to replace Program with a non-Program node", function() {
|
|
const ast = parse("var x = 3;");
|
|
expect(function() {
|
|
traverse(ast, {
|
|
Program(path) {
|
|
path.replaceWith(t.identifier("a"));
|
|
},
|
|
});
|
|
}).to.throw(/You can only replace a Program root node with another Program node/);
|
|
});
|
|
|
|
it("throws error when used with an array of nodes", function() {
|
|
const ast = parse("function abc() {}; var test = 17;");
|
|
expect(function() {
|
|
traverse(ast, {
|
|
NumericLiteral(path) {
|
|
path.replaceWith([
|
|
t.identifier("should"),
|
|
t.identifier("never"),
|
|
t.identifier("happen"),
|
|
]);
|
|
},
|
|
});
|
|
}).to.throw(
|
|
/Don't use `path\.replaceWith\(\)` with an array of nodes, use `path\.replaceWithMultiple\(\)`/
|
|
);
|
|
});
|
|
|
|
it("throws error when used with source string", function() {
|
|
const ast = parse("(function() { var x = 3; var y = 17; var c = x + y; })();");
|
|
expect(function() {
|
|
traverse(ast, {
|
|
BinaryExpression(path) {
|
|
path.replaceWith("17 + 23");
|
|
},
|
|
});
|
|
}).to.throw(
|
|
/Don't use `path\.replaceWith\(\)` with a source string, use `path\.replaceWithSourceString\(\)`/
|
|
);
|
|
});
|
|
|
|
it("throws error when trying to replace removed node", function() {
|
|
const ast = parse("var z = 'abc';");
|
|
expect(function() {
|
|
traverse(ast, {
|
|
StringLiteral(path) {
|
|
path.remove();
|
|
path.replaceWith(t.identifier("p"));
|
|
},
|
|
});
|
|
}).to.throw(
|
|
/You can't replace this node, we've already removed it/,
|
|
);
|
|
});
|
|
|
|
it("throws error when passed a falsy value", function() {
|
|
const ast = parse("var z = 'abc';");
|
|
expect(function() {
|
|
traverse(ast, {
|
|
StringLiteral(path) {
|
|
path.replaceWith();
|
|
},
|
|
});
|
|
}).to.throw(
|
|
/You passed `path\.replaceWith\(\)` a falsy node, use `path\.remove\(\)` instead/,
|
|
);
|
|
});
|
|
|
|
});
|
|
});
|