Home | History | Annotate | Download | only in testutils
      1 /*
      2  * Copyright (C) 2016 The Android Open Source Project
      3  *
      4  * Licensed under the Apache License, Version 2.0 (the "License");
      5  * you may not use this file except in compliance with the License.
      6  * You may obtain a copy of the License at
      7  *
      8  *      http://www.apache.org/licenses/LICENSE-2.0
      9  *
     10  * Unless required by applicable law or agreed to in writing, software
     11  * distributed under the License is distributed on an "AS IS" BASIS,
     12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
     13  * See the License for the specific language governing permissions and
     14  * limitations under the License.
     15  */
     16 package com.android.settings.testutils;
     17 
     18 import java.net.MalformedURLException;
     19 import java.net.URL;
     20 import org.junit.runners.model.InitializationError;
     21 import org.robolectric.RobolectricTestRunner;
     22 import org.robolectric.annotation.Config;
     23 import org.robolectric.manifest.AndroidManifest;
     24 import org.robolectric.res.Fs;
     25 import org.robolectric.res.ResourcePath;
     26 
     27 import java.util.List;
     28 
     29 /**
     30  * Custom test runner for the testing of BluetoothPairingDialogs. This is needed because the
     31  * default behavior for robolectric is just to grab the resource directory in the target package.
     32  * We want to override this to add several spanning different projects.
     33  */
     34 public class SettingsRobolectricTestRunner extends RobolectricTestRunner {
     35 
     36     /**
     37      * We don't actually want to change this behavior, so we just call super.
     38      */
     39     public SettingsRobolectricTestRunner(Class<?> testClass) throws InitializationError {
     40         super(testClass);
     41     }
     42 
     43     /**
     44      * We are going to create our own custom manifest so that we can add multiple resource
     45      * paths to it. This lets us access resources in both Settings and SettingsLib in our tests.
     46      */
     47     @Override
     48     protected AndroidManifest getAppManifest(Config config) {
     49         try {
     50             // Using the manifest file's relative path, we can figure out the application directory.
     51             final URL appRoot = new URL("file:packages/apps/Settings/");
     52             final URL manifestPath = new URL(appRoot, "AndroidManifest.xml");
     53             final URL resDir = new URL(appRoot, "tests/robotests/res");
     54             final URL assetsDir = new URL(appRoot, "tests/robotests/assets");
     55 
     56             // By adding any resources from libraries we need the AndroidManifest, we can access
     57             // them from within the parallel universe's resource loader.
     58             return new AndroidManifest(Fs.fromURL(manifestPath), Fs.fromURL(resDir),
     59                 Fs.fromURL(assetsDir), "com.android.settings") {
     60                 @Override
     61                 public List<ResourcePath> getIncludedResourcePaths() {
     62                     final List<ResourcePath> paths = super.getIncludedResourcePaths();
     63                     addIncludedResourcePaths(paths);
     64                     return paths;
     65                 }
     66             };
     67         } catch (MalformedURLException e) {
     68             throw new RuntimeException("SettingsRobolectricTestRunner failure", e);
     69         }
     70     }
     71 
     72     public static void addIncludedResourcePaths(List<ResourcePath> paths) {
     73         try {
     74             paths.add(new ResourcePath(null,
     75                 Fs.fromURL(new URL("file:packages/apps/Settings/res")), null));
     76             paths.add(new ResourcePath(null,
     77                 Fs.fromURL(new URL("file:frameworks/base/packages/SettingsLib/res")), null));
     78             paths.add(new ResourcePath(null,
     79                 Fs.fromURL(new URL("file:frameworks/base/core/res/res")), null));
     80             paths.add(new ResourcePath(null,
     81                 Fs.fromURL(new URL("file:frameworks/opt/setupwizard/library/main/res")), null));
     82             paths.add(new ResourcePath(null,
     83                 Fs.fromURL(new URL("file:frameworks/opt/setupwizard/library/gingerbread/res")), null));
     84             paths.add(new ResourcePath(null,
     85                 Fs.fromURL(new URL("file:frameworks/opt/setupwizard/library/recyclerview/res")), null));
     86             paths.add(new ResourcePath(null,
     87                 Fs.fromURL(new URL("file:frameworks/support/v7/appcompat/res")), null));
     88             paths.add(new ResourcePath(null,
     89                 Fs.fromURL(new URL("file:frameworks/support/v7/cardview/res")), null));
     90         } catch (MalformedURLException e) {
     91             throw new RuntimeException("SettingsRobolectricTestRunner failure", e);
     92         }
     93     }
     94 }
     95