GooglePlayMessageHandlerTest.java

// Copyright 2016 Google, Inc.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//      http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
//
////////////////////////////////////////////////////////////////////////////////

package com.firebase.jobdispatcher;

import static com.firebase.jobdispatcher.GooglePlayJobWriter.REQUEST_PARAM_TAG;
import static org.mockito.Matchers.any;
import static org.mockito.Matchers.eq;
import static org.mockito.Mockito.doThrow;
import static org.mockito.Mockito.never;
import static org.mockito.Mockito.verify;
import static org.mockito.Mockito.when;

import android.app.AppOpsManager;
import android.content.Context;
import android.os.Bundle;
import android.os.Looper;
import android.os.Message;
import android.os.Messenger;
import com.firebase.jobdispatcher.JobInvocation.Builder;
import org.junit.Before;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.mockito.Mock;
import org.mockito.MockitoAnnotations;
import org.robolectric.RobolectricTestRunner;
import org.robolectric.annotation.Config;

/**
 * Tests {@link GooglePlayMessageHandler}.
 */
@RunWith(RobolectricTestRunner.class)
@Config(constants = BuildConfig.class, manifest = Config.NONE, sdk = 21)
public class GooglePlayMessageHandlerTest {

    @Mock
    Looper looper;
    @Mock
    GooglePlayReceiver receiverMock;
    @Mock
    Context context;
    @Mock
    AppOpsManager appOpsManager;
    @Mock
    Messenger messengerMock;
    @Mock
    ExecutionDelegator executionDelegatorMock;

    GooglePlayMessageHandler handler;

    @Before
    public void setUp() throws Exception {
        MockitoAnnotations.initMocks(this);
        handler = new GooglePlayMessageHandler(looper, receiverMock);
        when(receiverMock.getExecutionDelegator()).thenReturn(executionDelegatorMock);
        when(receiverMock.getApplicationContext()).thenReturn(context);
        when(context.getSystemService(Context.APP_OPS_SERVICE)).thenReturn(appOpsManager);
    }

    @Test
    public void handleMessage_nullNoException() throws Exception {
        handler.handleMessage(null);
    }

    @Test
    public void handleMessage_ignoreIfSenderIsNotGcm() throws Exception {
        Message message = Message.obtain();
        message.what = GooglePlayMessageHandler.MSG_START_EXEC;
        Bundle data = new Bundle();
        data.putString(REQUEST_PARAM_TAG, "TAG");
        message.setData(data);
        message.replyTo = messengerMock;
        doThrow(new SecurityException()).when(appOpsManager)
                .checkPackage(message.sendingUid, GooglePlayDriver.BACKEND_PACKAGE);
        handler.handleMessage(message);
        verify(receiverMock, never()).prepareJob(eq(data), any(GooglePlayMessengerCallback.class));
    }

    @Test
    public void handleMessage_startExecution_noData() throws Exception {
        Message message = Message.obtain();
        message.what = GooglePlayMessageHandler.MSG_START_EXEC;
        message.replyTo = messengerMock;

        handler.handleMessage(message);
        verify(receiverMock, never())
                .prepareJob(any(Bundle.class), any(GooglePlayMessengerCallback.class));
    }

    @Test
    public void handleMessage_startExecution() throws Exception {
        Message message = Message.obtain();
        message.what = GooglePlayMessageHandler.MSG_START_EXEC;
        Bundle data = new Bundle();
        data.putString(REQUEST_PARAM_TAG, "TAG");
        message.setData(data);
        message.replyTo = messengerMock;
        JobInvocation jobInvocation = new Builder()
                .setTag("tag")
                .setService(TestJobService.class.getName())
                .setTrigger(Trigger.NOW).build();
        when(receiverMock.prepareJob(eq(data), any(GooglePlayMessengerCallback.class)))
                .thenReturn(jobInvocation);

        handler.handleMessage(message);

        verify(executionDelegatorMock).executeJob(jobInvocation);
    }

    @Test
    public void handleMessage_stopExecution_invalidNoCrash() throws Exception {
        Message message = Message.obtain();
        message.what = GooglePlayMessageHandler.MSG_STOP_EXEC;
        message.replyTo = messengerMock;

        handler.handleMessage(message);

        verify(executionDelegatorMock, never()).stopJob(any(JobInvocation.class));
    }
}