1 files changed, 22 insertions, 0 deletions
diff --git a/apk/tests/src/com/android/healthconnect/controller/tests/utils/ApiExtensions.kt b/apk/tests/src/com/android/healthconnect/controller/tests/utils/ApiExtensions.kt
new file mode 100644
index 00000000..3f60e2d6
--- /dev/null
+++ b/apk/tests/src/com/android/healthconnect/controller/tests/utils/ApiExtensions.kt
@@ -0,0 +1,22 @@
+package com.android.healthconnect.controller.tests.utils
+
+import android.health.connect.ReadRecordsRequestUsingFilters
+import android.health.connect.TimeInstantRangeFilter
+import android.health.connect.datatypes.Record
+
+fun ReadRecordsRequestUsingFilters<Record>.fromDataSource(packageName: String): Boolean {
+    return this.dataOrigins.any { dataOrigin -> dataOrigin.packageName == packageName }
+}
+
+fun ReadRecordsRequestUsingFilters<Record>.fromTimeRange(
+    sourceTimeFilter: TimeInstantRangeFilter
+): Boolean {
+    val thisTimeRangeFilter = this.timeRangeFilter
+    if (thisTimeRangeFilter !is TimeInstantRangeFilter) return false
+    return thisTimeRangeFilter.startTime == sourceTimeFilter.startTime &&
+        thisTimeRangeFilter.endTime == sourceTimeFilter.endTime
+}
+
+fun ReadRecordsRequestUsingFilters<Record>.forDataType(dataType: Class<out Record>): Boolean {
+    return this.recordType == dataType
+}