Index: base/test/android/javatests/src/org/chromium/base/test/util/AdvancedMockContext.java |
diff --git a/base/test/android/javatests/src/org/chromium/base/test/util/AdvancedMockContext.java b/base/test/android/javatests/src/org/chromium/base/test/util/AdvancedMockContext.java |
new file mode 100644 |
index 0000000000000000000000000000000000000000..a57a8b0319f118f24cab11a59227b889aaf9e3d6 |
--- /dev/null |
+++ b/base/test/android/javatests/src/org/chromium/base/test/util/AdvancedMockContext.java |
@@ -0,0 +1,103 @@ |
+// Copyright (c) 2013 The Chromium Authors. All rights reserved. |
+// Use of this source code is governed by a BSD-style license that can be |
+// found in the LICENSE file. |
+ |
+package org.chromium.base.test.util; |
+ |
+import android.content.ContentResolver; |
+import android.content.Context; |
+import android.content.ContextWrapper; |
+import android.content.SharedPreferences; |
+import android.test.mock.MockContentResolver; |
+import android.test.mock.MockContext; |
+ |
+import java.util.HashMap; |
+import java.util.Map; |
+ |
+/** |
+ * ContextWrapper that adds functionality for SharedPreferences and a way to set and retrieve flags. |
+ */ |
+public class AdvancedMockContext extends ContextWrapper { |
+ |
+ private final MockContentResolver mMockContentResolver = new MockContentResolver(); |
+ |
+ private final Map<String, SharedPreferences> mSharedPreferences = |
+ new HashMap<String, SharedPreferences>(); |
+ |
+ private final Map<String, Boolean> flags = new HashMap<String, Boolean>(); |
+ |
+ public AdvancedMockContext(Context base) { |
+ super(base); |
+ } |
+ |
+ public AdvancedMockContext() { |
+ super(new MockContext()); |
+ } |
+ |
+ @Override |
+ public String getPackageName() { |
+ return getBaseContext().getPackageName(); |
+ } |
+ |
+ @Override |
+ public Context getApplicationContext() { |
+ return this; |
+ } |
+ |
+ @Override |
+ public ContentResolver getContentResolver() { |
+ return mMockContentResolver; |
+ } |
+ |
+ public MockContentResolver getMockContentResolver() { |
+ return mMockContentResolver; |
+ } |
+ |
+ @Override |
+ public SharedPreferences getSharedPreferences(String name, int mode) { |
+ synchronized (mSharedPreferences) { |
+ if (!mSharedPreferences.containsKey(name)) { |
+ // Auto-create shared preferences to mimic Android Context behavior |
+ mSharedPreferences.put(name, new InMemorySharedPreferences()); |
+ } |
+ return mSharedPreferences.get(name); |
+ } |
+ } |
+ |
+ public void addSharedPreferences(String name, Map<String, Object> data) { |
+ synchronized (mSharedPreferences) { |
+ mSharedPreferences.put(name, new InMemorySharedPreferences(data)); |
+ } |
+ } |
+ |
+ public void setFlag(String key) { |
+ flags.put(key, true); |
+ } |
+ |
+ public void clearFlag(String key) { |
+ flags.remove(key); |
+ } |
+ |
+ public boolean isFlagSet(String key) { |
+ return flags.containsKey(key) && flags.get(key); |
+ } |
+ |
+ public static class MapBuilder { |
+ |
+ private final Map<String, Object> mData = new HashMap<String, Object>(); |
+ |
+ public static MapBuilder create() { |
+ return new MapBuilder(); |
+ } |
+ |
+ public MapBuilder add(String key, Object value) { |
+ mData.put(key, value); |
+ return this; |
+ } |
+ |
+ public Map<String, Object> build() { |
+ return mData; |
+ } |
+ |
+ } |
+} |