Use MediaMetadataRetriever.setDataSource(Context, Uri) for "content" scheme.

content/public/android/javatests/src/org/chromium/content/browser/MediaResourceGetterTest.java

 // Copyright 2014 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.content.browser;
 
 import android.annotation.SuppressLint;
 import android.content.Context;
 import android.content.pm.PackageManager;
 import android.media.MediaMetadataRetriever;
 import android.net.ConnectivityManager;
+import android.net.Uri;
 import android.test.InstrumentationTestCase;
 import android.test.mock.MockContext;
 import android.test.suitebuilder.annotation.SmallTest;
 import android.util.SparseArray;
 
 import org.chromium.content.browser.MediaResourceGetter.MediaMetadata;
 
 import java.io.File;
 import java.util.Collections;
 import java.util.HashMap;
 import java.util.Map;
 
 /**
  * Tests for MediaResourceGetter.
  */
 @SuppressLint("SdCardPath")
 public class MediaResourceGetterTest extends InstrumentationTestCase {
     private static final String TEST_HTTP_URL = "http://example.com";
     private static final String TEST_USER_AGENT = // Anything, really
             "Mozilla/5.0 (Windows NT 6.2; Win64; x64) AppleWebKit/537.36 "
             + "(KHTML, like Gecko) Chrome/32.0.1667.0 Safari/537.36";
     private static final String TEST_FILE_PATH = "/mnt/sdcard/test";
     private static final String TEST_FILE_URL = "file://" + TEST_FILE_PATH;
+    private static final String TEST_CONTENT_URI =
+            "content://com.android.providers.media.documents/document/video:113";
     private static final String TEST_COOKIES = "yum yum yum!";
     private static final MediaMetadata sEmptyMetadata = new MediaMetadata(0, 0, 0, false);
     private static final String sExternalStorageDirectory = "/test_external_storage";
 
     private static final Map<String, String> sHeadersCookieOnly;
     private static final Map<String, String> sHeadersCookieAndUA;
     private static final Map<String, String> sHeadersUAOnly;
     static {
         Map<String, String> headers = new HashMap<String, String>();
         headers.put("Cookie", TEST_COOKIES);
@@ skipping 45 matching lines @@
      * Helper class that fakes functionality we cannot perform without real
      * media resources. The class being faked here has been architected
      * carefully to allow most of the functionality to be tested. What remains
      * here is overrides of trivial functionality.
      */
     private static class FakeMediaResourceGetter extends MediaResourceGetter {
         // Read these back in tests to ensure proper values passed through
         String mUri = null;
         Map<String, String> mHeaders = null;
         String mPath = null;
+        String mContentUri = null;
         int mFd;
         long mOffset;
         long mLength;
 
         // Write these before tests to configure functionality
         SparseArray<String> mMetadata = null;
         Integer mNetworkType = null;
         boolean mThrowExceptionInConfigure = false;
         boolean mThrowExceptionInExtract = false;
         boolean mFileExists = false;
@@ skipping 23 matching lines @@
         @Override
         public void configure(String path) {
             if (mThrowExceptionInConfigure) {
                 throw new RuntimeException("test exception");
             }
             mPath = path;
         }
 
         // Can't use a real MediaMetadataRetriever as we have no media
         @Override
+        public void configure(Context context, Uri uri) {
+            if (mThrowExceptionInConfigure) {
+                throw new RuntimeException("test exception");
+            }
+            mContentUri = uri.toString();
+        }
+
+        // Can't use a real MediaMetadataRetriever as we have no media
+        @Override
         public String extractMetadata(int key) {
             if (mThrowExceptionInExtract) {
                 throw new RuntimeException("test exception");
             }
             if (mMetadata == null) {
                 return null;
             }
             return mMetadata.get(key);
         }
 
@@ skipping 135 matching lines @@
 
     @SmallTest
     public void testConfigure_Net_Allowed_Ethernet_Cookies_NoUA() {
         mMockContext.mAllowPermission = true;
         mFakeMRG.mNetworkType = ConnectivityManager.TYPE_ETHERNET;
         assertTrue(mFakeMRG.configure(mMockContext, TEST_HTTP_URL,
                                       TEST_COOKIES, null));
         assertEquals(TEST_HTTP_URL, mFakeMRG.mUri);
         assertEquals(sHeadersCookieOnly, mFakeMRG.mHeaders);
         assertNull(mFakeMRG.mPath);
+        assertNull(mFakeMRG.mContentUri);
     }
 
     @SmallTest
     public void testConfigure_Net_Allowed_Wifi_Cookies_NoUA() {
         mMockContext.mAllowPermission = true;
         mFakeMRG.mNetworkType = ConnectivityManager.TYPE_WIFI;
         assertTrue(mFakeMRG.configure(mMockContext, TEST_HTTP_URL,
                                       TEST_COOKIES, null));
         assertEquals(TEST_HTTP_URL, mFakeMRG.mUri);
         assertEquals(sHeadersCookieOnly, mFakeMRG.mHeaders);
         assertNull(mFakeMRG.mPath);
+        assertNull(mFakeMRG.mContentUri);
     }
 
     @SmallTest
     public void testConfigure_Net_Allowed_Ethernet_NoCookies_NoUA() {
         mMockContext.mAllowPermission = true;
         mFakeMRG.mNetworkType = ConnectivityManager.TYPE_ETHERNET;
         assertTrue(mFakeMRG.configure(mMockContext, TEST_HTTP_URL,
                                       "", null));
         assertEquals(TEST_HTTP_URL, mFakeMRG.mUri);
         assertEquals(Collections.emptyMap(), mFakeMRG.mHeaders);
         assertNull(mFakeMRG.mPath);
+        assertNull(mFakeMRG.mContentUri);
     }
 
     @SmallTest
     public void testConfigure_Net_Allowed_Ethernet_Cookies_WithUA() {
         mMockContext.mAllowPermission = true;
         mFakeMRG.mNetworkType = ConnectivityManager.TYPE_ETHERNET;
         assertTrue(mFakeMRG.configure(mMockContext, TEST_HTTP_URL,
                                       TEST_COOKIES, TEST_USER_AGENT));
         assertEquals(TEST_HTTP_URL, mFakeMRG.mUri);
         assertEquals(sHeadersCookieAndUA, mFakeMRG.mHeaders);
         assertNull(mFakeMRG.mPath);
+        assertNull(mFakeMRG.mContentUri);
     }
 
     @SmallTest
     public void testConfigure_Net_Allowed_Ethernet_NoCookies_WithUA() {
         mMockContext.mAllowPermission = true;
         mFakeMRG.mNetworkType = ConnectivityManager.TYPE_ETHERNET;
         assertTrue(mFakeMRG.configure(mMockContext, TEST_HTTP_URL,
                                       "", TEST_USER_AGENT));
         assertEquals(TEST_HTTP_URL, mFakeMRG.mUri);
         assertEquals(sHeadersUAOnly, mFakeMRG.mHeaders);
         assertNull(mFakeMRG.mPath);
+        assertNull(mFakeMRG.mContentUri);
     }
 
     @SmallTest
     public void testConfigure_Net_Allowed_Ethernet_Exception() {
         mMockContext.mAllowPermission = true;
         mFakeMRG.mThrowExceptionInConfigure = true;
         mFakeMRG.mNetworkType = ConnectivityManager.TYPE_ETHERNET;
         assertFalse(mFakeMRG.configure(mMockContext, TEST_HTTP_URL,
                                        "", TEST_USER_AGENT));
         assertNull(mFakeMRG.mUri);
         assertNull(mFakeMRG.mHeaders);
     }
 
     @SmallTest
     public void testConfigure_Net_Allowed_LocalHost_WithNoNetwork() {
         String[] localHostUrls = {
             "http://LocalHost",
             "https://127.0.0.1/",
             "http://[::1]:8888/",
         };
         mMockContext.mAllowPermission = true;
         mFakeMRG.mNetworkType = null;
         for (String localHostUrl : localHostUrls) {
             assertTrue(mFakeMRG.configure(mMockContext, localHostUrl,
                                           TEST_COOKIES, TEST_USER_AGENT));
             assertEquals(localHostUrl, mFakeMRG.mUri);
             assertEquals(sHeadersCookieAndUA, mFakeMRG.mHeaders);
             assertNull(mFakeMRG.mPath);
+            assertNull(mFakeMRG.mContentUri);
         }
     }
 
     @SmallTest
     public void testConfigure_File_Allowed_MntSdcard() {
         final String path = "/mnt/sdcard/test";
         final String url = "file://" + path;
         mFakeMRG.mFileExists = true;
         assertTrue(mFakeMRG.configure(mMockContext, url, "", null));
         assertEquals(path, mFakeMRG.mPath);
         assertNull(mFakeMRG.mUri);
+        assertNull(mFakeMRG.mContentUri);
         assertNull(mFakeMRG.mHeaders);
     }
 
     @SmallTest
     public void testConfigure_File_Allowed_Sdcard() {
         final String path = "/sdcard/test";
         final String url = "file://" + path;
         mFakeMRG.mFileExists = true;
         assertTrue(mFakeMRG.configure(mMockContext, url, "", null));
         assertEquals(path, mFakeMRG.mPath);
         assertNull(mFakeMRG.mUri);
+        assertNull(mFakeMRG.mContentUri);
         assertNull(mFakeMRG.mHeaders);
     }
 
     @SmallTest
     public void testConfigure_File_Allowed_Sdcard_DoesntExist() {
         final String path = "/sdcard/test";
         final String url = "file://" + path;
         mFakeMRG.mFileExists = false;
         assertFalse(mFakeMRG.configure(mMockContext, url, "", null));
         assertNull(mFakeMRG.mPath);
     }
 
     @SmallTest
     public void testConfigure_File_Allowed_ExternalStorage() {
         final String path = sExternalStorageDirectory + "/test";
         final String url = "file://" + path;
         mFakeMRG.mFileExists = true;
         assertTrue(mFakeMRG.configure(mMockContext, url, "", null));
         assertEquals(path, mFakeMRG.mPath);
         assertNull(mFakeMRG.mUri);
+        assertNull(mFakeMRG.mContentUri);
         assertNull(mFakeMRG.mHeaders);
     }
 
     @SmallTest
     public void testConfigure_File_Disallowed_Innocent() {
         final String path = "/malicious/path";
         final String url = "file://" + path;
         mFakeMRG.mFileExists = true;
         assertFalse(mFakeMRG.configure(mMockContext, url, "", null));
         assertNull(mFakeMRG.mPath);
@@ skipping 52 matching lines @@
     public void testConfigure_Blob_Disallowed_Other_Application() {
         final String path = "/data/data/org.some.other.app/cache/";
         final String url = path;
         mFakeMRG.mFileExists = true;
         mFakeMRG.mThrowExceptionInConfigure = true;
         assertFalse(mFakeMRG.configure(mMockContext, url, "", null));
         assertNull(mFakeMRG.mPath);
     }
 
     @SmallTest
+    public void testConfigure_Content_Uri_Allowed() {
+        assertTrue(mFakeMRG.configure(mMockContext, TEST_CONTENT_URI, "", null));
+        assertNull(mFakeMRG.mPath);
+        assertNull(mFakeMRG.mUri);
+        assertEquals(TEST_CONTENT_URI, mFakeMRG.mContentUri);
+    }
+
+    @SmallTest
+    public void testConfigure_Content_Uri_Disallowed() {
+        mFakeMRG.mThrowExceptionInConfigure = true;
+        assertFalse(mFakeMRG.configure(mMockContext, TEST_CONTENT_URI, "", null));
+        assertNull(mFakeMRG.mPath);
+        assertNull(mFakeMRG.mUri);
+        assertNull(mFakeMRG.mContentUri);
+    }
+
+    @SmallTest
     public void testExtract_NoMetadata() {
         mFakeMRG.mFileExists = true;
         assertEquals(sEmptyMetadata, mFakeMRG.extract(
                 mMockContext, TEST_FILE_URL, null, null));
         assertEquals("configured successfully when we shouldn't have",
                      TEST_FILE_PATH, mFakeMRG.mPath); // tricky
     }
 
     @SmallTest
     public void testExtract_WithMetadata_ValidDuration() {
@@ skipping 74 matching lines @@
         final MediaMetadata expected = new MediaMetadata(1, 2, 3, true);
         int fd = 1234;
         long offset = 1000;
         long length = 9000;
         assertEquals(expected, mFakeMRG.extract(fd, offset, length));
         assertEquals(fd, mFakeMRG.mFd);
         assertEquals(offset, mFakeMRG.mOffset);
         assertEquals(length, mFakeMRG.mLength);
     }
 }