Persist I/O usage state to the local database in CarWatchdogService.
Test: atest WatchdogStorageUnitTest
Bug: 192294393
Change-Id: I899ca6685370c65eb450de24b89611e994b966c8
Merged-In: I899ca6685370c65eb450de24b89611e994b966c8
(cherry picked from commit 5feb3831cb46290829f7b8732b359216a75c9970)
diff --git a/tests/carservice_unit_test/src/com/android/car/watchdog/IoUsageStatsEntrySubject.java b/tests/carservice_unit_test/src/com/android/car/watchdog/IoUsageStatsEntrySubject.java
new file mode 100644
index 0000000..e683618
--- /dev/null
+++ b/tests/carservice_unit_test/src/com/android/car/watchdog/IoUsageStatsEntrySubject.java
@@ -0,0 +1,165 @@
+/*
+ * Copyright (C) 2021 The Android Open Source Project
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package com.android.car.watchdog;
+
+import static com.google.common.truth.Truth.assertAbout;
+import static com.google.common.truth.Truth.assertWithMessage;
+
+import android.annotation.Nullable;
+import android.automotive.watchdog.IoOveruseStats;
+import android.automotive.watchdog.PerStateBytes;
+
+import com.google.common.truth.Correspondence;
+import com.google.common.truth.FailureMetadata;
+import com.google.common.truth.Subject;
+
+import java.util.Arrays;
+
+public final class IoUsageStatsEntrySubject extends Subject {
+ /* Boiler-plate Subject.Factory for IoUsageStatsEntrySubject. */
+ private static final Subject.Factory<
+ com.android.car.watchdog.IoUsageStatsEntrySubject,
+ Iterable<WatchdogStorage.IoUsageStatsEntry>> IO_OVERUSE_STATS_ENTRY_SUBJECT_FACTORY =
+ com.android.car.watchdog.IoUsageStatsEntrySubject::new;
+ private static final String NULL_ENTRY_STRING = "{NULL}";
+
+ private final Iterable<WatchdogStorage.IoUsageStatsEntry> mActual;
+
+ /* User-defined entry point. */
+ public static IoUsageStatsEntrySubject assertThat(
+ @Nullable Iterable<WatchdogStorage.IoUsageStatsEntry> stats) {
+ return assertAbout(IO_OVERUSE_STATS_ENTRY_SUBJECT_FACTORY).that(stats);
+ }
+
+ public static Subject.Factory<IoUsageStatsEntrySubject,
+ Iterable<WatchdogStorage.IoUsageStatsEntry>> resourceOveruseStats() {
+ return IO_OVERUSE_STATS_ENTRY_SUBJECT_FACTORY;
+ }
+
+ public void containsExactly(WatchdogStorage.IoUsageStatsEntry... stats) {
+ containsExactlyElementsIn(Arrays.asList(stats));
+ }
+
+ public void containsExactlyElementsIn(Iterable<WatchdogStorage.IoUsageStatsEntry> expected) {
+ assertWithMessage("Expected stats(%s) equals to actual stats(%s)", toString(expected),
+ toString(mActual)).that(mActual)
+ .comparingElementsUsing(Correspondence.from(
+ IoUsageStatsEntrySubject::isEquals, "is equal to"))
+ .containsExactlyElementsIn(expected);
+ }
+
+ public static boolean isEquals(WatchdogStorage.IoUsageStatsEntry actual,
+ WatchdogStorage.IoUsageStatsEntry expected) {
+ if (actual == null || expected == null) {
+ return (actual == null) && (expected == null);
+ }
+ return actual.userId == expected.userId && actual.packageName.equals(expected.packageName)
+ && actual.ioUsage.getTotalTimesKilled() == expected.ioUsage.getTotalTimesKilled()
+ && isEqualsPerStateBytes(actual.ioUsage.getForgivenWriteBytes(),
+ expected.ioUsage.getForgivenWriteBytes())
+ && isEqualsIoOveruseStats(actual.ioUsage.getInternalIoOveruseStats(),
+ expected.ioUsage.getInternalIoOveruseStats());
+ }
+
+ private static boolean isEqualsIoOveruseStats(android.automotive.watchdog.IoOveruseStats actual,
+ android.automotive.watchdog.IoOveruseStats expected) {
+ if (actual == null || expected == null) {
+ return (actual == null) && (expected == null);
+ }
+ return actual.killableOnOveruse == expected.killableOnOveruse
+ && isEqualsPerStateBytes(actual.remainingWriteBytes, expected.remainingWriteBytes)
+ && actual.startTime == expected.startTime
+ && actual.durationInSeconds == expected.durationInSeconds
+ && isEqualsPerStateBytes(actual.writtenBytes, expected.writtenBytes)
+ && actual.totalOveruses == expected.totalOveruses;
+ }
+
+ private static boolean isEqualsPerStateBytes(PerStateBytes actual, PerStateBytes expected) {
+ if (actual == null || expected == null) {
+ return (actual == null) && (expected == null);
+ }
+ return actual.foregroundBytes == expected.foregroundBytes
+ && actual.backgroundBytes == expected.backgroundBytes
+ && actual.garageModeBytes == expected.garageModeBytes;
+ }
+
+ private static String toString(Iterable<WatchdogStorage.IoUsageStatsEntry> entries) {
+ StringBuilder builder = new StringBuilder();
+ builder.append('[');
+ for (WatchdogStorage.IoUsageStatsEntry entry : entries) {
+ toStringBuilder(builder, entry).append(", ");
+ }
+ if (builder.length() > 1) {
+ builder.delete(builder.length() - 2, builder.length());
+ }
+ builder.append(']');
+ return builder.toString();
+ }
+
+ private static StringBuilder toStringBuilder(StringBuilder builder,
+ WatchdogStorage.IoUsageStatsEntry entry) {
+ builder.append("{UserId: ").append(entry.userId)
+ .append(", Package name: ").append(entry.packageName)
+ .append(", IoUsage: ");
+ toStringBuilder(builder, entry.ioUsage);
+ return builder.append('}');
+ }
+
+ private static StringBuilder toStringBuilder(StringBuilder builder,
+ WatchdogPerfHandler.PackageIoUsage ioUsage) {
+ builder.append("{IoOveruseStats: ");
+ toStringBuilder(builder, ioUsage.getInternalIoOveruseStats());
+ builder.append(", ForgivenWriteBytes: ");
+ toStringBuilder(builder, ioUsage.getForgivenWriteBytes());
+ return builder.append(", Total times killed: ").append(ioUsage.getTotalTimesKilled())
+ .append('}');
+ }
+
+ private static StringBuilder toStringBuilder(StringBuilder builder,
+ IoOveruseStats stats) {
+ if (stats == null) {
+ return builder.append(NULL_ENTRY_STRING);
+ }
+ builder.append("{Killable on overuse: ").append(stats.killableOnOveruse)
+ .append(", Remaining write bytes: ");
+ toStringBuilder(builder, stats.remainingWriteBytes);
+ builder.append(", Start time: ").append(stats.startTime)
+ .append(", Duration: ").append(stats.durationInSeconds).append(" seconds")
+ .append(", Total overuses: ").append(stats.totalOveruses)
+ .append(", Written bytes: ");
+ toStringBuilder(builder, stats.writtenBytes);
+ return builder.append('}');
+ }
+
+ private static StringBuilder toStringBuilder(
+ StringBuilder builder, PerStateBytes perStateBytes) {
+ if (perStateBytes == null) {
+ return builder.append(NULL_ENTRY_STRING);
+ }
+ return builder.append("{Foreground bytes: ").append(perStateBytes.foregroundBytes)
+ .append(", Background bytes: ").append(perStateBytes.backgroundBytes)
+ .append(", Garage mode bytes: ").append(perStateBytes.garageModeBytes)
+ .append('}');
+ }
+
+ private IoUsageStatsEntrySubject(FailureMetadata failureMetadata,
+ @Nullable Iterable<WatchdogStorage.IoUsageStatsEntry> iterableSubject) {
+ super(failureMetadata, iterableSubject);
+
+ mActual = iterableSubject;
+ }
+}