chrome/android/junit/src/org/chromium/chrome/browser/DelayedScreenLockIntentHandlerTest.java

// Copyright 2016 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

package org.chromium.chrome.browser;

import static org.mockito.Mockito.any;
import static org.mockito.Mockito.eq;
import static org.mockito.Mockito.never;
import static org.mockito.Mockito.verify;

import android.content.Context;
import android.content.Intent;
import android.content.IntentFilter;

import org.junit.Before;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.mockito.Mock;
import org.mockito.MockitoAnnotations;
import org.robolectric.annotation.Config;

import org.chromium.base.ContextUtils;
import org.chromium.testing.local.LocalRobolectricTestRunner;

/**
 * Unit tests for the DelayedScreenLockIntentHandlerTest class.
 */
@RunWith(LocalRobolectricTestRunner.class)
@Config(manifest = Config.NONE)
public class DelayedScreenLockIntentHandlerTest {
    @Mock private Context mContextMock;

    private DelayedScreenLockIntentHandler mIntentHandler;

    @Before
    public void setUp() {
        MockitoAnnotations.initMocks(this);
        mIntentHandler = new DelayedScreenLockIntentHandler();
        ContextUtils.initApplicationContextForTests(mContextMock);
    }

    @Test
    public void testReceiveBroadcast() {
        final Intent deferredIntent = new Intent();
        final Intent intent = new Intent(Intent.ACTION_USER_PRESENT);

        mIntentHandler.updateDeferredIntent(deferredIntent);
        mIntentHandler.onReceive(mContextMock, intent);

        verify(mContextMock).registerReceiver(eq(mIntentHandler), any(IntentFilter.class));
        verify(mContextMock).startActivity(deferredIntent);
        verify(mContextMock).unregisterReceiver(mIntentHandler);
    }

    @Test
    public void testReceiveBroadcastTwice() {
        final Intent deferredIntent = new Intent();
        final Intent intent = new Intent(Intent.ACTION_USER_PRESENT);

        mIntentHandler.updateDeferredIntent(deferredIntent);
        mIntentHandler.onReceive(mContextMock, intent);
        mIntentHandler.onReceive(mContextMock, intent);

        verify(mContextMock).registerReceiver(eq(mIntentHandler), any(IntentFilter.class));
        verify(mContextMock).startActivity(deferredIntent);
        verify(mContextMock).unregisterReceiver(mIntentHandler);
    }

    @Test
    public void testSecondDeferredIntentAction() {
        final Intent deferredIntent1 = new Intent();
        final Intent deferredIntent2 = new Intent();
        final Intent intent = new Intent(Intent.ACTION_USER_PRESENT);

        mIntentHandler.updateDeferredIntent(deferredIntent1);
        mIntentHandler.updateDeferredIntent(deferredIntent2);
        mIntentHandler.onReceive(mContextMock, intent);

        verify(mContextMock).registerReceiver(eq(mIntentHandler), any(IntentFilter.class));
        verify(mContextMock).startActivity(deferredIntent2);
        verify(mContextMock).unregisterReceiver(mIntentHandler);
    }

    @Test
    public void testNonExpectedIntentAction() {
        mIntentHandler.updateDeferredIntent(new Intent());
        try {
            mIntentHandler.onReceive(mContextMock, new Intent());
        } catch (AssertionError assertError) {
            // Ignore AssertErrors
        }

        verify(mContextMock).registerReceiver(eq(mIntentHandler), any(IntentFilter.class));
        verify(mContextMock, never()).startActivity(any(Intent.class));
        verify(mContextMock, never()).unregisterReceiver(mIntentHandler);
    }
}