@@ -52,17 +52,25 @@ export class Tonkeeper extends WalletExtension {
5252 await recoveryPhrasePage . getByRole ( 'button' , { name : 'Save' } ) . click ( )
5353 }
5454
55- async connect ( ) : Promise < void > {
55+ async connect ( confirm ?: boolean ) : Promise < void > {
5656 const acceptPage = await this . context . waitForEvent ( 'page' )
5757 await acceptPage . getByRole ( 'button' , { name : 'Connect wallet' } ) . click ( )
5858 await acceptPage . locator ( '#unlock-password' ) . fill ( this . password )
59- await acceptPage . getByRole ( 'button' , { name : 'Confirm' } ) . click ( )
59+ if ( confirm !== false ) {
60+ await acceptPage . getByRole ( 'button' , { name : 'Confirm' } ) . click ( )
61+ } else {
62+ await acceptPage . getByRole ( 'button' , { name : 'Cancel' } ) . click ( )
63+ }
6064 }
6165
62- async accept ( ) : Promise < void > {
66+ async accept ( confirm ?: boolean ) : Promise < void > {
6367 const acceptPage = await this . context . waitForEvent ( 'page' )
6468 await acceptPage . getByRole ( 'button' , { name : 'Confirm' } ) . click ( )
6569 await acceptPage . locator ( '#unlock-password' ) . fill ( this . password )
66- await acceptPage . getByRole ( 'button' , { name : 'Confirm' } ) . click ( )
70+ if ( confirm !== false ) {
71+ await acceptPage . getByRole ( 'button' , { name : 'Confirm' } ) . click ( )
72+ } else {
73+ await acceptPage . getByRole ( 'button' , { name : 'Cancel' } ) . click ( )
74+ }
6775 }
6876}
0 commit comments