11package com .d4rk .androidtutorials .java .utils ;
22
3- import static org .junit .Assert .assertEquals ;
43import static org .mockito .Mockito .mock ;
54import static org .mockito .Mockito .verify ;
65import static org .mockito .Mockito .when ;
1413import com .google .firebase .analytics .FirebaseAnalytics ;
1514
1615import org .junit .Test ;
17- import org .mockito .ArgumentCaptor ;
1816import org .mockito .MockedStatic ;
1917import org .mockito .Mockito ;
2018
19+ import java .util .EnumMap ;
2120import java .util .Map ;
2221
2322public class ConsentUtilsTest {
@@ -50,19 +49,18 @@ public void applyStoredConsent_readsPreferencesAndUpdatesFirebase() {
5049 verify (prefs ).getBoolean ("consent_ad_user_data" , true );
5150 verify (prefs ).getBoolean ("consent_ad_personalization" , true );
5251
53- ArgumentCaptor <Map <FirebaseAnalytics .ConsentType , FirebaseAnalytics .ConsentStatus >> captor =
54- ArgumentCaptor .forClass (Map .class ); // FIXME: Unchecked assignment: 'org.mockito.ArgumentCaptor' to 'org.mockito.ArgumentCaptor<java.util.Map<com.google.firebase.analytics.FirebaseAnalytics.ConsentType,com.google.firebase.analytics.FirebaseAnalytics.ConsentStatus>>' && Unchecked method 'forClass(Class<S>)' invocation
55- verify (analytics ).setConsent (captor .capture ());
56- Map <FirebaseAnalytics .ConsentType , FirebaseAnalytics .ConsentStatus > consentMap = captor .getValue ();
57-
58- assertEquals (FirebaseAnalytics .ConsentStatus .DENIED ,
59- consentMap .get (FirebaseAnalytics .ConsentType .ANALYTICS_STORAGE ));
60- assertEquals (FirebaseAnalytics .ConsentStatus .GRANTED ,
61- consentMap .get (FirebaseAnalytics .ConsentType .AD_STORAGE ));
62- assertEquals (FirebaseAnalytics .ConsentStatus .DENIED ,
63- consentMap .get (FirebaseAnalytics .ConsentType .AD_USER_DATA ));
64- assertEquals (FirebaseAnalytics .ConsentStatus .GRANTED ,
65- consentMap .get (FirebaseAnalytics .ConsentType .AD_PERSONALIZATION ));
52+ Map <FirebaseAnalytics .ConsentType , FirebaseAnalytics .ConsentStatus > expectedConsent =
53+ new EnumMap <>(FirebaseAnalytics .ConsentType .class );
54+ expectedConsent .put (FirebaseAnalytics .ConsentType .ANALYTICS_STORAGE ,
55+ FirebaseAnalytics .ConsentStatus .DENIED );
56+ expectedConsent .put (FirebaseAnalytics .ConsentType .AD_STORAGE ,
57+ FirebaseAnalytics .ConsentStatus .GRANTED );
58+ expectedConsent .put (FirebaseAnalytics .ConsentType .AD_USER_DATA ,
59+ FirebaseAnalytics .ConsentStatus .DENIED );
60+ expectedConsent .put (FirebaseAnalytics .ConsentType .AD_PERSONALIZATION ,
61+ FirebaseAnalytics .ConsentStatus .GRANTED );
62+
63+ verify (analytics ).setConsent (expectedConsent );
6664 }
6765 }
6866
@@ -85,19 +83,18 @@ public void updateFirebaseConsent_setsExpectedStatuses() {
8583 adUserDataConsent ,
8684 adPersonalizationConsent );
8785
88- ArgumentCaptor <Map <FirebaseAnalytics .ConsentType , FirebaseAnalytics .ConsentStatus >> captor =
89- ArgumentCaptor .forClass (Map .class ); // FIXME: Unchecked assignment: 'org.mockito.ArgumentCaptor' to 'org.mockito.ArgumentCaptor<java.util.Map<com.google.firebase.analytics.FirebaseAnalytics.ConsentType,com.google.firebase.analytics.FirebaseAnalytics.ConsentStatus>>' && Unchecked method 'forClass(Class<S>)' invocation
90- verify (analytics ).setConsent (captor .capture ());
91- Map <FirebaseAnalytics .ConsentType , FirebaseAnalytics .ConsentStatus > result = captor .getValue ();
92-
93- assertEquals (FirebaseAnalytics .ConsentStatus .GRANTED ,
94- result .get (FirebaseAnalytics .ConsentType .ANALYTICS_STORAGE ));
95- assertEquals (FirebaseAnalytics .ConsentStatus .DENIED ,
96- result .get (FirebaseAnalytics .ConsentType .AD_STORAGE ));
97- assertEquals (FirebaseAnalytics .ConsentStatus .GRANTED ,
98- result .get (FirebaseAnalytics .ConsentType .AD_USER_DATA ));
99- assertEquals (FirebaseAnalytics .ConsentStatus .DENIED ,
100- result .get (FirebaseAnalytics .ConsentType .AD_PERSONALIZATION ));
86+ Map <FirebaseAnalytics .ConsentType , FirebaseAnalytics .ConsentStatus > expectedConsent =
87+ new EnumMap <>(FirebaseAnalytics .ConsentType .class );
88+ expectedConsent .put (FirebaseAnalytics .ConsentType .ANALYTICS_STORAGE ,
89+ FirebaseAnalytics .ConsentStatus .GRANTED );
90+ expectedConsent .put (FirebaseAnalytics .ConsentType .AD_STORAGE ,
91+ FirebaseAnalytics .ConsentStatus .DENIED );
92+ expectedConsent .put (FirebaseAnalytics .ConsentType .AD_USER_DATA ,
93+ FirebaseAnalytics .ConsentStatus .GRANTED );
94+ expectedConsent .put (FirebaseAnalytics .ConsentType .AD_PERSONALIZATION ,
95+ FirebaseAnalytics .ConsentStatus .DENIED );
96+
97+ verify (analytics ).setConsent (expectedConsent );
10198 }
10299 }
103100
0 commit comments