X-Git-Url: https://git.pterodactylus.net/?a=blobdiff_plain;f=src%2Ftest%2Fjava%2Fnet%2Fpterodactylus%2Fsone%2Ftest%2FTestImageBuilder.java;fp=src%2Ftest%2Fjava%2Fnet%2Fpterodactylus%2Fsone%2Ftest%2FTestImageBuilder.java;h=edeeb1a7b75fa32237affdaf90fe6f464f31be4f;hb=eb32457356fac09e16ec98394c1b5d48f9dfba84;hp=0000000000000000000000000000000000000000;hpb=ec655d61682a70063c1cf540fe3e41613aaf71a5;p=Sone.git

diff --git a/src/test/java/net/pterodactylus/sone/test/TestImageBuilder.java b/src/test/java/net/pterodactylus/sone/test/TestImageBuilder.java
new file mode 100644
index 0000000..edeeb1a
--- /dev/null
+++ b/src/test/java/net/pterodactylus/sone/test/TestImageBuilder.java
@@ -0,0 +1,105 @@
+package net.pterodactylus.sone.test;
+
+import static java.util.UUID.randomUUID;
+import static org.mockito.Mockito.mock;
+import static org.mockito.Mockito.when;
+
+import net.pterodactylus.sone.data.Image;
+import net.pterodactylus.sone.data.Sone;
+import net.pterodactylus.sone.database.ImageBuilder;
+
+/**
+ * {@link ImageBuilder} implementation that returns a mocked {@link Image}.
+ *
+ * @author <a href="mailto:bombe@pterodactylus.net">David ‘Bombe’ Roden</a>
+ */
+public class TestImageBuilder implements ImageBuilder {
+
+	private final Image image;
+
+	public TestImageBuilder() {
+		image = mock(Image.class);
+		Image.Modifier imageModifier = new Image.Modifier() {
+			private Sone sone = image.getSone();
+			private long creationTime = image.getCreationTime();
+			private String key = image.getKey();
+			private String title = image.getTitle();
+			private String description = image.getDescription();
+			private int width = image.getWidth();
+			private int height = image.getHeight();
+
+			@Override
+			public Image.Modifier setSone(Sone sone) {
+				this.sone = sone;
+				return this;
+			}
+
+			@Override
+			public Image.Modifier setCreationTime(long creationTime) {
+				this.creationTime = creationTime;
+				return this;
+			}
+
+			@Override
+			public Image.Modifier setKey(String key) {
+				this.key = key;
+				return this;
+			}
+
+			@Override
+			public Image.Modifier setTitle(String title) {
+				this.title = title;
+				return this;
+			}
+
+			@Override
+			public Image.Modifier setDescription(String description) {
+				this.description = description;
+				return this;
+			}
+
+			@Override
+			public Image.Modifier setWidth(int width) {
+				this.width = width;
+				return this;
+			}
+
+			@Override
+			public Image.Modifier setHeight(int height) {
+				this.height = height;
+				return this;
+			}
+
+			@Override
+			public Image update() throws IllegalStateException {
+				when(image.getSone()).thenReturn(sone);
+				when(image.getCreationTime()).thenReturn(creationTime);
+				when(image.getKey()).thenReturn(key);
+				when(image.getTitle()).thenReturn(title);
+				when(image.getDescription()).thenReturn(description);
+				when(image.getWidth()).thenReturn(width);
+				when(image.getHeight()).thenReturn(height);
+				return image;
+			}
+		};
+		when(image.modify()).thenReturn(imageModifier);
+	}
+
+	@Override
+	public ImageBuilder randomId() {
+		when(image.getId()).thenReturn(randomUUID().toString());
+		return this;
+	}
+
+	@Override
+	public ImageBuilder withId(String id) {
+		when(image.getId()).thenReturn(id);
+		return this;
+	}
+
+	@Override
+	public Image build() throws IllegalStateException {
+		return image;
+	}
+
+}