Merge branch 'release-0.9.7'
[Sone.git] / src / test / java / net / pterodactylus / sone / template / AlbumAccessorTest.java
index b31eabf..56d2f29 100644 (file)
@@ -14,11 +14,11 @@ import java.util.Arrays;
 import java.util.Collections;
 import java.util.List;
 
-import net.pterodactylus.sone.TestUtil;
 import net.pterodactylus.sone.data.Album;
 import net.pterodactylus.sone.data.Image;
 import net.pterodactylus.sone.data.Profile;
 import net.pterodactylus.sone.data.Sone;
+import net.pterodactylus.sone.test.TestUtil;
 
 import org.hamcrest.Description;
 import org.hamcrest.Matcher;
@@ -104,7 +104,7 @@ public class AlbumAccessorTest {
                when(album.getImages()).thenReturn(albumImages);
                int matchedImage = 0;
                for (int i = 0; i < 1000; i++) {
-                       Image randomImage = (Image) albumAccessor.get(null, album, "randomImage");
+                       Image randomImage = (Image) albumAccessor.get(null, album, "albumImage");
                        if (randomImage == image) {
                                matchedImage++;
                        }
@@ -115,7 +115,7 @@ public class AlbumAccessorTest {
        @Test
        public void albumImageIsNullIfThereAreNoImagesInAnAlbum() {
                when(album.getImages()).thenReturn(Collections.<Image>emptyList());
-               assertThat(albumAccessor.get(null, album, "randomImage"), nullValue());
+               assertThat(albumAccessor.get(null, album, "albumImage"), nullValue());
        }
 
 }