5858import static com .flutterwave .raveandroid .RaveConstants .fieldCvv ;
5959import static com .flutterwave .raveandroid .RaveConstants .fieldEmail ;
6060import static com .flutterwave .raveandroid .RaveConstants .fieldcardNoStripped ;
61- import static com .flutterwave .raveandroid .RaveConstants .invalidChargeCode ;
6261import static com .flutterwave .raveandroid .RaveConstants .noResponse ;
6362import static com .flutterwave .raveandroid .RaveConstants .success ;
6463import static com .flutterwave .raveandroid .RaveConstants .tokenExpired ;
6564import static com .flutterwave .raveandroid .RaveConstants .tokenNotFound ;
6665import static com .flutterwave .raveandroid .RaveConstants .transactionError ;
6766import static com .flutterwave .raveandroid .RaveConstants .unknownAuthmsg ;
68- import static com .flutterwave .raveandroid .RaveConstants .unknownResCodemsg ;
6967import static com .flutterwave .raveandroid .RaveConstants .validExpiryDatePrompt ;
7068import static org .junit .Assert .assertEquals ;
7169import static org .mockito .ArgumentMatchers .any ;
@@ -870,7 +868,7 @@ public void chargeCardWithSuggestedAuthModel_onSuccessWithPIN_showOTPLayoutCalle
870868 cardPresenter .chargeCardWithSuggestedAuthModel (generatePayload (), generateRandomString (), generateRandomString (), generateRandomString ());
871869 ArgumentCaptor <Callbacks .OnChargeRequestComplete > captor = ArgumentCaptor .forClass (Callbacks .OnChargeRequestComplete .class );
872870 verify (networkRequest ).charge (any (ChargeRequestBody .class ), captor .capture ());
873- captor .getAllValues ().get (0 ).onSuccess (generateValidChargeResponseWithAuth (PIN ), generateRandomString ());
871+ captor .getAllValues ().get (0 ).onSuccess (generateValidChargeResponseWithAuthModelUsed (PIN ), generateRandomString ());
874872 verify (view ).showOTPLayout (anyString (), anyString ());
875873
876874 }
@@ -881,8 +879,8 @@ public void chargeCardWithSuggestedAuthModel_onSuccessWithAVS_VBVSECURECODE_onAV
881879 cardPresenter .chargeCardWithSuggestedAuthModel (generatePayload (), generateRandomString (), generateRandomString (), generateRandomString ());
882880 ArgumentCaptor <Callbacks .OnChargeRequestComplete > captor = ArgumentCaptor .forClass (Callbacks .OnChargeRequestComplete .class );
883881 verify (networkRequest ).charge (any (ChargeRequestBody .class ), captor .capture ());
884- captor .getAllValues ().get (0 ).onSuccess (generateValidChargeResponseWithAuth (AVS_VBVSECURECODE ), generateRandomString ());
885- verify (view ).onAVSVBVSecureCodeModelUsed (anyString (), anyString ());
882+ captor .getAllValues ().get (0 ).onSuccess (generateValidChargeResponseWithAuthModelUsed (AVS_VBVSECURECODE ), generateRandomString ());
883+ verify (view ).onVBVAuthModelUsed (anyString (), anyString ());
886884
887885 }
888886
@@ -894,7 +892,7 @@ public void chargeCardWithSuggestedAuthModel_onSuccess_unknownResCodemsgReturned
894892 ArgumentCaptor <Callbacks .OnChargeRequestComplete > captor = ArgumentCaptor .forClass (Callbacks .OnChargeRequestComplete .class );
895893 verify (networkRequest ).charge (any (ChargeRequestBody .class ), captor .capture ());
896894 captor .getAllValues ().get (0 ).onSuccess (generateRandomChargeResponse (), generateRandomString ());
897- verify (view ).onPaymentError (unknownResCodemsg );
895+ verify (view ).onPaymentError (anyString () );
898896
899897 }
900898
@@ -916,7 +914,7 @@ public void chargeCardWithSuggestedAuthModel_onSuccess_invalidChargeCodeReturned
916914 ArgumentCaptor <Callbacks .OnChargeRequestComplete > captor = ArgumentCaptor .forClass (Callbacks .OnChargeRequestComplete .class );
917915 verify (networkRequest ).charge (any (ChargeRequestBody .class ), captor .capture ());
918916 captor .getAllValues ().get (0 ).onSuccess (generateNullChargeResponse (), generateRandomString ());
919- verify (view ).onPaymentError (invalidChargeCode );
917+ verify (view ).onPaymentError (anyString () );
920918
921919 }
922920
@@ -997,7 +995,7 @@ public void chargeCardWithAVSModel_onSuccess_02_authModelVBV_showOTPLayoutCalled
997995 captor .getAllValues ().get (0 ).onSuccess (chargeResponse , generateRandomString ());
998996
999997 //assert
1000- verify (view ).onAVSVBVSecureCodeModelUsed (authurl , flwref );
998+ verify (view ).onVBVAuthModelUsed (authurl , flwref );
1001999
10021000 }
10031001
@@ -1015,7 +1013,7 @@ public void chargeCardWithAVSModel_onSuccess_unknownResCodemsgReturned_onPayment
10151013 captor .getAllValues ().get (0 ).onSuccess (generateRandomChargeResponse (), generateRandomString ());
10161014
10171015 //assert
1018- verify (view ).onPaymentError (unknownResCodemsg );
1016+ verify (view ).onPaymentError (anyString () );
10191017
10201018 }
10211019
@@ -1049,7 +1047,7 @@ public void chargeCardWithAVSModel_onSuccess_invalidChargeCodeReturned_onPayment
10491047 captor .getAllValues ().get (0 ).onSuccess (generateNullChargeResponse (), generateRandomString ());
10501048
10511049 //assert
1052- verify (view ).onPaymentError (invalidChargeCode );
1050+ verify (view ).onPaymentError (any ( String . class ) );
10531051
10541052 }
10551053
@@ -1139,6 +1137,15 @@ private ChargeResponse generateValidChargeResponseWithAuth(String auth) {
11391137 return chargeResponse ;
11401138 }
11411139
1140+ private ChargeResponse generateValidChargeResponseWithAuthModelUsed (String auth ) {
1141+ ChargeResponse chargeResponse = generateRandomChargeResponse ();
1142+ chargeResponse .getData ().setAuthModelUsed (auth );
1143+ chargeResponse .getData ().setAuthurl (generateRandomString ());
1144+ chargeResponse .getData ().setFlwRef (generateRandomString ());
1145+ chargeResponse .getData ().setChargeResponseCode ("02" );
1146+ return chargeResponse ;
1147+ }
1148+
11421149 private Payload generatePayload () {
11431150 List <Meta > metas = new ArrayList <>();
11441151 List <SubAccount > subAccounts = new ArrayList <>();
0 commit comments