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

Commit 86fb5b41 authored by Fabian Kozynski's avatar Fabian Kozynski
Browse files

Unit test for CastControllerImp 

Added basic Unit test for CastControllerImp testing adding and removing
callbacks and that the system recognizes that there were changes.

Change-Id: I992935bdc2a822d00ac66700c6a0da133036fcc1
Test: systemui test passes
parent 3c1a3a41

packages/SystemUI/tests/src/com/android/systemui/statusbar/policy/CastControllerImplTest.java

0 → 100644
+77 −0
Original line number Diff line number Diff line 
package com.android.systemui.statusbar.policy;





import static org.mockito.Mockito.mock;

import static org.mockito.Mockito.never;

import static org.mockito.Mockito.times;

import static org.mockito.Mockito.verify;

import static org.mockito.Mockito.when;



import android.content.Context;

import android.media.MediaRouter;

import android.media.projection.MediaProjectionInfo;

import android.media.projection.MediaProjectionManager;

import android.support.test.filters.SmallTest;

import android.testing.AndroidTestingRunner;

import android.testing.TestableLooper;



import com.android.systemui.SysuiTestCase;

import com.android.systemui.statusbar.policy.CastController.Callback;



import org.junit.Before;

import org.junit.runner.RunWith;

import org.mockito.Mock;

import org.mockito.MockitoAnnotations;

import org.junit.Test;



@SmallTest

@RunWith(AndroidTestingRunner.class)

@TestableLooper.RunWithLooper

public class CastControllerImplTest extends SysuiTestCase {



    @Mock

    MediaRouter mMediaRouter;

    @Mock

    MediaProjectionManager mMediaProjectionManager;

    @Mock

    MediaProjectionInfo mProjection;



    private CastControllerImpl mController;



    @Before

    public void setUp() {

        MockitoAnnotations.initMocks(this);

        mContext.addMockSystemService(MediaRouter.class, mMediaRouter);

        mContext.addMockSystemService(MediaProjectionManager.class, mMediaProjectionManager);

        when(mMediaProjectionManager.getActiveProjectionInfo()).thenReturn(mProjection);



        mController = new CastControllerImpl(mContext);

    }



    @Test

    public void testAddCallback(){

        Callback mockCallback = mock(Callback.class);



        mController.addCallback(mockCallback);

        verify(mockCallback,times(1)).onCastDevicesChanged();

    }



    @Test

    public void testRemoveCallback(){

        Callback mockCallback = mock(Callback.class);



        mController.addCallback(mockCallback);

        verify(mockCallback, times(1)).onCastDevicesChanged();



        mController.removeCallback(mockCallback);

        verify(mockCallback, times(1)).onCastDevicesChanged();

    }



    @Test

    public void testRemoveCallbackFromEmptyList(){

        Callback mockCallback = mock(Callback.class);



        mController.removeCallback(mockCallback);

        verify(mockCallback, never()).onCastDevicesChanged();

    }

}