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

Commit bbd54b61 authored by Amin Shaikh's avatar Amin Shaikh
Browse files

Fix ThemePicker robo test build. 

Test: make
Change-Id: If66b31d4ee5f21430864946c5adf1bf623b1b2b2
parent 8204b164

robolectric_tests/src/com/android/customization/model/grid/GridOptionsManagerTest.java

+3 −1
Original line number Diff line number Diff line
@@ -24,6 +24,7 @@ import static org.mockito.Mockito.when; 
import androidx.annotation.Nullable;



import com.android.customization.model.CustomizationManager.Callback;

import com.android.customization.module.ThemesUserEventLogger;



import org.junit.Before;

import org.junit.Test;
@@ -37,12 +38,13 @@ import org.robolectric.RobolectricTestRunner; 
public class GridOptionsManagerTest {



    @Mock LauncherGridOptionsProvider mProvider;

    @Mock ThemesUserEventLogger mThemesUserEventLogger;

    private GridOptionsManager mManager;



    @Before

    public void setUp() {

        MockitoAnnotations.initMocks(this);

        mManager = new GridOptionsManager(mProvider);

        mManager = new GridOptionsManager(mProvider, mThemesUserEventLogger);

    }



    @Test

robolectric_tests/src/com/android/customization/model/theme/ThemeManagerTest.java

+6 −3
Original line number Diff line number Diff line
@@ -44,6 +44,7 @@ import androidx.annotation.Nullable; 
import androidx.fragment.app.FragmentActivity;



import com.android.customization.model.CustomizationManager.Callback;

import com.android.customization.module.ThemesUserEventLogger;

import com.android.customization.testutils.Condition;

import com.android.customization.testutils.OverlayManagerMocks;

import com.android.customization.testutils.Wait;
@@ -65,6 +66,7 @@ public class ThemeManagerTest { 


    @Mock OverlayManagerCompat mMockOm;

    @Mock WallpaperSetter mMockWallpaperSetter;

    @Mock ThemesUserEventLogger mThemesUserEventLogger;

    private OverlayManagerMocks mMockOmHelper;

    private ThemeManager mThemeManager;

    private FragmentActivity mActivity;
@@ -77,7 +79,8 @@ public class ThemeManagerTest { 
        mMockOmHelper = new OverlayManagerMocks();

        mMockOmHelper.setUpMock(mMockOm);

        ThemeBundleProvider provider = mock(ThemeBundleProvider.class);

        mThemeManager = new ThemeManager(provider, activity, mMockWallpaperSetter, mMockOm);

        mThemeManager = new ThemeManager(

                provider, activity, mMockWallpaperSetter, mMockOm, mThemesUserEventLogger);

    }



    @After
@@ -100,7 +103,7 @@ public class ThemeManagerTest { 
        mMockOmHelper.addOverlay("test.package.name_sysui", SYSUI_PACKAGE,

                OVERLAY_CATEGORY_ICON_SYSUI, true, 0);



        ThemeBundle defaultTheme = new ThemeBundle.Builder().asDefault().build();

        ThemeBundle defaultTheme = new ThemeBundle.Builder().asDefault().build(mActivity);



        applyThemeAndWaitForCondition(defaultTheme, "Overlays didn't get disabled", () -> {

            verify(mMockOm, times(6)).disableOverlay(anyString(), anyInt());
@@ -128,7 +131,7 @@ public class ThemeManagerTest { 
        ThemeBundle theme = new ThemeBundle.Builder()

                .addOverlayPackage(OVERLAY_CATEGORY_COLOR, bundleColorPackage)

                .addOverlayPackage(OVERLAY_CATEGORY_FONT, bundleFontPackage)

                .build();

                .build(mActivity);



        applyThemeAndWaitForCondition(theme, "Overlays didn't get enabled", () -> {

            verify(mMockOm, times(2)).setEnabledExclusiveInCategory(anyString(), anyInt());