1 files changed, 77 insertions, 0 deletions
diff --git a/v1/src/main/java/com/xtremelabs/robolectric/shadows/ShadowMimeTypeMap.java b/v1/src/main/java/com/xtremelabs/robolectric/shadows/ShadowMimeTypeMap.java
new file mode 100644
index 000000000..c344b6616
--- /dev/null
+++ b/v1/src/main/java/com/xtremelabs/robolectric/shadows/ShadowMimeTypeMap.java
@@ -0,0 +1,77 @@
+package com.xtremelabs.robolectric.shadows;
+
+import android.webkit.MimeTypeMap;
+import com.xtremelabs.robolectric.Robolectric;
+import com.xtremelabs.robolectric.internal.Implementation;
+import com.xtremelabs.robolectric.internal.Implements;
+
+import java.util.HashMap;
+import java.util.Map;
+
+import static com.xtremelabs.robolectric.Robolectric.shadowOf;
+
+/**
+ * Shadow for {@code MimeTypeMap} that allows custom extension <-> mimetype mapping to be set up by tests.
+ */
+@Implements(MimeTypeMap.class)
+public class ShadowMimeTypeMap {
+
+    Map<String, String> extensionToMimeTypeMap = new HashMap<String, String>();
+    Map<String, String> mimeTypeToExtensionMap = new HashMap<String, String>();
+
+    static MimeTypeMap sSingleton = null;
+    static Object sSingletonLock = new Object();
+
+    @Implementation
+    public static MimeTypeMap getSingleton() {
+        if (sSingleton == null) {
+            synchronized (sSingletonLock) {
+                if (sSingleton == null) {
+                    sSingleton = Robolectric.newInstanceOf(MimeTypeMap.class);
+                }
+            }
+        }
+
+        return sSingleton;
+    }
+
+    public static void reset() {
+        shadowOf(getSingleton()).clearMappings();
+    }
+
+    @Implementation
+    public String getMimeTypeFromExtension(String extension) {
+        if (extensionToMimeTypeMap.containsKey(extension))
+            return extensionToMimeTypeMap.get(extension);
+
+        return null;
+    }
+
+    @Implementation
+    public String getExtensionFromMimeType(String mimeType) {
+        if (mimeTypeToExtensionMap.containsKey(mimeType))
+            return mimeTypeToExtensionMap.get(mimeType);
+
+        return null;
+    }
+
+    public void addExtensionMimeTypMapping(String extension, String mimeType) {
+        extensionToMimeTypeMap.put(extension, mimeType);
+        mimeTypeToExtensionMap.put(mimeType, extension);
+    }
+
+    public void clearMappings() {
+        extensionToMimeTypeMap.clear();
+        mimeTypeToExtensionMap.clear();
+    }
+
+    @Implementation
+    public boolean hasExtension(String extension) {
+        return extensionToMimeTypeMap.containsKey(extension);
+    }
+
+    @Implementation
+    public boolean hasMimeType(String mimeType) {
+        return mimeTypeToExtensionMap.containsKey(mimeType);
+    }
+}