summaryrefslogtreecommitdiff
path: root/adservices/tests/unittest/service-core/src/com/android/adservices/data/measurement/AbstractDbIntegrationTest.java
diff options
context:
space:
mode:
authorAbhimanyu <abmehta@google.com>2022-08-04 22:39:33 +0000
committerAbhimanyu <abmehta@google.com>2022-08-12 20:02:32 +0000
commite3c647daab2afccd083ab0f781ee6ab3660f8ae5 (patch)
tree86c0ffd2dfa0acd8cdc84e89614fbd4ceff3f643 /adservices/tests/unittest/service-core/src/com/android/adservices/data/measurement/AbstractDbIntegrationTest.java
parentec48b483fcf3c482e704ece3f83161eb983d6a05 (diff)
downloadAdServices-e3c647daab2afccd083ab0f781ee6ab3660f8ae5.tar.gz
Account for attribution rate limit for fake reports
Bug: b/234775723 Test: atest com.android.adservices.service.measurement && atest com.android.adservices.data.measurement Details: When `Source` chooses to generate fake reports per the probabality, we want to attribute for them. For each fake reports generation positive decision, whether a state was chosen to insert 0 fake reports or more, we account a single attribution rate limit increment for each destination (app or web). Separate fake attribution for each destination's presence because their rate limits are separate. Also, updated the DAO's rate limit insertion method signature. Change-Id: I894b74235381048eaf6bd2ff68176c5eec9c6e2c
Diffstat (limited to 'adservices/tests/unittest/service-core/src/com/android/adservices/data/measurement/AbstractDbIntegrationTest.java')
-rw-r--r--adservices/tests/unittest/service-core/src/com/android/adservices/data/measurement/AbstractDbIntegrationTest.java11
1 files changed, 6 insertions, 5 deletions
diff --git a/adservices/tests/unittest/service-core/src/com/android/adservices/data/measurement/AbstractDbIntegrationTest.java b/adservices/tests/unittest/service-core/src/com/android/adservices/data/measurement/AbstractDbIntegrationTest.java
index f2a7eab4bd..8d9210bb97 100644
--- a/adservices/tests/unittest/service-core/src/com/android/adservices/data/measurement/AbstractDbIntegrationTest.java
+++ b/adservices/tests/unittest/service-core/src/com/android/adservices/data/measurement/AbstractDbIntegrationTest.java
@@ -24,6 +24,7 @@ import android.database.sqlite.SQLiteException;
import androidx.test.core.app.ApplicationProvider;
import com.android.adservices.data.DbHelper;
+import com.android.adservices.service.measurement.Attribution;
import com.android.adservices.service.measurement.EventReport;
import com.android.adservices.service.measurement.Source;
import com.android.adservices.service.measurement.Trigger;
@@ -214,7 +215,8 @@ public abstract class AbstractDbIntegrationTest {
for (EventReport eventReport : input.mEventReportList) {
insertToDb(eventReport, db);
}
- for (Attribution attr : input.mAttrRateLimitList) {
+ for (com.android.adservices.service.measurement.Attribution attr :
+ input.mAttrRateLimitList) {
insertToDb(attr, db);
}
for (AggregateReport aggregateReport : input.mAggregateReportList) {
@@ -323,10 +325,9 @@ public abstract class AbstractDbIntegrationTest {
throws SQLiteException {
ContentValues values = new ContentValues();
values.put(MeasurementTables.AttributionContract.ID, attribution.getId());
- values.put(MeasurementTables.AttributionContract.SOURCE_SITE,
- attribution.getSourceSite());
- values.put(MeasurementTables.AttributionContract.SOURCE_ORIGIN,
- attribution.getSourceOrigin());
+ values.put(MeasurementTables.AttributionContract.SOURCE_SITE, attribution.getSourceSite());
+ values.put(
+ MeasurementTables.AttributionContract.SOURCE_ORIGIN, attribution.getSourceOrigin());
values.put(MeasurementTables.AttributionContract.DESTINATION_SITE,
attribution.getDestinationSite());
values.put(MeasurementTables.AttributionContract.DESTINATION_ORIGIN,