com.liferay.mobile.push.DeviceRegistrationTest.java Source code

Introduction

Here is the source code for com.liferay.mobile.push.DeviceRegistrationTest.java

Source

/**
 * Copyright (c) 2000-present Liferay, Inc. All rights reserved.
 *
 * This library is free software; you can redistribute it and/or modify it under
 * the terms of the GNU Lesser General Public License as published by the Free
 * Software Foundation; either version 2.1 of the License, or (at your option)
 * any later version.
 *
 * This library is distributed in the hope that it will be useful, but WITHOUT
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
 * FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public License for more
 * details.
 */

package com.liferay.mobile.push;

import android.content.Context;

import com.liferay.mobile.push.util.GoogleServices;

import junit.framework.Assert;

import org.json.JSONException;
import org.json.JSONObject;

import org.junit.Test;
import org.junit.runner.RunWith;

import org.mockito.Mockito;

import org.robolectric.Robolectric;
import org.robolectric.RobolectricTestRunner;
import org.robolectric.annotation.Config;

/**
 * @author Bruno Farache
 */
@RunWith(RobolectricTestRunner.class)
@Config(manifest = "src/main/AndroidManifest.xml", emulateSdk = 18)
public class DeviceRegistrationTest extends BaseTest {

    @Test
    public void registerWithRegistrationId() throws Exception {
        final String registrationId = "123";

        push.onSuccess(new Push.OnSuccess() {

            @Override
            public void onSuccess(JSONObject device) {
                try {
                    Assert.assertNotNull(device);
                    Assert.assertEquals(Push.ANDROID, device.getString("platform"));
                    Assert.assertEquals(registrationId, device.getString("token"));
                } catch (JSONException je) {
                    Assert.fail();
                }
            }

        }).onFailure(new Push.OnFailure() {

            @Override
            public void onFailure(Exception e) {
                Assert.fail(e.getMessage());
            }

        }).register(registrationId);

        Robolectric.runBackgroundTasks();
    }

    @Test
    public void registerWithSenderId() throws Exception {
        GoogleServices googleServices = Mockito.mock(GoogleServices.class);
        Context context = Robolectric.application;
        String senderId = "senderId";

        final String registrationId = "123";

        Mockito.when(googleServices.getRegistrationId(context, senderId)).thenReturn(registrationId);

        push.setGoogleServices(googleServices);

        push.onSuccess(new Push.OnSuccess() {

            @Override
            public void onSuccess(JSONObject device) {
                try {
                    Assert.assertNotNull(device);
                    Assert.assertEquals(Push.ANDROID, device.getString("platform"));
                    Assert.assertEquals(registrationId, device.getString("token"));
                } catch (JSONException je) {
                    Assert.fail();
                }
            }

        }).register(context, senderId);

        Robolectric.runBackgroundTasks();
    }

    @Test
    public void unregister() throws Exception {
        final String registrationId = "123";

        register(registrationId);

        push.onSuccess(new Push.OnSuccess() {

            @Override
            public void onSuccess(JSONObject device) {
                try {
                    Assert.assertNotNull(device);
                    Assert.assertEquals(registrationId, device.getString("token"));
                } catch (JSONException je) {
                    Assert.fail();
                }
            }

        }).unregister(registrationId);

        Robolectric.runBackgroundTasks();
    }

    protected void register(String registrationId) throws Exception {
        push.onFailure(new Push.OnFailure() {

            @Override
            public void onFailure(Exception e) {
                Assert.fail(e.getMessage());
            }

        }).register(registrationId);

        Robolectric.runBackgroundTasks();
    }

}