@@ -15,8 +15,8 @@ const mockPage = {
15
15
evaluateJavaScript : vi . fn ( ) ,
16
16
size : vi . fn ( ) . mockResolvedValue ( { width : 1920 , height : 1080 , dpr : 1 } ) ,
17
17
url : vi . fn ( ) . mockResolvedValue ( 'https://example.com' ) ,
18
- getContext : vi . fn ( ) . mockImplementation ( async function ( ) {
19
- return await WebPageContextParser ( this ) ;
18
+ getContext : vi . fn ( ) . mockImplementation ( async function ( this : WebPage ) {
19
+ return await WebPageContextParser ( this , { } ) ;
20
20
} ) ,
21
21
} as unknown as WebPage ;
22
22
@@ -31,7 +31,8 @@ describe('PageAgent freeze/unfreeze page context', () => {
31
31
// Create mock contexts
32
32
mockContext = {
33
33
size : { width : 1920 , height : 1080 , dpr : 1 } ,
34
- screenshotBase64 : 'mock-screenshot-base64-1' ,
34
+ screenshotBase64 :
35
+ 'data:image/png;base64,iVBORw0KGgoAAAANSUhEUgAAAAEAAAABCAYAAAAfFcSJAAAADUlEQVR42mNk+M9QDwADhgGAWjR9awAAAABJRU5ErkJggg==' ,
35
36
tree : [
36
37
{
37
38
id : 'element1' ,
@@ -46,7 +47,8 @@ describe('PageAgent freeze/unfreeze page context', () => {
46
47
47
48
mockContext2 = {
48
49
size : { width : 1920 , height : 1080 , dpr : 1 } ,
49
- screenshotBase64 : 'mock-screenshot-base64-2' ,
50
+ screenshotBase64 :
51
+ 'data:image/png;base64,iVBORw0KGgoAAAANSUhEUgAAAAEAAAABCAYAAAAfFcSJAAAADUlEQVR42mP8/5+hHgAHggJ/PchI7wAAAABJRU5ErkJggg==' ,
50
52
tree : [
51
53
{
52
54
id : 'element2' ,
@@ -138,7 +140,9 @@ describe('PageAgent freeze/unfreeze page context', () => {
138
140
// Frozen context should be marked
139
141
const frozenContext = ( agent as any ) . frozenUIContext ;
140
142
expect ( frozenContext . _isFrozen ) . toBe ( true ) ;
141
- expect ( frozenContext . screenshotBase64 ) . toBe ( mockContext . screenshotBase64 ) ;
143
+ expect ( frozenContext . screenshotBase64 ) . toBe (
144
+ 'data:image/png;base64,iVBORw0KGgoAAAANSUhEUgAAAAEAAAABCAYAAAAfFcSJAAAADUlEQVR42mNk+M9QDwADhgGAWjR9awAAAABJRU5ErkJggg==' ,
145
+ ) ;
142
146
expect ( frozenContext . tree ) . toBe ( mockContext . tree ) ;
143
147
} ) ;
144
148
0 commit comments