Fix ScribeTest.

Mock the consumption limit so that the Scribe can properly adjust
values.

Bug: 223553004
Test: atest com.android.server.tare.ScribeTest#testWriteHighLevelStateToDisk
Merged-In: I104eef812889ffb543132ae46c7ac869b12c9882
Change-Id: I36cde7d5e5a755e0607fb9d0152f739b31026435
This commit is contained in:
Kweku Adams
2022-02-16 17:24:48 +00:00
parent 7073d3bbe8
commit ff2793c086

View File

@@ -109,6 +109,7 @@ public class ScribeTest {
long lastReclamationTime = System.currentTimeMillis();
long remainingConsumableNarcs = 2000L;
long consumptionLimit = 500_000L;
when(mIrs.getConsumptionLimitLocked()).thenReturn(consumptionLimit);
Ledger ledger = mScribeUnderTest.getLedgerLocked(TEST_USER_ID, TEST_PACKAGE);
ledger.recordTransaction(new Ledger.Transaction(0, 1000L, 1, null, 2000, 0));
@@ -119,8 +120,13 @@ public class ScribeTest {
mScribeUnderTest.setConsumptionLimitLocked(consumptionLimit);
mScribeUnderTest.adjustRemainingConsumableNarcsLocked(
remainingConsumableNarcs - consumptionLimit);
mScribeUnderTest.writeImmediatelyForTesting();
assertEquals(lastReclamationTime, mScribeUnderTest.getLastReclamationTimeLocked());
assertEquals(remainingConsumableNarcs,
mScribeUnderTest.getRemainingConsumableNarcsLocked());
assertEquals(consumptionLimit, mScribeUnderTest.getSatiatedConsumptionLimitLocked());
mScribeUnderTest.writeImmediatelyForTesting();
mScribeUnderTest.loadFromDiskLocked();
assertEquals(lastReclamationTime, mScribeUnderTest.getLastReclamationTimeLocked());