@@ -153,7 +153,7 @@ export function getArgvValue<ParseFuncReturnType>(
153153 * Creates a it function for the test file
154154 * @TODO : improve this
155155 */
156- export function itFunction ( { additionalSearchParams, clip, clipSelector, folders : { baselineFolder } , framework, skipStories, storyData, storybookUrl } : CreateItContent ) {
156+ export function itFunction ( { additionalSearchParams, clip, clipSelector, compareOptions , folders : { baselineFolder } , framework, skipStories, storyData, storybookUrl } : CreateItContent ) {
157157 const { id } = storyData
158158 const screenshotType = clip ? 'n element' : ' viewport'
159159 const DEFAULT_IT_TEXT = 'it'
@@ -173,7 +173,8 @@ export function itFunction({ additionalSearchParams, clip, clipSelector, folders
173173
174174 // Setup the folder structure
175175 const { category, component } = extractCategoryAndComponent ( id )
176- const methodOptions = {
176+ const checkMethodOptions = {
177+ ...compareOptions ,
177178 baselineFolder : join ( baselineFolder , `./${ category } /${ component } /` ) ,
178179 }
179180
@@ -186,8 +187,8 @@ export function itFunction({ additionalSearchParams, clip, clipSelector, folders
186187 additionalSearchParams: new URLSearchParams('${ additionalSearchParams . toString ( ) } '),
187188 });
188189 ${ clip
189- ? `await expect($('${ clipSelector } ')).toMatchElementSnapshot('${ id } -element', ${ JSON . stringify ( methodOptions ) } )`
190- : `await expect(browser).toMatchScreenSnapshot('${ id } ', ${ JSON . stringify ( methodOptions ) } )` }
190+ ? `await expect($('${ clipSelector } ')).toMatchElementSnapshot('${ id } -element', ${ JSON . stringify ( checkMethodOptions ) } )`
191+ : `await expect(browser).toMatchScreenSnapshot('${ id } ', ${ JSON . stringify ( checkMethodOptions ) } )` }
191192 });
192193 `
193194 return it
@@ -211,11 +212,11 @@ export function writeTestFile(directoryPath: string, fileID: string, testContent
211212 * Create the test content
212213 */
213214export function createTestContent (
214- { additionalSearchParams, clip, clipSelector, folders, framework, skipStories, stories, storybookUrl } : CreateTestContent ,
215+ { additionalSearchParams, clip, clipSelector, compareOptions , folders, framework, skipStories, stories, storybookUrl } : CreateTestContent ,
215216 // For testing purposes only
216217 itFunc = itFunction
217218) : string {
218- const itFunctionOptions = { additionalSearchParams, clip, clipSelector, folders, framework, skipStories, storybookUrl }
219+ const itFunctionOptions = { additionalSearchParams, clip, clipSelector, compareOptions , folders, framework, skipStories, storybookUrl }
219220
220221 return stories . reduce ( ( acc , storyData ) => acc + itFunc ( { ...itFunctionOptions , storyData } ) , '' )
221222}
@@ -318,14 +319,14 @@ function filterStories(storiesJson: Stories): StorybookData[] {
318319 * Create the test files
319320 */
320321export function createTestFiles (
321- { additionalSearchParams, clip, clipSelector, directoryPath, folders, framework, numShards, skipStories, storiesJson, storybookUrl } : CreateTestFileOptions ,
322+ { additionalSearchParams, clip, clipSelector, compareOptions , directoryPath, folders, framework, numShards, skipStories, storiesJson, storybookUrl } : CreateTestFileOptions ,
322323 // For testing purposes only
323324 createTestCont = createTestContent ,
324325 createFileD = createFileData ,
325326 writeTestF = writeTestFile
326327) {
327328 const fileNamePrefix = 'visual-storybook'
328- const createTestContentData = { additionalSearchParams, clip, clipSelector, folders, framework, skipStories, stories : storiesJson , storybookUrl }
329+ const createTestContentData = { additionalSearchParams, clip, clipSelector, compareOptions , folders, framework, skipStories, stories : storiesJson , storybookUrl }
329330
330331 if ( numShards === 1 ) {
331332 const testContent = createTestCont ( createTestContentData )
0 commit comments