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

Commit e25dbaa6 authored by Christine Franks's avatar Christine Franks
Browse files

Fix broken retaildemo tests

Bug: 38162695
Test: runtest frameworks-services -c
com.android.server.retaildemo.RetailDemoModeServiceTest &&
runtest frameworks-services -c
com.android.server.retaildemo.PreloadAppsInstallerTest

Change-Id: Iac60202414753b3e04e89cb29271cbd150a3acdd
parent 1c38d294
Loading
Loading
Loading
Loading
+9 −4
Original line number Diff line number Diff line
@@ -24,6 +24,7 @@ import static org.mockito.Mockito.verify;
import static org.mockito.Mockito.when;

import android.content.Context;
import android.content.ContextWrapper;
import android.content.pm.IPackageInstallObserver2;
import android.content.pm.IPackageManager;
import android.content.pm.PackageManager;
@@ -32,6 +33,7 @@ import android.os.UserHandle;
import android.provider.Settings;
import android.support.test.InstrumentationRegistry;
import android.support.test.filters.SmallTest;
import android.support.test.runner.AndroidJUnit4;
import android.test.mock.MockContentResolver;

import com.android.internal.util.test.FakeSettingsProvider;
@@ -40,20 +42,20 @@ import org.junit.After;
import org.junit.Before;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.junit.runners.JUnit4;
import org.mockito.ArgumentCaptor;
import org.mockito.Mock;
import org.mockito.Mockito;
import org.mockito.MockitoAnnotations;

import java.io.File;
import java.util.ArrayList;

@RunWith(JUnit4.class)
@RunWith(AndroidJUnit4.class)
@SmallTest
public class PreloadAppsInstallerTest {
    private static final int TEST_DEMO_USER = 111;

    private @Mock Context mContext;
    private Context mContext;
    private @Mock IPackageManager mIpm;
    private MockContentResolver mContentResolver;
    private File mPreloadsAppsDirectory;
@@ -66,6 +68,7 @@ public class PreloadAppsInstallerTest {
    @Before
    public void setUp() throws Exception {
        MockitoAnnotations.initMocks(this);
        mContext = Mockito.spy(new ContextWrapper(InstrumentationRegistry.getTargetContext()));
        mContentResolver = new MockContentResolver(mContext);
        mContentResolver.addProvider(Settings.AUTHORITY, new FakeSettingsProvider());
        when(mContext.getContentResolver()).thenReturn(mContentResolver);
@@ -89,8 +92,10 @@ public class PreloadAppsInstallerTest {

    @After
    public void tearDown() {
        if (mPreloadsAppsDirectory != null) {
            FileUtils.deleteContentsAndDir(mPreloadsAppsDirectory);
        }
    }

    @Test
    public void testInstallApps() throws Exception {
+14 −13
Original line number Diff line number Diff line
@@ -38,6 +38,7 @@ import android.app.job.JobScheduler;
import android.content.BroadcastReceiver;
import android.content.ContentResolver;
import android.content.Context;
import android.content.ContextWrapper;
import android.content.Intent;
import android.content.IntentFilter;
import android.content.pm.ActivityInfo;
@@ -59,6 +60,7 @@ import android.provider.MediaStore;
import android.provider.Settings;
import android.support.test.InstrumentationRegistry;
import android.support.test.filters.SmallTest;
import android.support.test.runner.AndroidJUnit4;
import android.test.mock.MockContentProvider;
import android.test.mock.MockContentResolver;
import android.util.ArrayMap;
@@ -72,9 +74,9 @@ import org.junit.After;
import org.junit.Before;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.junit.runners.JUnit4;
import org.mockito.ArgumentCaptor;
import org.mockito.Mock;
import org.mockito.Mockito;
import org.mockito.MockitoAnnotations;
import org.mockito.compat.ArgumentMatcher;

@@ -82,7 +84,7 @@ import java.io.File;
import java.util.concurrent.CountDownLatch;
import java.util.concurrent.TimeUnit;

@RunWith(JUnit4.class)
@RunWith(AndroidJUnit4.class)
@SmallTest
public class RetailDemoModeServiceTest {
    private static final int TEST_DEMO_USER = 111;
@@ -90,7 +92,7 @@ public class RetailDemoModeServiceTest {
    private static final String TEST_CAMERA_PKG = "test.cameraapp";
    private static final String TEST_PRELOADS_DIR_NAME = "test_preloads";

    private @Mock Context mContext;
    private Context mContext;
    private @Mock UserManager mUm;
    private @Mock PackageManager mPm;
    private @Mock IPackageManager mIpm;
@@ -113,12 +115,11 @@ public class RetailDemoModeServiceTest {
    @Before
    public void setUp() throws Exception {
        MockitoAnnotations.initMocks(this);
        Context originalContext = InstrumentationRegistry.getContext();
        when(mContext.getApplicationInfo()).thenReturn(originalContext.getApplicationInfo());
        when(mContext.getResources()).thenReturn(originalContext.getResources());
        mContext = Mockito.spy(new ContextWrapper(InstrumentationRegistry.getTargetContext()));
        when(mContext.getSystemServiceName(eq(JobScheduler.class))).thenReturn(
                Context.JOB_SCHEDULER_SERVICE);
        when(mContext.getSystemService(Context.JOB_SCHEDULER_SERVICE)).thenReturn(mJobScheduler);
        when(mContext.getSystemService(Context.USER_SERVICE)).thenReturn(mUm);
        mContentResolver = new MockContentResolver(mContext);
        mContentResolver.addProvider(Settings.AUTHORITY, new FakeSettingsProvider());
        mContactsProvider = new MockContactsProvider(mContext);
@@ -129,12 +130,10 @@ public class RetailDemoModeServiceTest {
        mTestPreloadsDir = new File(InstrumentationRegistry.getContext().getFilesDir(),
                TEST_PRELOADS_DIR_NAME);

        Settings.Global.putString(mContentResolver,
                Settings.Global.RETAIL_DEMO_MODE_CONSTANTS, "");
        Settings.Global.putInt(mContentResolver,
                Settings.Global.DEVICE_PROVISIONED, 1);
        Settings.Global.putInt(mContentResolver,
                Settings.Global.DEVICE_DEMO_MODE, 1);
        Settings.Global.putString(mContentResolver, Settings.Global.RETAIL_DEMO_MODE_CONSTANTS, "");
        Settings.Global.putInt(mContentResolver, Settings.Global.DEVICE_PROVISIONED, 1);
        Settings.Global.putInt(mContentResolver, Settings.Global.DEVICE_DEMO_MODE, 1);

        // Initialize RetailDemoModeService
        mInjector = new TestInjector();
        mService = new RetailDemoModeService(mInjector);
@@ -143,8 +142,10 @@ public class RetailDemoModeServiceTest {

    @After
    public void tearDown() {
        if (mTestPreloadsDir != null) {
            FileUtils.deleteContentsAndDir(mTestPreloadsDir);
        }
    }

    @Test
    public void testDemoUserSetup() throws Exception {