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

Commit 8c8b67fe authored by Tim Peng's avatar Tim Peng
Browse files

Fix SettingsLib test cases 

-Update test cases according to ag/14909662

Bug: 194743689
Test: make -j50 RunSettingsLibRoboTests ROBOTEST_FILTER=InfoMediaManagerTest
Change-Id: Iebdf4b7610bf065bfe45442956bc6ea1197308fb
parent d309a0f0

packages/SettingsLib/tests/robotests/src/com/android/settingslib/media/InfoMediaManagerTest.java

+7 −11
Original line number Diff line number Diff line
@@ -27,7 +27,6 @@ import static android.media.MediaRoute2ProviderService.REASON_UNKNOWN_ERROR; 
import static com.google.common.truth.Truth.assertThat;



import static org.mockito.ArgumentMatchers.any;

import static org.mockito.ArgumentMatchers.anyString;

import static org.mockito.Mockito.doReturn;

import static org.mockito.Mockito.mock;

import static org.mockito.Mockito.spy;
@@ -108,7 +107,7 @@ public class InfoMediaManagerTest { 


        final List<MediaRoute2Info> routes = new ArrayList<>();

        routes.add(info);

        mShadowRouter2Manager.setAvailableRoutes(routes);

        mShadowRouter2Manager.setTransferableRoutes(routes);



        final MediaDevice mediaDevice = mInfoMediaManager.findMediaDevice(TEST_ID);

        assertThat(mediaDevice).isNull();
@@ -159,7 +158,7 @@ public class InfoMediaManagerTest { 


        final List<MediaRoute2Info> routes = new ArrayList<>();

        routes.add(info);

        mShadowRouter2Manager.setAvailableRoutes(routes);

        mShadowRouter2Manager.setTransferableRoutes(routes);



        final MediaDevice mediaDevice = mInfoMediaManager.findMediaDevice(TEST_ID);

        assertThat(mediaDevice).isNull();
@@ -195,7 +194,7 @@ public class InfoMediaManagerTest { 


        final List<MediaRoute2Info> routes = new ArrayList<>();

        routes.add(info);

        mShadowRouter2Manager.setAvailableRoutes(routes);

        mShadowRouter2Manager.setTransferableRoutes(routes);



        final MediaDevice mediaDevice = mInfoMediaManager.findMediaDevice(TEST_ID);

        assertThat(mediaDevice).isNull();
@@ -618,7 +617,7 @@ public class InfoMediaManagerTest { 


        final List<MediaRoute2Info> routes = new ArrayList<>();

        routes.add(info);

        mShadowRouter2Manager.setAvailableRoutes(routes);

        mShadowRouter2Manager.setTransferableRoutes(routes);



        final MediaDevice mediaDevice = mInfoMediaManager.findMediaDevice(TEST_ID);

        assertThat(mediaDevice).isNull();
@@ -728,9 +727,8 @@ public class InfoMediaManagerTest { 
        final MediaRoute2Info info = mock(MediaRoute2Info.class);

        final List<MediaRoute2Info> infos = new ArrayList<>();

        infos.add(info);

        mShadowRouter2Manager.setAvailableRoutes(infos);

        mShadowRouter2Manager.setTransferableRoutes(infos);



        when(mRouterManager.getAvailableRoutes(anyString())).thenReturn(infos);

        when(info.getType()).thenReturn(TYPE_REMOTE_SPEAKER);



        assertThat(mInfoMediaManager.shouldDisableMediaOutput("test")).isTrue();
@@ -741,9 +739,8 @@ public class InfoMediaManagerTest { 
        final MediaRoute2Info info = mock(MediaRoute2Info.class);

        final List<MediaRoute2Info> infos = new ArrayList<>();

        infos.add(info);

        mShadowRouter2Manager.setAvailableRoutes(infos);

        mShadowRouter2Manager.setTransferableRoutes(infos);



        when(mRouterManager.getAvailableRoutes(anyString())).thenReturn(infos);

        when(info.getType()).thenReturn(TYPE_BUILTIN_SPEAKER);



        assertThat(mInfoMediaManager.shouldDisableMediaOutput("test")).isFalse();
@@ -757,9 +754,8 @@ public class InfoMediaManagerTest { 
        final List<MediaRoute2Info> infos = new ArrayList<>();

        infos.add(info);

        infos.add(info2);

        mShadowRouter2Manager.setAvailableRoutes(infos);

        mShadowRouter2Manager.setTransferableRoutes(infos);



        when(mRouterManager.getAvailableRoutes(anyString())).thenReturn(infos);

        when(info.getType()).thenReturn(TYPE_REMOTE_SPEAKER);

        when(info2.getType()).thenReturn(TYPE_REMOTE_SPEAKER);

packages/SettingsLib/tests/robotests/testutils/com/android/settingslib/testutils/shadow/ShadowRouter2Manager.java

+9 −0
Original line number Diff line number Diff line
@@ -85,4 +85,13 @@ public class ShadowRouter2Manager { 
        return (ShadowRouter2Manager) Shadow.extract(

                MediaRouter2Manager.getInstance(RuntimeEnvironment.application));

    }



    @Implementation

    protected List<MediaRoute2Info> getTransferableRoutes(String packageName) {

        return mAvailableRoutes;

    }



    public void setTransferableRoutes(List<MediaRoute2Info> infos) {

        mAvailableRoutes = infos;

    }

}