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

Commit 9e92c766 authored by Marvin Ramin's avatar Marvin Ramin
Browse files

Fix and improve ArcInitiationActionFromAvrTest 

Update test to be more realistic. Un-Ignore a test. Add more tests to
cover more cases.

Bug: 120845532
Test: atest ArcInitiationActionFromAvrTest
Change-Id: I049221e8a2bb5014f30aa3605eca3c225487a576
parent 6d411c83

services/tests/servicestests/src/com/android/server/hdmi/ArcInitiationActionFromAvrTest.java

+120 −67
Original line number Diff line number Diff line
@@ -19,11 +19,18 @@ import static com.android.server.hdmi.HdmiControlService.INITIATED_BY_ENABLE_CEC 


import static com.google.common.truth.Truth.assertThat;



import android.annotation.Nullable;

import android.app.Instrumentation;

import android.hardware.hdmi.HdmiDeviceInfo;

import static org.mockito.Mockito.spy;

import static org.mockito.Mockito.when;



import android.content.Context;

import android.content.ContextWrapper;

import android.hardware.tv.cec.V1_0.SendMessageResult;

import android.media.AudioManager;

import android.os.Handler;

import android.os.IPowerManager;

import android.os.IThermalService;

import android.os.Looper;

import android.os.PowerManager;

import android.os.test.TestLooper;

import android.platform.test.annotations.Presubmit;
@@ -31,10 +38,11 @@ import androidx.test.InstrumentationRegistry; 
import androidx.test.filters.SmallTest;



import org.junit.Before;

import org.junit.Ignore;

import org.junit.Test;

import org.junit.runner.RunWith;

import org.junit.runners.JUnit4;

import org.mockito.Mock;

import org.mockito.MockitoAnnotations;



import java.util.ArrayList;
@@ -44,7 +52,7 @@ import java.util.ArrayList; 
@RunWith(JUnit4.class)

public class ArcInitiationActionFromAvrTest {



    private HdmiDeviceInfo mDeviceInfoForTests;

    private Context mContextSpy;

    private HdmiCecLocalDeviceAudioSystem mHdmiCecLocalDeviceAudioSystem;

    private HdmiCecController mHdmiCecController;

    private HdmiControlService mHdmiControlService;
@@ -52,56 +60,43 @@ public class ArcInitiationActionFromAvrTest { 
    private ArcInitiationActionFromAvr mAction;



    private TestLooper mTestLooper = new TestLooper();

    private boolean mSendCecCommandSuccess;

    private boolean mShouldDispatchARCInitiated;

    private boolean mArcInitSent;

    private boolean mRequestActiveSourceSent;

    private Instrumentation mInstrumentation;

    private ArrayList<HdmiCecLocalDevice> mLocalDevices = new ArrayList<>();



    @Mock private IPowerManager mIPowerManagerMock;

    @Mock private IThermalService mIThermalServiceMock;

    @Mock private AudioManager mAudioManager;



    @Before

    public void setUp() {

        mDeviceInfoForTests = new HdmiDeviceInfo(1000, 1);

    public void setUp() throws Exception {

        MockitoAnnotations.initMocks(this);



        mContextSpy = spy(new ContextWrapper(InstrumentationRegistry.getTargetContext()));



        mInstrumentation = InstrumentationRegistry.getInstrumentation();

        PowerManager powerManager = new PowerManager(mContextSpy, mIPowerManagerMock,

                mIThermalServiceMock, new Handler(mTestLooper.getLooper()));

        when(mContextSpy.getSystemService(Context.POWER_SERVICE)).thenReturn(powerManager);

        when(mContextSpy.getSystemService(PowerManager.class)).thenReturn(powerManager);

        when(mIPowerManagerMock.isInteractive()).thenReturn(true);



        mHdmiControlService =

                new HdmiControlService(mInstrumentation.getTargetContext()) {

                new HdmiControlService(mContextSpy) {

                    @Override

                    void sendCecCommand(

                            HdmiCecMessage command, @Nullable SendMessageCallback callback) {

                        switch (command.getOpcode()) {

                            case Constants.MESSAGE_REQUEST_ACTIVE_SOURCE:

                                if (callback != null) {

                                    callback.onSendCompleted(

                                            mSendCecCommandSuccess

                                                    ? SendMessageResult.SUCCESS

                                                    : SendMessageResult.NACK);

                                }

                                mRequestActiveSourceSent = true;

                                break;

                            case Constants.MESSAGE_INITIATE_ARC:

                                if (callback != null) {

                                    callback.onSendCompleted(

                                            mSendCecCommandSuccess

                                                    ? SendMessageResult.SUCCESS

                                                    : SendMessageResult.NACK);

                                }

                                mArcInitSent = true;

                                if (mShouldDispatchARCInitiated) {

                                    mHdmiCecLocalDeviceAudioSystem.dispatchMessage(

                                            HdmiCecMessageBuilder.buildReportArcInitiated(

                                                    Constants.ADDR_TV,

                                                    Constants.ADDR_AUDIO_SYSTEM));

                    boolean isPowerStandby() {

                        return false;

                    }

                                break;

                            default:



                    @Override

                    void wakeUp() {

                    }



                    @Override

                    PowerManager getPowerManager() {

                        return powerManager;

                    }



                    @Override

                    boolean isPowerStandby() {

                        return false;

                    AudioManager getAudioManager() {

                        return mAudioManager;

                    }



                    @Override
@@ -115,21 +110,9 @@ public class ArcInitiationActionFromAvrTest { 
                    }

                };



        mHdmiCecLocalDeviceAudioSystem =

                new HdmiCecLocalDeviceAudioSystem(mHdmiControlService) {

                    @Override

                    HdmiDeviceInfo getDeviceInfo() {

                        return mDeviceInfoForTests;

                    }



        mHdmiCecLocalDeviceAudioSystem = new HdmiCecLocalDeviceAudioSystem(mHdmiControlService) {

            @Override

                    void setArcStatus(boolean enabled) {

                        // do nothing

                    }



                    @Override

                    protected boolean isSystemAudioActivated() {

                        return true;

            protected void setPreferredAddress(int addr) {

            }

        };
@@ -150,18 +133,88 @@ public class ArcInitiationActionFromAvrTest { 
        mTestLooper.dispatchAll();

    }



    @Ignore("b/120845532")

    @Test

    public void arcInitiation_requestActiveSource() {

        mSendCecCommandSuccess = true;

        mShouldDispatchARCInitiated = true;

        mRequestActiveSourceSent = false;

        mArcInitSent = false;

    public void arcInitiation_initiated() {

        mHdmiCecLocalDeviceAudioSystem.addAndStartAction(mAction);

        mTestLooper.dispatchAll();

        HdmiCecMessage initiateArc = HdmiCecMessageBuilder.buildInitiateArc(

                Constants.ADDR_AUDIO_SYSTEM, Constants.ADDR_TV);



        assertThat(mNativeWrapper.getResultMessages()).contains(initiateArc);



        mHdmiControlService.sendCecCommand(

                HdmiCecMessageBuilder.buildReportArcInitiated(

                        Constants.ADDR_TV,

                        Constants.ADDR_AUDIO_SYSTEM));

        mTestLooper.dispatchAll();



        assertThat(mHdmiCecLocalDeviceAudioSystem.isArcEnabled()).isTrue();

    }



    @Test

    public void arcInitiation_sendFailed() {

        mNativeWrapper.setMessageSendResult(Constants.MESSAGE_INITIATE_ARC, SendMessageResult.NACK);

        mHdmiCecLocalDeviceAudioSystem.addAndStartAction(mAction);

        mTestLooper.dispatchAll();

        HdmiCecMessage initiateArc = HdmiCecMessageBuilder.buildInitiateArc(

                Constants.ADDR_AUDIO_SYSTEM, Constants.ADDR_TV);



        assertThat(mArcInitSent).isTrue();

        assertThat(mRequestActiveSourceSent).isTrue();

        assertThat(mNativeWrapper.getResultMessages()).contains(initiateArc);



        assertThat(mHdmiCecLocalDeviceAudioSystem.isArcEnabled()).isFalse();

    }



    @Test

    public void arcInitiation_terminated() {

        mHdmiCecLocalDeviceAudioSystem.addAndStartAction(mAction);

        mTestLooper.dispatchAll();



        HdmiCecMessage initiateArc = HdmiCecMessageBuilder.buildInitiateArc(

                Constants.ADDR_AUDIO_SYSTEM, Constants.ADDR_TV);



        assertThat(mNativeWrapper.getResultMessages()).contains(initiateArc);



        mHdmiControlService.handleCecCommand(HdmiCecMessageBuilder.buildReportArcTerminated(

                Constants.ADDR_TV,

                Constants.ADDR_AUDIO_SYSTEM));

        mTestLooper.dispatchAll();



        assertThat(mHdmiCecLocalDeviceAudioSystem.isArcEnabled()).isFalse();

    }



    @Test

    public void arcInitiation_abort() {

        mHdmiCecLocalDeviceAudioSystem.addAndStartAction(mAction);

        mTestLooper.dispatchAll();



        HdmiCecMessage initiateArc = HdmiCecMessageBuilder.buildInitiateArc(

                Constants.ADDR_AUDIO_SYSTEM, Constants.ADDR_TV);



        assertThat(mNativeWrapper.getResultMessages()).contains(initiateArc);



        mHdmiControlService.handleCecCommand(

                HdmiCecMessageBuilder.buildFeatureAbortCommand(

                        Constants.ADDR_TV,

                        Constants.ADDR_AUDIO_SYSTEM, Constants.MESSAGE_INITIATE_ARC,

                        Constants.ABORT_REFUSED));

        mTestLooper.dispatchAll();



        assertThat(mHdmiCecLocalDeviceAudioSystem.isArcEnabled()).isFalse();

    }



    //Fail

    @Test

    public void arcInitiation_timeout() {

        mHdmiCecLocalDeviceAudioSystem.addAndStartAction(mAction);

        mTestLooper.dispatchAll();



        HdmiCecMessage initiateArc = HdmiCecMessageBuilder.buildInitiateArc(

                Constants.ADDR_AUDIO_SYSTEM, Constants.ADDR_TV);



        assertThat(mNativeWrapper.getResultMessages()).contains(initiateArc);



        mTestLooper.moveTimeForward(1001);

        mTestLooper.dispatchAll();

        assertThat(mHdmiCecLocalDeviceAudioSystem.isArcEnabled()).isTrue();

    }

}

services/tests/servicestests/src/com/android/server/hdmi/FakeNativeWrapper.java

+9 −2
Original line number Diff line number Diff line
@@ -24,6 +24,7 @@ import com.android.server.hdmi.HdmiCecController.NativeWrapper; 
import com.google.common.collect.Iterables;



import java.util.ArrayList;

import java.util.HashMap;

import java.util.List;



/** Fake {@link NativeWrapper} useful for testing. */
@@ -48,6 +49,7 @@ final class FakeNativeWrapper implements NativeWrapper { 
            };



    private final List<HdmiCecMessage> mResultMessages = new ArrayList<>();

    private final HashMap<Integer, Integer> mMessageSendResult = new HashMap<>();

    private int mMyPhysicalAddress = 0;

    private HdmiPortInfo[] mHdmiPortInfo = null;
@@ -65,9 +67,10 @@ final class FakeNativeWrapper implements NativeWrapper { 
        if (body.length == 0) {

            return mPollAddressResponse[dstAddress];

        } else {

            mResultMessages.add(HdmiCecMessageBuilder.of(srcAddress, dstAddress, body));

            HdmiCecMessage message = HdmiCecMessageBuilder.of(srcAddress, dstAddress, body);

            mResultMessages.add(message);

            return mMessageSendResult.getOrDefault(message.getOpcode(), SendMessageResult.SUCCESS);

        }

        return SendMessageResult.SUCCESS;

    }



    @Override
@@ -132,6 +135,10 @@ final class FakeNativeWrapper implements NativeWrapper { 
        mPollAddressResponse[logicalAddress] = response;

    }



    public void setMessageSendResult(int opcode, int result) {

        mMessageSendResult.put(opcode, result);

    }



    @VisibleForTesting

    protected void setPhysicalAddress(int physicalAddress) {

        mMyPhysicalAddress = physicalAddress;