11import 'jest' ;
22import * as ts from 'typescript' ;
33import * as fs from 'fs' ;
4- import * as glob from 'glob' ;
54import transformer from '../transformer' ;
65
76const printer = ts . createPrinter ( ) ;
@@ -34,11 +33,11 @@ ${indent(obj.transformed)}
3433} ) ;
3534
3635function expectTransform ( filename : string ) {
37- const content = fs . readFileSync ( './src/__tests__ /baselines/' + filename ) . toString ( ) ;
36+ const content = fs . readFileSync ( __dirname + ' /baselines/' + filename ) . toString ( ) ;
3837 const sourceFile = ts . createSourceFile ( filename , content , ts . ScriptTarget . Latest ) ;
3938 const source = printer . printFile ( sourceFile ) ;
40- const transformedFile = ts . transform ( sourceFile , [ transformer ] ) ;
41- const transformed = printer . printFile ( transformedFile . transformed [ 0 ] ) ;
39+ const transformedFile = ts . transform ( sourceFile , [ transformer ] ) . transformed [ 0 ] ;
40+ const transformed = printer . printFile ( transformedFile ) ;
4241
4342 const snapshot : TransformBaseline = {
4443 type : 'transform-baseline' ,
@@ -48,9 +47,10 @@ function expectTransform(filename: string) {
4847 transformed
4948 } ;
5049
51- expect ( snapshot ) . toMatchSnapshot ( ) ;
50+ expect ( snapshot ) . toMatchSnapshot ( filename ) ;
5251}
5352
54- const files = [ 'sample1.ts' ] ;
53+ const files = fs . readdirSync ( __dirname + '/baselines' )
54+ . filter ( f => f . toLowerCase ( ) . endsWith ( '.tsx' ) || f . toLowerCase ( ) . endsWith ( '.ts' ) ) ;
5555
5656files . forEach ( file => it ( file , ( ) => expectTransform ( file ) ) ) ;
0 commit comments