Refactor delete integration test into files

Bug: 237661937

Test: atest com.android.adservices.data.measurement.DeleteApiIntegrationTest
Change-Id: Idee8a975478977cd4e33c446d9ded67d7080037f

1 files changed, 28 insertions, 0 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 3aab1fb4da..698d367880 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
@@ -152,6 +152,34 @@ public abstract class AbstractDbIntegrationTest {
         return testCases;
     }
 
+    public static Collection<Object[]> getTestCasesFromMultipleStreams(
+            List<InputStream> inputStreams, CheckedJsonFunction prepareAdditionalData)
+            throws IOException, JSONException {
+        List<Object[]> testCases = new ArrayList<>();
+        for (InputStream inputStream : inputStreams) {
+            int size = inputStream.available();
+            byte[] buffer = new byte[size];
+            inputStream.read(buffer);
+            inputStream.close();
+            String json = new String(buffer, StandardCharsets.UTF_8);
+            JSONObject testObj = new JSONObject(json);
+            String name = testObj.getString("name");
+            JSONObject input = testObj.getJSONObject("input");
+            JSONObject output = testObj.getJSONObject("output");
+            DbState inputState = new DbState(input);
+            DbState outputState = new DbState(output);
+            if (prepareAdditionalData != null) {
+                testCases.add(
+                        new Object[] {
+                            inputState, outputState, prepareAdditionalData.apply(testObj), name
+                        });
+            } else {
+                testCases.add(new Object[] {inputState, outputState, name});
+            }
+        }
+        return testCases;
+    }
+
     /**
      * Compares two lists of the same measurement record type.
      * (Caller enforces the element types.)