[Overlay resource] Support video from overlay package

support video from overlay package.

bug: 110588815
Test: Manually tested and verified
Change-Id: Ia5acc27602309ed7a0a081ef5a9629186f77fb8d

1 files changed, 67 insertions, 0 deletions

diff --git a/library/test/robotest/src/com/android/setupwizardlib/view/IllustrationVideoViewTest.java b/library/test/robotest/src/com/android/setupwizardlib/view/IllustrationVideoViewTest.java
index a39ffb9..a6c1808 100644
--- a/library/test/robotest/src/com/android/setupwizardlib/view/IllustrationVideoViewTest.java
+++ b/library/test/robotest/src/com/android/setupwizardlib/view/IllustrationVideoViewTest.java
@@ -190,6 +190,64 @@ public class IllustrationVideoViewTest {
         assertThat(aspectRatio).isEqualTo(0.0f);
     }
 
+    @Test
+    public void setVideoResId_resetDiffVideoResFromDiffPackage_videoResShouldBeSet() {
+        // VideoRes default set as android.R.color.white with
+        // default package(com.android.setupwizardlib)
+        createDefaultView();
+
+        // reset different videoRes from different package
+        String newPackageName = "com.android.fakepackage";
+        @RawRes int black = android.R.color.black;
+        addMediaInfo(black, newPackageName);
+        mView.setVideoResource(black, newPackageName);
+
+        // should be reset to black with the new package
+        mShadowMediaPlayer = (ShadowMediaPlayer) Shadows.shadowOf(mView.mMediaPlayer);
+        assertThat(mShadowMediaPlayer.getSourceUri().toString())
+                .isEqualTo("android.resource://" + newPackageName + "/"
+                        + android.R.color.black);
+    }
+
+    @Test
+    public void setVideoResId_resetDiffVideoResFromSamePackage_videoResShouldBeSet() {
+        // VideoRes default set as android.R.color.white with
+        // default package(com.android.setupwizardlib)
+        createDefaultView();
+
+        // reset different videoRes from the same package(default package)
+        String defaultPackageName = "com.android.setupwizardlib";
+        @RawRes int black = android.R.color.black;
+        addMediaInfo(black);
+        mView.setVideoResource(black, defaultPackageName);
+
+        // should be reset to black with the same package(default package)
+        mShadowMediaPlayer = (ShadowMediaPlayer) Shadows.shadowOf(mView.mMediaPlayer);
+        assertThat(mShadowMediaPlayer.getSourceUri().toString())
+                .isEqualTo("android.resource://" + defaultPackageName + "/"
+                        + android.R.color.black);
+    }
+
+    @Test
+    public void setVideoResId_resetSameVideoResFromDifferentPackage_videoResShouldBeSet() {
+        // VideoRes default set as android.R.color.white with
+        // default package(com.android.setupwizardlib)
+        createDefaultView();
+
+
+        // reset same videoRes from different package
+        @RawRes int white = android.R.color.white;
+        String newPackageName = "com.android.fakepackage";
+        addMediaInfo(white, newPackageName);
+        mView.setVideoResource(white, newPackageName);
+
+        // should be white with the new package
+        mShadowMediaPlayer = (ShadowMediaPlayer) Shadows.shadowOf(mView.mMediaPlayer);
+        assertThat(mShadowMediaPlayer.getSourceUri().toString())
+                .isEqualTo("android.resource://" + newPackageName + "/"
+                        + android.R.color.white);
+    }
+
     private void createDefaultView() {
         mView = new IllustrationVideoView(
                 application,
@@ -227,6 +285,15 @@ public class IllustrationVideoViewTest {
                 new MediaInfo(5000, 1));
     }
 
+    private void addMediaInfo(@RawRes int res, String packageName) {
+        ShadowMediaPlayer.addMediaInfo(
+                DataSource.toDataSource(
+                        application,
+                        Uri.parse("android.resource://" + packageName + "/" + res),
+                        null),
+                new MediaInfo(5000, 1));
+    }
+
     @Implements(Surface.class)
     @TargetApi(VERSION_CODES.HONEYCOMB)
     public static class ShadowSurface extends org.robolectric.shadows.ShadowSurface {