v1/src/test/java/com/xtremelabs/robolectric/res/RobolectricPackageManagerTest.java

package com.xtremelabs.robolectric.res;

import static org.hamcrest.CoreMatchers.*;
import static org.junit.Assert.assertThat;

import com.xtremelabs.robolectric.Robolectric;
import com.xtremelabs.robolectric.WithTestDefaultsRunner;
import org.junit.After;
import org.junit.Before;
import org.junit.Test;
import org.junit.runner.RunWith;

import android.content.ComponentName;
import android.content.Intent;
import android.content.pm.ApplicationInfo;
import android.content.pm.PackageInfo;
import android.content.pm.PackageManager;
import android.content.pm.ResolveInfo;
import android.graphics.drawable.BitmapDrawable;
import android.graphics.drawable.Drawable;

import java.util.List;

@RunWith(WithTestDefaultsRunner.class)
public class RobolectricPackageManagerTest {

    private static final String TEST_PACKAGE_NAME = "com.some.other.package";
    private static final String TEST_PACKAGE_LABEL = "My Little App";

    RobolectricPackageManager rpm;

    @Before
    public void setUp() throws Exception {
        rpm = (RobolectricPackageManager) Robolectric.application.getPackageManager();
    }

    @After
    public void tearDown() throws Exception {
    }

    @Test
    public void getApplicationInfo__ThisApplication() throws Exception {
        ApplicationInfo info = rpm.getApplicationInfo(Robolectric.application.getPackageName(), 0);
        assertThat(info, notNullValue());
        assertThat(info.packageName, equalTo(Robolectric.application.getPackageName()));
    }

    @Test
    public void getApplicationInfo__OtherApplication() throws Exception {
        PackageInfo packageInfo = new PackageInfo();
        packageInfo.packageName = TEST_PACKAGE_NAME;
        packageInfo.applicationInfo = new ApplicationInfo();
        packageInfo.applicationInfo.packageName = TEST_PACKAGE_NAME;
        packageInfo.applicationInfo.name = TEST_PACKAGE_LABEL;
        rpm.addPackage(packageInfo);

        ApplicationInfo info = rpm.getApplicationInfo(TEST_PACKAGE_NAME, 0);
        assertThat(info, notNullValue());
        assertThat(info.packageName, equalTo(TEST_PACKAGE_NAME));
        assertThat(rpm.getApplicationLabel(info).toString(), equalTo(TEST_PACKAGE_LABEL));
    }

    @Test
    public void queryIntentActivities__EmptyResult() throws Exception {
        Intent i = new Intent(Intent.ACTION_MAIN, null);
        i.addCategory(Intent.CATEGORY_LAUNCHER);

        List<ResolveInfo> activities = rpm.queryIntentActivities(i, 0);
        assertThat(activities, notNullValue());        // empty list, not null
        assertThat(activities.size(), equalTo(0));
    }

    @Test
    public void queryIntentActivities__Match() throws Exception {
        Intent i = new Intent(Intent.ACTION_MAIN, null);
        i.addCategory(Intent.CATEGORY_LAUNCHER);

        ResolveInfo info = new ResolveInfo();
        info.nonLocalizedLabel = TEST_PACKAGE_LABEL;

        rpm.addResolveInfoForIntent(i, info);

        List<ResolveInfo> activities = rpm.queryIntentActivities(i, 0);
        assertThat(activities, notNullValue());
        assertThat(activities.size(), equalTo(1));
        assertThat(activities.get(0).nonLocalizedLabel.toString(), equalTo(TEST_PACKAGE_LABEL));
    }

    @Test
    public void queryBroadcastReceivers__EmptyResult() throws Exception {
        Intent i = new Intent(Intent.ACTION_MAIN, null);
        i.addCategory(Intent.CATEGORY_LAUNCHER);

        List<ResolveInfo> broadCastReceivers = rpm.queryBroadcastReceivers(i, 0);
        assertThat(broadCastReceivers.size(), equalTo(0));
    }

    @Test
    public void queryBroadcastReceivers__Match() throws Exception {
        Intent i = new Intent(Intent.ACTION_MAIN, null);

        ResolveInfo info = new ResolveInfo();
        info.nonLocalizedLabel = TEST_PACKAGE_LABEL;

        rpm.addResolveInfoForIntent(i, info);

        List<ResolveInfo> broadCastReceivers = rpm.queryBroadcastReceivers(i, 0);
        assertThat(broadCastReceivers.size(), equalTo(1));
        assertThat(broadCastReceivers.get(0).nonLocalizedLabel.toString(),
                equalTo(TEST_PACKAGE_LABEL));
    }

    @Test
    public void resolveActivity__Match() throws Exception {
        Intent i = new Intent(Intent.ACTION_MAIN, null).addCategory(Intent.CATEGORY_LAUNCHER);
        ResolveInfo info = new ResolveInfo();
        info.nonLocalizedLabel = TEST_PACKAGE_LABEL;
        rpm.addResolveInfoForIntent(i, info);

        assertThat(rpm.resolveActivity(i, 0), sameInstance(info));
    }

    @Test
    public void resolveActivity__NoMatch() throws Exception {
        Intent i = new Intent();
        i.setComponent(new ComponentName("foo.bar", "No Activity"));
        assertThat(rpm.resolveActivity(i, 0), nullValue());
    }

    @Test
    public void resolveService__Match() throws Exception {
        Intent i = new Intent(Intent.ACTION_MAIN, null);
        i.addCategory(Intent.CATEGORY_LAUNCHER);

        ResolveInfo info = new ResolveInfo();
        info.nonLocalizedLabel = TEST_PACKAGE_LABEL;
        rpm.addResolveInfoForIntent(i, info);

        assertThat(rpm.resolveService(i, 0), sameInstance(info));
    }

    @Test
    public void resolveService__NoMatch() throws Exception {
        Intent i = new Intent();
        i.setComponent(new ComponentName("foo.bar", "No Activity"));
        assertThat(rpm.resolveService(i, 0), nullValue());
    }

    @Test
    public void queryActivityIcons__Match() throws Exception {
        Intent i = rpm.getLaunchIntentForPackage(TEST_PACKAGE_NAME);
        Drawable d = new BitmapDrawable();

        rpm.addActivityIcon(i, d);

        assertThat(rpm.getActivityIcon(i), sameInstance(d));
        assertThat(rpm.getActivityIcon(i.getComponent()), sameInstance(d));
    }

    @Test
    public void hasSystemFeature() throws Exception {
        // uninitialized
        assertThat(rpm.hasSystemFeature(PackageManager.FEATURE_CAMERA), equalTo(false));

        // positive
        rpm.setSystemFeature(PackageManager.FEATURE_CAMERA, true);
        assertThat(rpm.hasSystemFeature(PackageManager.FEATURE_CAMERA), equalTo(true));

        // negative
        rpm.setSystemFeature(PackageManager.FEATURE_CAMERA, false);
        assertThat(rpm.hasSystemFeature(PackageManager.FEATURE_CAMERA), equalTo(false));
    }

}