102 lines
2.9 KiB
JavaScript
102 lines
2.9 KiB
JavaScript
import traverse from "../lib";
|
|
import assert from "assert";
|
|
import { parse } from "babylon";
|
|
import * as t from "@babel/types";
|
|
|
|
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"));
|
|
},
|
|
});
|
|
}).toThrow(
|
|
/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"),
|
|
]);
|
|
},
|
|
});
|
|
}).toThrow(
|
|
/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");
|
|
},
|
|
});
|
|
}).toThrow(
|
|
/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"));
|
|
},
|
|
});
|
|
}).toThrow(/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();
|
|
},
|
|
});
|
|
}).toThrow(
|
|
/You passed `path\.replaceWith\(\)` a falsy node, use `path\.remove\(\)` instead/,
|
|
);
|
|
});
|
|
});
|
|
});
|