@@ -250,22 +250,25 @@ public Object answer(InvocationOnMock invocation) {
250250 // given
251251 MockedStatic mockArgument = mockStatic (Arguments .class );
252252 MockedStatic mockReactApplicationContext = mockStatic (ReactApplicationContext .class );
253+ final OnSdkDismissCallback .DismissType dismissType = OnSdkDismissCallback .DismissType .CANCEL ;
254+ final OnSdkDismissCallback .ReportType reportType = OnSdkDismissCallback .ReportType .BUG ;
253255
254256 // when
255257 when (Arguments .createMap ()).thenReturn (new JavaOnlyMap ());
256258 mockBugReporting .when (() -> BugReporting .setOnDismissCallback (any (OnSdkDismissCallback .class ))).thenAnswer (new Answer () {
257259 public Object answer (InvocationOnMock invocation ) {
258260 InstabugUtil .sendEvent (any (),any (),any ());
259261 ((OnSdkDismissCallback ) invocation .getArguments ()[0 ])
260- .call (OnSdkDismissCallback . DismissType . CANCEL , OnSdkDismissCallback . ReportType . BUG );
262+ .call (dismissType , reportType );
261263 return null ;
262264 }});
263265 bugReportingModule .setOnDismissHandler (null );
264266
265267 // then
266268 WritableMap params = new JavaOnlyMap ();
267- params .putString ("dismissType" , OnSdkDismissCallback .DismissType .CANCEL .toString ());
268- params .putString ("reportType" , OnSdkDismissCallback .ReportType .BUG .toString ());
269+ params .putString ("dismissType" , ArgsRegistry .dismissTypes .getKey (dismissType ));
270+ params .putString ("reportType" , ArgsRegistry .sdkDismissReportTypes .getKey (reportType ));
271+
269272 verify (InstabugUtil .class ,VerificationModeFactory .times (1 ));
270273 InstabugUtil .sendEvent (any (ReactApplicationContext .class ), eq (Constants .IBG_POST_INVOCATION_HANDLER ), eq (params ));
271274 mockArgument .close ();
0 commit comments