|
1 |
| -/*global jasmine*/ |
| 1 | +/* eslint-env jest */ |
2 | 2 |
|
3 |
| -var recast = require('recast'); |
| 3 | +const recast = require('recast'); |
| 4 | +const diff = require('jest-diff'); |
| 5 | +const utils = require('jest-matcher-utils'); |
| 6 | + |
| 7 | +const matchers = { |
| 8 | + toEqualASTNode: function (received, expected) { |
| 9 | + if (!expected || typeof expected !== 'object') { |
| 10 | + throw new Error( |
| 11 | + 'Expected value must be an object representing an AST node.\n' + |
| 12 | + 'Got ' + expected + '(' + typeof expected + ') instead.' |
| 13 | + ); |
| 14 | + } |
4 | 15 |
|
5 |
| -var matchers = { |
6 |
| - toEqualASTNode: function () { |
7 | 16 | return {
|
8 |
| - compare: function (actual, expected) { |
9 |
| - if (!expected || typeof expected !== 'object') { |
10 |
| - throw new Error( |
11 |
| - 'Expected value must be an object representing an AST node.\n' + |
12 |
| - 'Got ' + expected + '(' + typeof expected + ') instead.' |
13 |
| - ); |
14 |
| - } |
| 17 | + pass: recast.types.astNodesAreEquivalent(received, expected), |
| 18 | + message: () => { |
| 19 | + const diffString = diff(expected, received); |
15 | 20 |
|
16 |
| - return {pass: recast.types.astNodesAreEquivalent(actual, expected)}; |
17 |
| - } |
| 21 | + return ( |
| 22 | + 'Expected value to be (using ast-types):\n' + |
| 23 | + ` ${utils.printExpected(expected)}\n` + |
| 24 | + 'Received:\n' + |
| 25 | + ` ${utils.printReceived(received)}` + |
| 26 | + (diffString ? `\n\nDifference:\n\n${diffString}` : '') |
| 27 | + ); |
| 28 | + }, |
18 | 29 | };
|
19 |
| - } |
| 30 | + }, |
20 | 31 | };
|
21 | 32 |
|
22 |
| - |
23 |
| -jasmine.getEnv().beforeEach(function() { |
24 |
| - jasmine.addMatchers(matchers); |
25 |
| -}); |
| 33 | +expect.extend(matchers); |
0 commit comments