replicant-vendor_cmsdk/tests/src/org/cyanogenmod/tests/customtiles/unit/CMStatusBarManagerTest.java

/**
 * Copyright (c) 2015, The CyanogenMod Project
 *
 * 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 org.cyanogenmod.tests.customtiles.unit;

import android.app.PendingIntent;
import android.content.ComponentName;
import android.content.Intent;
import android.os.Binder;
import android.os.IBinder;
import android.os.RemoteException;
import android.os.UserHandle;
import android.test.AndroidTestCase;
import android.test.suitebuilder.annotation.MediumTest;
import android.test.suitebuilder.annotation.SmallTest;

import android.text.TextUtils;
import android.util.Log;

import cyanogenmod.app.CMContextConstants;
import cyanogenmod.app.CMStatusBarManager;
import cyanogenmod.app.CustomTile;
import cyanogenmod.app.CustomTileListenerService;
import cyanogenmod.app.ICMStatusBarManager;
import org.cyanogenmod.tests.R;

import java.util.concurrent.CountDownLatch;

public class CMStatusBarManagerTest extends AndroidTestCase {
    private static final String TAG = CMStatusBarManagerTest.class.getSimpleName();
    private static final int COUNTDOWN = 1;
    private CMStatusBarManager mCMStatusBarManager;

    @Override
    protected void setUp() throws Exception {
        super.setUp();
        // Only run this if we support cm status bar service
        org.junit.Assume.assumeTrue(mContext.getPackageManager().hasSystemFeature(
                CMContextConstants.Features.STATUSBAR));
        mCMStatusBarManager = CMStatusBarManager.getInstance(mContext);
    }

    @SmallTest
    public void testManagerExists() {
        assertNotNull(mCMStatusBarManager);
    }

    @SmallTest
    public void testManagerServiceIsAvailable() {
        ICMStatusBarManager icmHardwareManagerService = mCMStatusBarManager.getService();
        assertNotNull(icmHardwareManagerService);
    }

    @MediumTest
    public void testCustomTileListenerServiceRegisterAndUnregisterAsSystemService() {
        CustomTileListenerService customTileListenerService = new CustomTileListenerService();
        registerCustomTileListenerService(customTileListenerService);
        unregisterCustomTileListenerService(customTileListenerService);
    }

    @MediumTest
    public void testCustomTileListenerServiceOnListenerConnected() {
        final CountDownLatch signal = new CountDownLatch(COUNTDOWN);
        CustomTileListenerService customTileListenerService =
                new CustomTileListenerService() {
                    @Override
                    public IBinder onBind(Intent intent) {
                        Log.d(TAG, "Bound");
                        signal.countDown();
                        return super.onBind(intent);
                    }
                    @Override
                    public void onListenerConnected() {
                        Log.d(TAG, "Connected");
                        super.onListenerConnected();
                        signal.countDown();
                    }
                };

        registerCustomTileListenerService(customTileListenerService);

        // Lock
        try {
            signal.await();
        } catch (InterruptedException e) {
            throw new AssertionError(e);
        }

        unregisterCustomTileListenerService(customTileListenerService);
    }

    @MediumTest
    public void testCustomTileListenerServiceOnCustomTilePosted() {
        final CustomTile expectedCustomTile = createSampleCustomTile();

        final CountDownLatch signal = new CountDownLatch(COUNTDOWN);
        CustomTileListenerService customTileListenerService =
                new CustomTileListenerService() {
                    @Override
                    public void onListenerConnected() {
                        super.onListenerConnected();
                        Log.d(TAG, "Connected");
                        // publish
                        mCMStatusBarManager.publishTile(1337, expectedCustomTile);
                    }

                    @Override
                    public void onCustomTilePosted(cyanogenmod.app.StatusBarPanelCustomTile sbc) {
                        super.onCustomTilePosted(sbc);
                        Log.d(TAG, "Posted " + sbc.getCustomTile());
                        if (TextUtils.equals(expectedCustomTile.label, sbc.getCustomTile().label)) {
                            signal.countDown();
                        }
                    }
                };

        registerCustomTileListenerService(customTileListenerService);

        // Lock
        try {
            signal.await();
        } catch (InterruptedException e) {
            throw new AssertionError(e);
        }

        unregisterCustomTileListenerService(customTileListenerService);
    }

    @MediumTest
    public void testCustomTileListenerServiceOnCustomTileRemoved() {
        final CustomTile expectedCustomTile = createSampleCustomTile();

        final CountDownLatch signal = new CountDownLatch(COUNTDOWN);
        CustomTileListenerService customTileListenerService =
                new CustomTileListenerService() {
                    @Override
                    public void onListenerConnected() {
                        super.onListenerConnected();
                        Log.d(TAG, "Connected");
                        // publish
                        mCMStatusBarManager.publishTile(1338, expectedCustomTile);
                    }

                    @Override
                    public void onCustomTilePosted(cyanogenmod.app.StatusBarPanelCustomTile sbc) {
                        super.onCustomTilePosted(sbc);
                        Log.d(TAG, "Posted " + sbc.getCustomTile());
                        if (TextUtils.equals(expectedCustomTile.label, sbc.getCustomTile().label)) {
                            removeCustomTile(mContext.getPackageName(), null, 1338);
                        }
                    }

                    @Override
                    public void onCustomTileRemoved(cyanogenmod.app.StatusBarPanelCustomTile sbc) {
                        super.onCustomTileRemoved(sbc);
                        Log.d(TAG, "Removed " + sbc.getCustomTile());
                        if (TextUtils.equals(expectedCustomTile.label, sbc.getCustomTile().label)) {
                            signal.countDown();
                        }
                    }
                };

        registerCustomTileListenerService(customTileListenerService);

        // Lock
        try {
            signal.await();
        } catch (InterruptedException e) {
            throw new AssertionError(e);
        }

        unregisterCustomTileListenerService(customTileListenerService);
    }

    private CustomTile createSampleCustomTile() {
        Intent intent = new Intent(Intent.ACTION_DIAL);
        PendingIntent pendingIntent = PendingIntent.getActivity(mContext, 0, intent, 0);
        return new CustomTile.Builder(mContext)
                .setLabel("YOLO")
                .setIcon(R.drawable.ic_launcher)
                .setOnClickIntent(pendingIntent)
                .build();
    }

    private void registerCustomTileListenerService(
            CustomTileListenerService customTileListenerService) {
        try {
            Log.d(TAG, "Registering " + customTileListenerService
                    + " custom tile listener service");
            customTileListenerService.registerAsSystemService(mContext,
                    new ComponentName(mContext.getPackageName(),
                            CMStatusBarManagerTest.this.getClass().getCanonicalName()),
                    UserHandle.USER_ALL);
        } catch (RemoteException e) {
            throw new AssertionError(e);
        }
    }

    private void unregisterCustomTileListenerService(CustomTileListenerService customTileListenerService) {
        try {
            Log.d(TAG, "Unregistering " + customTileListenerService
                    + " custom tile listener service");
            customTileListenerService.unregisterAsSystemService();
        } catch (RemoteException e) {
            throw new AssertionError(e);
        }
    }
}