1 files changed, 12 insertions, 1 deletions
diff --git a/tests/src/com/android/camera/gallery/ImageListUberUnitTests.java b/tests/src/com/android/camera/gallery/ImageListUberUnitTests.java
index a123d05..457dbe4 100644
--- a/tests/src/com/android/camera/gallery/ImageListUberUnitTests.java
+++ b/tests/src/com/android/camera/gallery/ImageListUberUnitTests.java
@@ -14,16 +14,20 @@ public class ImageListUberUnitTests extends AndroidTestCase {
         listB.addImage(new MockImage(1, 1));
         ImageListUber uber = new ImageListUber(
                 new IImageList[] {listA, listB}, ImageManager.SORT_DESCENDING);
+        uber.open(getContext().getContentResolver());
 
         assertEquals(2, uber.getImageAt(0).fullSizeImageId());
         assertEquals(1, uber.getImageAt(1).fullSizeImageId());
         assertEquals(0, uber.getImageAt(2).fullSizeImageId());
+        uber.close();
 
         uber = new ImageListUber(
                 new IImageList[] {listA, listB}, ImageManager.SORT_DESCENDING);
+        uber.open(getContext().getContentResolver());
         assertEquals(2, uber.getImageAt(0).fullSizeImageId());
         assertEquals(1, uber.getImageAt(1).fullSizeImageId());
         assertEquals(0, uber.getImageAt(2).fullSizeImageId());
+        uber.close();
     }
 
     public void testTheOrderOfGetImageAtCaseTwo() {
@@ -35,17 +39,20 @@ public class ImageListUberUnitTests extends AndroidTestCase {
         ImageListUber uber = new ImageListUber(
                 new IImageList[] {listB, listA}, ImageManager.SORT_DESCENDING);
 
+        uber.open(getContext().getContentResolver());
         assertEquals(2, uber.getImageAt(0).fullSizeImageId());
         assertEquals(1, uber.getImageAt(1).fullSizeImageId());
         assertEquals(0, uber.getImageAt(2).fullSizeImageId());
+        uber.close();
 
         uber = new ImageListUber(
                 new IImageList[] {listA, listB}, ImageManager.SORT_DESCENDING);
 
+        uber.open(getContext().getContentResolver());
         assertEquals(2, uber.getImageAt(0).fullSizeImageId());
         assertEquals(1, uber.getImageAt(1).fullSizeImageId());
         assertEquals(0, uber.getImageAt(2).fullSizeImageId());
-
+        uber.close();
     }
 
     public void testRemoveImage() {
@@ -57,12 +64,14 @@ public class ImageListUberUnitTests extends AndroidTestCase {
         listB.addImage(target);
         ImageListUber uber = new ImageListUber(
                 new IImageList[] {listB, listA}, ImageManager.SORT_DESCENDING);
+        uber.open(getContext().getContentResolver());
         uber.removeImage(target);
         assertEquals(2, uber.getImageAt(0).fullSizeImageId());
         assertEquals(0, uber.getImageAt(1).fullSizeImageId());
 
         assertEquals(0, uber.getImageIndex(uber.getImageAt(0)));
         assertEquals(1, uber.getImageIndex(uber.getImageAt(1)));
+        uber.close();
     }
 
     public void testRemoveImageAt() {
@@ -74,11 +83,13 @@ public class ImageListUberUnitTests extends AndroidTestCase {
         listB.addImage(target);
         ImageListUber uber = new ImageListUber(
                 new IImageList[] {listB, listA}, ImageManager.SORT_DESCENDING);
+        uber.open(getContext().getContentResolver());
         uber.removeImageAt(1);
         assertEquals(2, uber.getImageAt(0).fullSizeImageId());
         assertEquals(0, uber.getImageAt(1).fullSizeImageId());
 
         assertEquals(0, uber.getImageIndex(uber.getImageAt(0)));
         assertEquals(1, uber.getImageIndex(uber.getImageAt(1)));
+        uber.close();
     }
 }