1 files changed, 17 insertions, 1 deletions
diff --git a/adservices/tests/unittest/service-core/src/com/android/adservices/service/measurement/access/ManifestBasedAdtechAccessResolverTest.java b/adservices/tests/unittest/service-core/src/com/android/adservices/service/measurement/access/ManifestBasedAdtechAccessResolverTest.java
index 553cb029bb..02a7724e3b 100644
--- a/adservices/tests/unittest/service-core/src/com/android/adservices/service/measurement/access/ManifestBasedAdtechAccessResolverTest.java
+++ b/adservices/tests/unittest/service-core/src/com/android/adservices/service/measurement/access/ManifestBasedAdtechAccessResolverTest.java
@@ -68,13 +68,14 @@ public class ManifestBasedAdtechAccessResolverTest {
         mMockEnrollmentDao = mock(EnrollmentDao.class);
         doReturn(EnrollmentFixture.getValidEnrollment())
                 .when(mMockEnrollmentDao)
-                .getEnrollmentDataFromMeasurementUrl(eq(ENROLLED_AD_TECH_URL.toString()));
+                .getEnrollmentDataFromMeasurementUrl(eq(ENROLLED_AD_TECH_URL));
         mMockitoSession =
                 ExtendedMockito.mockitoSession()
                         .mockStatic(AppManifestConfigHelper.class)
                         .strictness(Strictness.LENIENT)
                         .initMocks(this)
                         .startMocking();
+        when(mFlags.isEnrollmentBlocklisted(any())).thenReturn(false);
     }
 
     @After
@@ -193,6 +194,21 @@ public class ManifestBasedAdtechAccessResolverTest {
     }
 
     @Test
+    public void isNotAllowed_enrollmentInBlocklist() {
+        mClassUnderTest =
+                new ManifestBasedAdtechAccessResolver(
+                        mMockEnrollmentDao, mFlags, PACKAGE, ENROLLED_AD_TECH_URL);
+        when(mFlags.isDisableMeasurementEnrollmentCheck()).thenReturn(false);
+        when(AppManifestConfigHelper.isAllowedAttributionAccess(any(), any(), any()))
+                .thenReturn(true);
+
+        String enrollmentId = EnrollmentFixture.getValidEnrollment().getEnrollmentId();
+        when(mFlags.isEnrollmentBlocklisted(enrollmentId)).thenReturn(true);
+
+        assertFalse(mClassUnderTest.isAllowed(CONTEXT));
+    }
+
+    @Test
     public void getErrorMessage() {
         mClassUnderTest =
                 new ManifestBasedAdtechAccessResolver(