Donate to e Foundation | Murena handsets with /e/OS | Own a part of Murena! Learn more

Commit a46e0d41 authored by Tetiana Meronyk's avatar Tetiana Meronyk
Browse files

Add I/O tests for UserWakeupStore 

Tests verify that after adding entries to the UserWakeupStore they are correctly persisted in the file.

Bug: 330264023
Test: atest UserWakeupStoreTest
Flag: EXEMPT adding tests to existing feature
Change-Id: I7d6d5e8e86a90895c872322adcd9eb327e90bd0e
parent 173e287c

services/tests/mockingservicestests/res/xml/expectedUserWakeupList_1.xml

0 → 100644
+20 −0
Original line number Diff line number Diff line 
<?xml version="1.0" encoding="UTF-8"?>

<!--

  Copyright 2025 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.

  -->

<users version="1">

    <user user_id="10" />

    <user user_id="11" />

</users>
 
 No newline at end of file

services/tests/mockingservicestests/res/xml/expectedUserWakeupList_2.xml

0 → 100644
+19 −0
Original line number Diff line number Diff line 
<?xml version="1.0" encoding="UTF-8"?>

<!--

  Copyright 2025 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.

  -->

<users version="1">

    <user user_id="10" />

</users>
 
 No newline at end of file

services/tests/mockingservicestests/src/com/android/server/alarm/UserWakeupStoreTest.java

+54 −2
Original line number Diff line number Diff line
@@ -23,15 +23,21 @@ import static com.android.server.alarm.UserWakeupStore.USER_START_TIME_DEVIATION 


import static org.junit.Assert.assertEquals;

import static org.junit.Assert.assertTrue;

import static org.junit.Assert.fail;



import android.content.res.AssetManager;

import android.content.res.XmlResourceParser;

import android.os.Environment;

import android.os.FileUtils;

import android.os.SystemClock;

import android.util.Xml;



import androidx.test.platform.app.InstrumentationRegistry;

import androidx.test.runner.AndroidJUnit4;



import com.android.internal.os.BackgroundThread;

import com.android.internal.util.XmlUtils;

import com.android.modules.utils.TypedXmlPullParser;

import com.android.modules.utils.testing.ExtendedMockitoRule;



import org.junit.After;
@@ -40,8 +46,11 @@ import org.junit.Rule; 
import org.junit.Test;

import org.junit.runner.RunWith;

import org.mockito.Mockito;

import org.xmlpull.v1.XmlPullParserException;



import java.io.File;

import java.io.FileInputStream;

import java.io.IOException;

import java.util.ArrayList;

import java.util.Collections;

import java.util.concurrent.ExecutorService;
@@ -55,6 +64,7 @@ public class UserWakeupStoreTest { 
    private static final File TEST_SYSTEM_DIR = new File(InstrumentationRegistry

            .getInstrumentation().getContext().getDataDir(), "alarmsTestDir");

    private static final File ROOT_DIR = new File(TEST_SYSTEM_DIR, UserWakeupStore.ROOT_DIR_NAME);

    private static final String USERS_FILE_NAME = "usersWithAlarmClocks.xml";

    private ExecutorService mMockExecutorService = null;

    UserWakeupStore mUserWakeupStore;
@@ -105,7 +115,7 @@ public class UserWakeupStoreTest { 
        Collections.sort(userWakeups);

        assertEquals(userIds, userWakeups);



        final File file = new File(ROOT_DIR , "usersWithAlarmClocks.xml");

        final File file = new File(ROOT_DIR, USERS_FILE_NAME);

        assertTrue(file.exists());

    }
@@ -178,5 +188,47 @@ public class UserWakeupStoreTest { 
        assertTrue(mUserWakeupStore.getWakeupTimeForUser(USER_ID_2) - realtime

                < 3 * BUFFER_TIME_MS + USER_START_TIME_DEVIATION_LIMIT_MS);

    }

    //TODO: b/330264023 - Add tests for I/O in usersWithAlarmClocks.xml.



    @Test

    public void testWriteWakeups_xmlIsOrdered() {

        mUserWakeupStore.addUserWakeup(USER_ID_1, TEST_TIMESTAMP - 19_000);

        mUserWakeupStore.addUserWakeup(USER_ID_2, TEST_TIMESTAMP - 7_000);

        assertFileContentsMatchExpectedXml("res/xml/expectedUserWakeupList_1.xml");

    }



    @Test

    public void testWriteWakeups_containsOneEntryPerUser() {

        mUserWakeupStore.addUserWakeup(USER_ID_1, TEST_TIMESTAMP - 19_000);

        mUserWakeupStore.addUserWakeup(USER_ID_1, TEST_TIMESTAMP - 7_000);

        assertFileContentsMatchExpectedXml("res/xml/expectedUserWakeupList_2.xml");

    }



    private static void assertFileContentsMatchExpectedXml(String expectedContentsFile) {

        final File actual = new File(ROOT_DIR, USERS_FILE_NAME);

        AssetManager assetManager =

                InstrumentationRegistry.getInstrumentation().getContext().getAssets();

        try (FileInputStream actualFis = new FileInputStream(actual)) {

            final TypedXmlPullParser actualParser = Xml.resolvePullParser(actualFis);

            final XmlResourceParser expectedParser = assetManager.openXmlResourceParser(

                    expectedContentsFile);

            for (XmlUtils.nextElement(expectedParser), XmlUtils.nextElement(actualParser);

                    actualParser.getEventType() != XmlResourceParser.END_DOCUMENT

                            && expectedParser.getEventType() != XmlResourceParser.END_DOCUMENT;

                    XmlUtils.nextElement(actualParser), XmlUtils.nextElement(expectedParser)) {

                assertEquals("Event types differ ", expectedParser.getEventType(),

                        actualParser.getEventType());

                for (int i = 0; i < expectedParser.getAttributeCount(); i++) {

                    assertEquals("Attribute names differ at index " + i,

                            expectedParser.getAttributeName(i), actualParser.getAttributeName(i));

                    assertEquals("Attribute values differ at index " + i,

                            expectedParser.getAttributeValue(i), actualParser.getAttributeValue(i));

                }

            }

            // Ensure they are both at the end of document

            assertEquals("One of the parsers has not reached the EOF",

                    expectedParser.getEventType(), actualParser.getEventType());

        } catch (IOException | XmlPullParserException e) {

            fail(e.getLocalizedMessage());

        }

    }

}