import { expect } from 'chai';
import { prettyPrint, parse } from 'recast';
import * as types from 'ast-types';

function printAst(ast: types.ASTNode): string {
  return prettyPrint(ast, { tabWidth: 2 }).code;
}

export default function expectAstEqual(compareFrom: string, compareTo: string): void {
  const astA = parse(compareFrom);
  const astB = parse(compareTo);

  expect(printAst(astA)).to.eq(printAst(astB));
}