1 files changed, 160 insertions, 0 deletions
diff --git a/library/test/robotest/src/com/android/setupwizardlib/robolectric/ExternalResources.java b/library/test/robotest/src/com/android/setupwizardlib/robolectric/ExternalResources.java
new file mode 100644
index 0000000..06ef508
--- /dev/null
+++ b/library/test/robotest/src/com/android/setupwizardlib/robolectric/ExternalResources.java
@@ -0,0 +1,160 @@
+/*
+ * 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 com.android.setupwizardlib.robolectric;
+
+import static org.robolectric.RuntimeEnvironment.application;
+import static org.robolectric.Shadows.shadowOf;
+
+import android.content.pm.PackageInfo;
+import android.content.pm.PackageManager.NameNotFoundException;
+import android.content.res.AssetManager;
+import android.content.res.Configuration;
+import androidx.annotation.AnyRes;
+import androidx.annotation.NonNull;
+import androidx.annotation.Nullable;
+import android.util.DisplayMetrics;
+import java.util.HashMap;
+import java.util.Map;
+import org.robolectric.res.ResName;
+import org.robolectric.res.ResType;
+import org.robolectric.res.TypedResource;
+import org.robolectric.shadows.ShadowPackageManager;
+
+/**
+ * Utility class to inject resources for an "external" application in Robolectric tests. This can be
+ * used with {@link org.robolectric.shadows.ShadowPackageManager#resources} to simulate loading
+ * resources from another package.
+ */
+public final class ExternalResources {
+
+  public static Resources injectExternalResources(String packageName) {
+    return injectExternalResources(createPackageInfo(packageName));
+  }
+
+  public static Resources injectExternalResources(PackageInfo packageInfo) {
+    try {
+      application.getPackageManager().getPackageInfo(packageInfo.packageName, 0);
+    } catch (NameNotFoundException e) {
+      // Add the package if it does not exist
+      shadowOf(application.getPackageManager()).addPackage(packageInfo);
+    }
+    Resources resources = Resources.forPackageName(packageInfo.packageName);
+    ShadowPackageManager.resources.put(packageInfo.packageName, resources);
+    return resources;
+  }
+
+  /**
+   * Constructed resources for testing, representing resources external to the current package under
+   * test.
+   */
+  public static class Resources extends android.content.res.Resources {
+
+    private final String packageName;
+
+    public static Resources forPackageName(String packageName) {
+      android.content.res.Resources res = application.getResources();
+      return new Resources(
+          packageName, res.getAssets(), res.getDisplayMetrics(), res.getConfiguration());
+    }
+
+    private Resources(
+        String packageName, AssetManager assets, DisplayMetrics metrics, Configuration config) {
+      super(assets, metrics, config);
+      this.packageName = packageName;
+    }
+
+    @Override
+    public int getIdentifier(String name, String defType, String defPackage) {
+      Integer resourceId = resourceIds.get(ResName.qualifyResName(name, defPackage, defType));
+      if (resourceId == null) {
+        return 0;
+      }
+      return resourceId;
+    }
+
+    @Override
+    public int getInteger(int id) {
+      return (int) get(id, ResType.INTEGER);
+    }
+
+    public void putInteger(String name, int value) {
+      put(
+          ResName.qualifyResName(name, packageName, "integer"),
+          new TypedResource<>(value, ResType.INTEGER, null));
+    }
+
+    @Override
+    public int getColor(int id) {
+      return (int) get(id, ResType.COLOR);
+    }
+
+    @Override
+    public int getColor(int id, @Nullable Theme theme) {
+      return (int) get(id, ResType.COLOR);
+    }
+
+    public void putColor(String name, int value) {
+      put(
+          ResName.qualifyResName(name, packageName, "color"),
+          new TypedResource<>(value, ResType.COLOR, null));
+    }
+
+    @NonNull
+    @Override
+    public CharSequence getText(int id) {
+      return (CharSequence) get(id, ResType.CHAR_SEQUENCE);
+    }
+
+    @NonNull
+    @Override
+    public String getString(int id) {
+      return get(id, ResType.CHAR_SEQUENCE).toString();
+    }
+
+    public void putText(String name, CharSequence value) {
+      put(
+          ResName.qualifyResName(name, packageName, "string"),
+          new TypedResource<>(value, ResType.CHAR_SEQUENCE, null));
+    }
+
+    private final Map<Integer, TypedResource<?>> overrideResources = new HashMap<>();
+    private final Map<ResName, Integer> resourceIds = new HashMap<>();
+    private int nextId = 1;
+
+    private <T> void put(ResName resName, TypedResource<T> value) {
+      int id = nextId++;
+      overrideResources.put(id, value);
+      resourceIds.put(resName, id);
+    }
+
+    private Object get(@AnyRes int id, ResType type) {
+      TypedResource<?> override = overrideResources.get(id);
+      if (override != null && override.getResType() == type) {
+        return override.getData();
+      }
+      throw new NotFoundException();
+    }
+  }
+
+  private static PackageInfo createPackageInfo(String packageName) {
+    PackageInfo packageInfo = new PackageInfo();
+    packageInfo.packageName = packageName;
+    return packageInfo;
+  }
+
+  private ExternalResources() {}
+}