1 files changed, 159 insertions, 0 deletions
diff --git a/adservices/tests/unittest/service-core/src/com/android/adservices/service/measurement/reporting/AggregateReportingJobHandlerTest.java b/adservices/tests/unittest/service-core/src/com/android/adservices/service/measurement/reporting/AggregateReportingJobHandlerTest.java
index e5700638e..fd7668d93 100644
--- a/adservices/tests/unittest/service-core/src/com/android/adservices/service/measurement/reporting/AggregateReportingJobHandlerTest.java
+++ b/adservices/tests/unittest/service-core/src/com/android/adservices/service/measurement/reporting/AggregateReportingJobHandlerTest.java
@@ -54,6 +54,7 @@ import com.android.adservices.service.measurement.aggregation.AggregateCryptoFix
 import com.android.adservices.service.measurement.aggregation.AggregateEncryptionKey;
 import com.android.adservices.service.measurement.aggregation.AggregateEncryptionKeyManager;
 import com.android.adservices.service.measurement.aggregation.AggregateReport;
+import com.android.adservices.service.measurement.aggregation.AggregateReportFixture;
 import com.android.adservices.service.measurement.util.UnsignedLong;
 import com.android.adservices.service.stats.AdServicesLogger;
 import com.android.adservices.service.stats.MeasurementReportsStats;
@@ -989,6 +990,164 @@ public class AggregateReportingJobHandlerTest {
         verify(mTransaction, times(1)).end();
     }
 
+    @Test
+    public void performReport_normalReportWithDebugKeys_hasDebugModeEnabled()
+            throws DatastoreException, IOException, JSONException {
+        AggregateReport aggregateReport =
+                AggregateReportFixture.getValidAggregateReportBuilder()
+                        .setSourceDebugKey(
+                                AggregateReportFixture.ValidAggregateReportParams.SOURCE_DEBUG_KEY)
+                        .setTriggerDebugKey(
+                                AggregateReportFixture.ValidAggregateReportParams.TRIGGER_DEBUG_KEY)
+                        .build();
+        executeDebugModeVerification(aggregateReport, mSpyAggregateReportingJobHandler, "enabled");
+        verify(mMeasurementDao, times(1))
+                .markAggregateReportStatus(
+                        eq(aggregateReport.getId()), eq(AggregateReport.Status.DELIVERED));
+    }
+
+    @Test
+    public void performReport_normalReportWithOnlySourceDebugKey_hasDebugModeNull()
+            throws DatastoreException, IOException, JSONException {
+        // Setup
+        AggregateReport aggregateReport =
+                AggregateReportFixture.getValidAggregateReportBuilder()
+                        .setSourceDebugKey(
+                                AggregateReportFixture.ValidAggregateReportParams.SOURCE_DEBUG_KEY)
+                        .setTriggerDebugKey(null)
+                        .build();
+        executeDebugModeVerification(aggregateReport, mSpyAggregateReportingJobHandler, "");
+        verify(mMeasurementDao, times(1))
+                .markAggregateReportStatus(
+                        eq(aggregateReport.getId()), eq(AggregateReport.Status.DELIVERED));
+    }
+
+    @Test
+    public void performReport_normalReportWithOnlyTriggerDebugKey_hasDebugModeNull()
+            throws DatastoreException, IOException, JSONException {
+        AggregateReport aggregateReport =
+                AggregateReportFixture.getValidAggregateReportBuilder()
+                        .setSourceDebugKey(null)
+                        .setTriggerDebugKey(
+                                AggregateReportFixture.ValidAggregateReportParams.TRIGGER_DEBUG_KEY)
+                        .build();
+        executeDebugModeVerification(aggregateReport, mSpyAggregateReportingJobHandler, "");
+        verify(mMeasurementDao, times(1))
+                .markAggregateReportStatus(
+                        eq(aggregateReport.getId()), eq(AggregateReport.Status.DELIVERED));
+    }
+
+    @Test
+    public void performReport_normalReportWithNoDebugKey_hasDebugModeNull()
+            throws DatastoreException, IOException, JSONException {
+        AggregateReport aggregateReport =
+                AggregateReportFixture.getValidAggregateReportBuilder()
+                        .setSourceDebugKey(null)
+                        .setTriggerDebugKey(null)
+                        .build();
+        executeDebugModeVerification(aggregateReport, mSpyAggregateReportingJobHandler, "");
+        verify(mMeasurementDao, times(1))
+                .markAggregateReportStatus(
+                        eq(aggregateReport.getId()), eq(AggregateReport.Status.DELIVERED));
+    }
+
+    @Test
+    public void performReport_debugReportWithDebugKeys_hasDebugModeEnabled()
+            throws DatastoreException, IOException, JSONException {
+        AggregateReport aggregateReport =
+                AggregateReportFixture.getValidAggregateReportBuilder()
+                        .setSourceDebugKey(
+                                AggregateReportFixture.ValidAggregateReportParams.SOURCE_DEBUG_KEY)
+                        .setTriggerDebugKey(
+                                AggregateReportFixture.ValidAggregateReportParams.TRIGGER_DEBUG_KEY)
+                        .build();
+        executeDebugModeVerification(
+                aggregateReport, mSpyDebugAggregateReportingJobHandler, "enabled");
+        verify(mMeasurementDao, times(1))
+                .markAggregateDebugReportDelivered(eq(aggregateReport.getId()));
+    }
+
+    @Test
+    public void performReport_debugReportWithOnlySourceDebugKey_hasDebugModeNull()
+            throws DatastoreException, IOException, JSONException {
+        AggregateReport aggregateReport =
+                AggregateReportFixture.getValidAggregateReportBuilder()
+                        .setSourceDebugKey(
+                                AggregateReportFixture.ValidAggregateReportParams.SOURCE_DEBUG_KEY)
+                        .setTriggerDebugKey(null)
+                        .build();
+        executeDebugModeVerification(aggregateReport, mSpyDebugAggregateReportingJobHandler, "");
+        verify(mMeasurementDao, times(1))
+                .markAggregateDebugReportDelivered(eq(aggregateReport.getId()));
+    }
+
+    @Test
+    public void performReport_debugReportWithOnlyTriggerDebugKey_hasDebugModeNull()
+            throws DatastoreException, IOException, JSONException {
+        AggregateReport aggregateReport =
+                AggregateReportFixture.getValidAggregateReportBuilder()
+                        .setSourceDebugKey(null)
+                        .setTriggerDebugKey(
+                                AggregateReportFixture.ValidAggregateReportParams.TRIGGER_DEBUG_KEY)
+                        .build();
+        executeDebugModeVerification(aggregateReport, mSpyDebugAggregateReportingJobHandler, "");
+    }
+
+    @Test
+    public void performReport_debugReportWithNoDebugKey_hasDebugModeNull()
+            throws DatastoreException, IOException, JSONException {
+        AggregateReport aggregateReport =
+                AggregateReportFixture.getValidAggregateReportBuilder()
+                        .setSourceDebugKey(null)
+                        .setTriggerDebugKey(null)
+                        .build();
+        executeDebugModeVerification(aggregateReport, mSpyDebugAggregateReportingJobHandler, "");
+        verify(mMeasurementDao, times(1))
+                .markAggregateDebugReportDelivered(eq(aggregateReport.getId()));
+    }
+
+    private void executeDebugModeVerification(
+            AggregateReport aggregateReport,
+            AggregateReportingJobHandler aggregateReportingJobHandler,
+            String expectedDebugMode)
+            throws DatastoreException, IOException, JSONException {
+        when(mMeasurementDao.getAggregateReport(aggregateReport.getId()))
+                .thenReturn(aggregateReport);
+        doReturn(HttpURLConnection.HTTP_OK)
+                .when(aggregateReportingJobHandler)
+                .makeHttpPostRequest(eq(REPORTING_URI), Mockito.any());
+
+        doNothing()
+                .when(mMeasurementDao)
+                .markAggregateReportStatus(
+                        aggregateReport.getId(), AggregateReport.Status.DELIVERED);
+        ArgumentCaptor<JSONObject> aggregateReportBodyCaptor =
+                ArgumentCaptor.forClass(JSONObject.class);
+
+        // Execution
+        Assert.assertEquals(
+                AdServicesStatusUtils.STATUS_SUCCESS,
+                aggregateReportingJobHandler.performReport(
+                        aggregateReport.getId(),
+                        AggregateCryptoFixture.getKey(),
+                        new ReportingStatus()));
+
+        // Assertion
+        verify(aggregateReportingJobHandler)
+                .makeHttpPostRequest(eq(REPORTING_URI), aggregateReportBodyCaptor.capture());
+        verify(mTransaction, times(2)).begin();
+        verify(mTransaction, times(2)).end();
+
+        JSONObject aggregateReportBody = aggregateReportBodyCaptor.getValue();
+        JSONObject sharedInfo =
+                new JSONObject(
+                        aggregateReportBody.getString(
+                                AggregateReportBody.PayloadBodyKeys.SHARED_INFO));
+        assertEquals(
+                expectedDebugMode,
+                sharedInfo.optString(AggregateReportBody.SharedInfoKeys.DEBUG_MODE));
+    }
+
     private static JSONObject createASampleAggregateReportBody(AggregateReport aggregateReport)
             throws JSONException {
         return new AggregateReportBody.Builder()