tests/tests/classloaderfactory/src/android/app/classloaderfactory/cts/AppComponentFactoryTest.java

/*
 * Copyright (C) 2018 The Android Open Source 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 android.app.classloaderfactory.cts;

import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertNotEquals;

import android.content.ComponentName;
import android.content.Context;
import android.content.Intent;
import android.content.pm.ApplicationInfo;

import androidx.test.InstrumentationRegistry;
import androidx.test.runner.AndroidJUnit4;

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

import java.io.File;
import java.io.FileOutputStream;
import java.io.InputStream;

@RunWith(AndroidJUnit4.class)
public class AppComponentFactoryTest {
    private static final String SECONDARY_APK_RESOURCE = "ClassLoaderFactoryTestSecondary.apk";
    private static final String CLASS_PACKAGE_NAME =
            AppComponentFactoryTest.class.getPackage().getName();

    static File writeSecondaryApkToDisk(ApplicationInfo aInfo) throws Exception {
        File outFile = File.createTempFile(AppComponentFactoryTest.class.getSimpleName(), ".apk",
                new File(aInfo.dataDir));

        InputStream is = AppComponentFactoryTest.class.getClassLoader().getResourceAsStream(
                SECONDARY_APK_RESOURCE);
        FileOutputStream os = new FileOutputStream(outFile);
        outFile.setReadOnly();
        final byte[] buffer = new byte[4096];
        int n;
        while ((n = is.read(buffer)) >= 0) {
            os.write(buffer, 0, n);
        }
        os.close();

        return outFile;
    }

    private final Context mContext = InstrumentationRegistry.getContext();

    @Test
    public void testApplication() {
        // Test that application context is MyApplication class from the secondary APK.
        // The assertion fails if framework did not use the custom class loader.
        assertEquals(mContext.getApplicationContext().getClass().getName(),
                CLASS_PACKAGE_NAME + ".MyApplication");
        assertNotEquals(mContext.getApplicationContext().getClass().getClassLoader(),
                AppComponentFactoryTest.class.getClassLoader());
    }

    @Test
    public void testActivity() {
        // Start MyActivity from the secondary APK. This will throw ActivityNotFoundException
        // if framework does not use the custom class loader.
        mContext.startActivity(new Intent()
                .setComponent(new ComponentName(mContext.getPackageName(),
                        CLASS_PACKAGE_NAME + ".MyActivity"))
                .addFlags(Intent.FLAG_ACTIVITY_NEW_TASK));
    }
}