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

Commit 8ea2ed21 authored by Paul Hu's avatar Paul Hu Committed by android-build-merger
Browse files

Merge "Fix flaky... 

Merge "Fix flaky IpMemoryStoreServiceTest.testFullMaintenance/testInterruptMaintenance" am: 6f0cef79 am: 5fa01563 am: 3af5611a
am: dd892d62

Change-Id: Iae901434f3c79371dd88686087a82b1b94364b99
parents 9b4b9f13 dd892d62

tests/unit/res/raw/test.db

0 → 100644
+204 KiB

File added.

No diff preview for this file type.

View file

tests/unit/src/com/android/server/connectivity/ipmemorystore/IpMemoryStoreServiceTest.java

+47 −11
Original line number Original line Diff line number Diff line
@@ -48,6 +48,10 @@ import androidx.test.InstrumentationRegistry; 
import androidx.test.filters.SmallTest;

import androidx.test.filters.SmallTest;

import androidx.test.runner.AndroidJUnit4;

import androidx.test.runner.AndroidJUnit4;





import com.android.server.networkstack.tests.R;



import libcore.io.Streams;



import org.junit.After;

import org.junit.After;

import org.junit.Before;

import org.junit.Before;

import org.junit.Test;

import org.junit.Test;
@@ -56,6 +60,9 @@ import org.mockito.Mock; 
import org.mockito.MockitoAnnotations;

import org.mockito.MockitoAnnotations;





import java.io.File;

import java.io.File;

import java.io.FileOutputStream;

import java.io.InputStream;

import java.io.OutputStream;

import java.lang.reflect.Modifier;

import java.lang.reflect.Modifier;

import java.net.Inet4Address;

import java.net.Inet4Address;

import java.net.Inet6Address;

import java.net.Inet6Address;
@@ -73,12 +80,15 @@ import java.util.function.Consumer; 
public class IpMemoryStoreServiceTest {

public class IpMemoryStoreServiceTest {

    private static final String TEST_CLIENT_ID = "testClientId";

    private static final String TEST_CLIENT_ID = "testClientId";

    private static final String TEST_DATA_NAME = "testData";

    private static final String TEST_DATA_NAME = "testData";

    private static final String TEST_DATABASE_NAME = "test.db";





    private static final int TEST_DATABASE_SIZE_THRESHOLD = 100 * 1024; //100KB

    private static final int TEST_DATABASE_SIZE_THRESHOLD = 100 * 1024; //100KB

    private static final int DEFAULT_TIMEOUT_MS = 5000;

    private static final int DEFAULT_TIMEOUT_MS = 5000;

    private static final int LONG_TIMEOUT_MS = 30000;

    private static final int LONG_TIMEOUT_MS = 30000;

    private static final int FAKE_KEY_COUNT = 20;

    private static final int FAKE_KEY_COUNT = 20;

    private static final long LEASE_EXPIRY_NULL = -1L;

    private static final long LEASE_EXPIRY_NULL = -1L;

    private static final long UNIX_TIME_MS_2000_01_01 = 946652400000L;

    private static final long UNIX_TIME_MS_2100_01_01 = 4102412400000L;

    private static final int MTU_NULL = -1;

    private static final int MTU_NULL = -1;

    private static final String[] FAKE_KEYS;

    private static final String[] FAKE_KEYS;

    private static final byte[] TEST_BLOB_DATA = new byte[] { -3, 6, 8, -9, 12,

    private static final byte[] TEST_BLOB_DATA = new byte[] { -3, 6, 8, -9, 12,
@@ -103,7 +113,7 @@ public class IpMemoryStoreServiceTest { 
        MockitoAnnotations.initMocks(this);

        MockitoAnnotations.initMocks(this);

        final Context context = InstrumentationRegistry.getContext();

        final Context context = InstrumentationRegistry.getContext();

        final File dir = context.getFilesDir();

        final File dir = context.getFilesDir();

        mDbFile = new File(dir, "test.db");

        mDbFile = new File(dir, TEST_DATABASE_NAME);

        doReturn(mDbFile).when(mMockContext).getDatabasePath(anyString());

        doReturn(mDbFile).when(mMockContext).getDatabasePath(anyString());

        doReturn(mMockJobScheduler).when(mMockContext)

        doReturn(mMockJobScheduler).when(mMockContext)

                .getSystemService(Context.JOB_SCHEDULER_SERVICE);

                .getSystemService(Context.JOB_SCHEDULER_SERVICE);
@@ -129,6 +139,16 @@ public class IpMemoryStoreServiceTest { 
        mDbFile.delete();

        mDbFile.delete();

    }

    }





    private void copyTestData(final File file) throws Exception {

        try (

                InputStream in = InstrumentationRegistry.getContext()

                        .getResources().openRawResource(R.raw.test);

                OutputStream out = new FileOutputStream(file)

        ) {

            Streams.copy(in, out);

        }

    }



    /** Helper method to build test network attributes */

    /** Helper method to build test network attributes */

    private static NetworkAttributes.Builder buildTestNetworkAttributes(

    private static NetworkAttributes.Builder buildTestNetworkAttributes(

            final Inet4Address ipAddress, final long expiry, final String hint,

            final Inet4Address ipAddress, final long expiry, final String hint,
@@ -321,20 +341,32 @@ public class IpMemoryStoreServiceTest { 
                })));

                })));

    }

    }





    /** Insert large data that db size will be over threshold for maintenance test usage. */

    /**

    private void insertFakeDataAndOverThreshold() {

     * This method is used to generate test.db file.

     *

     * Here are the steps to update the test.db file if you need to change value in DB.

     * 1. Create a new test like "testGenerateDB" and have only one line code "generateFakeData()".

     * 2. Comment out "mDbFile.delete()" in tearDown() method.

     * 3. Run "atest IpMemoryStoreServiceTest#testGenerateDB".

     * 4. Run "adb root; adb pull /data/data/com.android.server.networkstack.tests/files/test.db

     *    $YOUR_CODE_BASE/package/module/NetworkStack/tests/unit/res/raw/test.db".

     *

     */

    private void generateFakeData() {

        final int fakeDataCount = 1000;

        final int expiredRecordsCount = 100;

        try {

        try {

            final NetworkAttributes.Builder na = buildTestNetworkAttributes(

            final NetworkAttributes.Builder na = buildTestNetworkAttributes(

                    (Inet4Address) Inet4Address.getByName("1.2.3.4"), LEASE_EXPIRY_NULL,

                    (Inet4Address) Inet4Address.getByName("1.2.3.4"), LEASE_EXPIRY_NULL,

                    "hint1", Arrays.asList(Inet6Address.getByName("0A1C:2E40:480A::1CA6")),

                    "hint1", Arrays.asList(Inet6Address.getByName("0A1C:2E40:480A::1CA6")),

                    219);

                    219);

            final long time = System.currentTimeMillis() - 1;

            final long time = System.currentTimeMillis() - 1;

            for (int i = 0; i < 1000; i++) {

            for (int i = 0; i < fakeDataCount; i++) {

                int errorCode = IpMemoryStoreDatabase.storeNetworkAttributes(

                int errorCode = IpMemoryStoreDatabase.storeNetworkAttributes(

                        mService.mDb,

                        mService.mDb,

                        "fakeKey" + i,

                        "fakeKey" + i,

                        // Let first 100 records get expiry.

                        i < expiredRecordsCount

                        i < 100 ? time : time + TimeUnit.HOURS.toMillis(i),

                                ? UNIX_TIME_MS_2000_01_01 : UNIX_TIME_MS_2100_01_01 + i,

                        na.build());

                        na.build());

                assertEquals(errorCode, Status.SUCCESS);

                assertEquals(errorCode, Status.SUCCESS);
@@ -344,7 +376,7 @@ public class IpMemoryStoreServiceTest { 
                assertEquals(errorCode, Status.SUCCESS);

                assertEquals(errorCode, Status.SUCCESS);

            }

            }





            // After added 5000 records, db size is larger than fake threshold(100KB).

            // After inserted fake data, db size should be larger than threshold.

            assertTrue(mService.isDbSizeOverThreshold());

            assertTrue(mService.isDbSizeOverThreshold());

        } catch (final UnknownHostException e) {

        } catch (final UnknownHostException e) {

            fail("Insert fake data fail");

            fail("Insert fake data fail");
@@ -634,8 +666,10 @@ public class IpMemoryStoreServiceTest { 
    }

    }





    @Test

    @Test

    public void testFullMaintenance() {

    public void testFullMaintenance() throws Exception {

        insertFakeDataAndOverThreshold();

        copyTestData(mDbFile);

        // After inserted test data, db size should be larger than threshold.

        assertTrue(mService.isDbSizeOverThreshold());





        final InterruptMaintenance im = new InterruptMaintenance(0/* Fake JobId */);

        final InterruptMaintenance im = new InterruptMaintenance(0/* Fake JobId */);

        // Do full maintenance and then db size should go down and meet the threshold.

        // Do full maintenance and then db size should go down and meet the threshold.
@@ -651,8 +685,10 @@ public class IpMemoryStoreServiceTest { 
    }

    }





    @Test

    @Test

    public void testInterruptMaintenance() {

    public void testInterruptMaintenance() throws Exception {

        insertFakeDataAndOverThreshold();

        copyTestData(mDbFile);

        // After inserted test data, db size should be larger than threshold.

        assertTrue(mService.isDbSizeOverThreshold());





        final InterruptMaintenance im = new InterruptMaintenance(0/* Fake JobId */);

        final InterruptMaintenance im = new InterruptMaintenance(0/* Fake JobId */);