@@ -4,7 +4,11 @@ import type { PlaywrightLauncher } from '@web/test-runner-playwright';
44import type { WebdriverLauncher } from '@web/test-runner-webdriver' ;
55
66import { defaultOptions , VisualRegressionPluginOptions } from './config' ;
7- import { visualDiffCommand , VisualDiffCommandResult } from './visualDiffCommand' ;
7+ import {
8+ visualDiffCommand ,
9+ VisualDiffCommandContext ,
10+ VisualDiffCommandResult ,
11+ } from './visualDiffCommand' ;
812import { VisualRegressionError } from './VisualRegressionError' ;
913
1014interface Payload {
@@ -49,6 +53,11 @@ export function visualRegressionPlugin(
4953 return ;
5054 }
5155
56+ const context : VisualDiffCommandContext = {
57+ testFile : session . testFile ,
58+ browser : session . browser . name ,
59+ } ;
60+
5261 if ( session . browser . type === 'puppeteer' ) {
5362 const browser = session . browser as ChromeLauncher ;
5463 const page = browser . getPage ( session . id ) ;
@@ -67,7 +76,7 @@ export function visualRegressionPlugin(
6776 }
6877
6978 const screenshot = ( await element . screenshot ( { encoding : 'binary' } ) ) as Buffer ;
70- return visualDiffCommand ( mergedOptions , screenshot , session . browser . name , payload . name ) ;
79+ return visualDiffCommand ( mergedOptions , screenshot , payload . name , context ) ;
7180 }
7281
7382 if ( session . browser . type === 'playwright' ) {
@@ -88,7 +97,7 @@ export function visualRegressionPlugin(
8897 }
8998
9099 const screenshot = await element . screenshot ( ) ;
91- return visualDiffCommand ( mergedOptions , screenshot , session . browser . name , payload . name ) ;
100+ return visualDiffCommand ( mergedOptions , screenshot , payload . name , context ) ;
92101 }
93102
94103 if ( session . browser . type === 'webdriver' ) {
@@ -106,7 +115,7 @@ export function visualRegressionPlugin(
106115 ` ;
107116
108117 const screenshot = await browser . takeScreenshot ( session . id , locator ) ;
109- return visualDiffCommand ( mergedOptions , screenshot , session . browser . name , payload . name ) ;
118+ return visualDiffCommand ( mergedOptions , screenshot , payload . name , context ) ;
110119 }
111120
112121 throw new Error (
0 commit comments