import * as babel from "@babel/core";
import plugin from "babel-plugin-transform-comment-to-assert";

function replace(code) {
  return babel
    .transform(code, { babelrc: false, plugins: [plugin] })
    .code.replace(/\n/g, " ")
    .replace(/\s+/g, " ")
    .trim();
}

replace("1 //=> 1");
//=> 'assert.deepEqual(1, 1);'

replace("a = { a: 1 }; a //=> {a: 1}");
//=> 'a = { a: 1 }; assert.deepEqual(a, { a: 1 });'

replace("(() => 'foo')() //=> 'bar'");
//=> "assert.deepEqual((() => 'foo')(), 'bar');"

replace("console.log('foo') //=> 'bar'");
//=> "console.log('foo'); assert.deepEqual('foo', 'bar');"

replace(`
const a = () => {
  throw new Error('fail');
};
a() // throws Error
`);
//=> `const a = () => { throw new Error('fail'); }; assert.throws(() => a(), Error);`

replace(`
const a = () => {
  throw new Error('fail');
};
a() // throws /fail/
`);
//*=> `const a = () => { throw new Error('fail'); }; assert.throws(() => a(), /fail/);`