X-Git-Url: https://git.pterodactylus.net/?p=Sone.git;a=blobdiff_plain;f=src%2Ftest%2Fjava%2Fnet%2Fpterodactylus%2Fsone%2Ftemplate%2FAlbumAccessorTest.java;h=28f6fd16d5770df56114485dca11bcfaa9b51cc8;hp=b31eabf7b64f25ef0d718e18fdba407b5dad5f32;hb=62573c314957b1851f4fbe693b8746686caa940a;hpb=d9206828d23e4c0092499c46ffd75f6792c928bc

diff --git a/src/test/java/net/pterodactylus/sone/template/AlbumAccessorTest.java b/src/test/java/net/pterodactylus/sone/template/AlbumAccessorTest.java
index b31eabf..28f6fd1 100644
--- a/src/test/java/net/pterodactylus/sone/template/AlbumAccessorTest.java
+++ b/src/test/java/net/pterodactylus/sone/template/AlbumAccessorTest.java
@@ -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;
@@ -28,8 +28,6 @@ import org.junit.Test;
 
 /**
  * Unit test for {@link AlbumAccessor}.
- *
- * @author <a href="mailto:bombe@pterodactylus.net">David ‘Bombe’ Roden</a>
  */
 public class AlbumAccessorTest {
 
@@ -104,7 +102,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 +113,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());
 	}
 
 }