LoyaltyCardViewActivityTest: Update test to handle MaterialDatePicker…

… migration Signed-off-by: Aayush Gupta <[email protected]>
CatimaLoyalty · Oct 24, 2023 · cc74ac8 · cc74ac8
1 parent 6905efc
commit cc74ac8
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/app/src/test/java/protect/card_locker/LoyaltyCardViewActivityTest.java b/app/src/test/java/protect/card_locker/LoyaltyCardViewActivityTest.java
@@ -2,6 +2,7 @@
 
 import android.app.Activity;
 import android.app.DatePickerDialog;
+import android.app.Dialog;
 import android.content.Context;
 import android.content.Intent;
 import android.content.SharedPreferences;
@@ -725,9 +726,9 @@ public void startWithLoyaltyCardNoExpirySetExpiry() throws IOException {
 
         shadowOf(getMainLooper()).idle();
 
-        DatePickerDialog datePickerDialog = (DatePickerDialog) (ShadowDialog.getLatestDialog());
+        Dialog datePickerDialog = ShadowDialog.getLatestDialog();
         assertNotNull(datePickerDialog);
-        datePickerDialog.getButton(DatePickerDialog.BUTTON_POSITIVE).performClick();
+        datePickerDialog.findViewById(com.google.android.material.R.id.confirm_button).performClick();
 
         shadowOf(getMainLooper()).idle();