@@ -5,6 +5,8 @@ import * as typeScriptPlugin from 'prettier/plugins/typescript';
55import { format } from 'prettier/standalone' ;
66import { expect } from 'vitest' ;
77
8+ import type { Fragment } from '../src/utils' ;
9+
810const PRETTIER_OPTIONS : Parameters < typeof format > [ 1 ] = {
911 arrowParens : 'always' ,
1012 parser : 'typescript' ,
@@ -17,18 +19,22 @@ const PRETTIER_OPTIONS: Parameters<typeof format>[1] = {
1719 useTabs : false ,
1820} ;
1921
20- export function renderMapContains ( renderMap : RenderMap , key : string , expected : ( RegExp | string ) [ ] | RegExp | string ) {
22+ export function renderMapContains (
23+ renderMap : RenderMap < Fragment > ,
24+ key : string ,
25+ expected : ( RegExp | string ) [ ] | RegExp | string ,
26+ ) {
2127 expect ( renderMap . has ( key ) , `RenderMap is missing key "${ key } ".` ) . toBe ( true ) ;
22- return codeContains ( getFromRenderMap ( renderMap , key ) , expected ) ;
28+ return codeContains ( getFromRenderMap ( renderMap , key ) . content , expected ) ;
2329}
2430
2531export function renderMapDoesNotContain (
26- renderMap : RenderMap ,
32+ renderMap : RenderMap < Fragment > ,
2733 key : string ,
2834 expected : ( RegExp | string ) [ ] | RegExp | string ,
2935) {
3036 expect ( renderMap . has ( key ) , `RenderMap is missing key "${ key } ".` ) . toBe ( true ) ;
31- return codeDoesNotContain ( getFromRenderMap ( renderMap , key ) , expected ) ;
37+ return codeDoesNotContain ( getFromRenderMap ( renderMap , key ) . content , expected ) ;
3238}
3339
3440export async function codeContains ( actual : string , expected : ( RegExp | string ) [ ] | RegExp | string ) {
@@ -55,18 +61,22 @@ export async function codeDoesNotContain(actual: string, expected: (RegExp | str
5561 } ) ;
5662}
5763
58- export function renderMapContainsImports ( renderMap : RenderMap , key : string , expectedImports : Record < string , string [ ] > ) {
64+ export function renderMapContainsImports (
65+ renderMap : RenderMap < Fragment > ,
66+ key : string ,
67+ expectedImports : Record < string , string [ ] > ,
68+ ) {
5969 expect ( renderMap . has ( key ) , `RenderMap is missing key "${ key } ".` ) . toBe ( true ) ;
60- return codeContainsImports ( getFromRenderMap ( renderMap , key ) , expectedImports ) ;
70+ return codeContainsImports ( getFromRenderMap ( renderMap , key ) . content , expectedImports ) ;
6171}
6272
6373export function renderMapDoesNotContainImports (
64- renderMap : RenderMap ,
74+ renderMap : RenderMap < Fragment > ,
6575 key : string ,
6676 expectedImports : Record < string , string [ ] > ,
6777) {
6878 expect ( renderMap . has ( key ) , `RenderMap is missing key "${ key } ".` ) . toBe ( true ) ;
69- return codeDoesNotContainImports ( getFromRenderMap ( renderMap , key ) , expectedImports ) ;
79+ return codeDoesNotContainImports ( getFromRenderMap ( renderMap , key ) . content , expectedImports ) ;
7080}
7181
7282export async function codeContainsImports ( actual : string , expectedImports : Record < string , string [ ] > ) {
0 commit comments