X-Git-Url: https://git.pterodactylus.net/?p=Sone.git;a=blobdiff_plain;f=src%2Fmain%2Fjava%2Fnet%2Fpterodactylus%2Fsone%2Fdata%2Fimpl%2FAbstractAlbumBuilder.java;h=8e15b6fae77afa54bcde67869230170e9b5d9ca6;hp=312c79550b51f71204eb673d09f623e5663205c5;hb=ec06ae64c86f0b06bb0cf9f8b289e7907e81dffa;hpb=62bb6456a7c2667dec2f8b7b88db465eb6c07257

diff --git a/src/main/java/net/pterodactylus/sone/data/impl/AbstractAlbumBuilder.java b/src/main/java/net/pterodactylus/sone/data/impl/AbstractAlbumBuilder.java
index 312c795..8e15b6f 100644
--- a/src/main/java/net/pterodactylus/sone/data/impl/AbstractAlbumBuilder.java
+++ b/src/main/java/net/pterodactylus/sone/data/impl/AbstractAlbumBuilder.java
@@ -19,6 +19,7 @@ package net.pterodactylus.sone.data.impl;
 
 import static com.google.common.base.Preconditions.checkState;
 
+import net.pterodactylus.sone.data.Sone;
 import net.pterodactylus.sone.database.AlbumBuilder;
 
 /**
@@ -34,6 +35,7 @@ public abstract class AbstractAlbumBuilder implements AlbumBuilder {
 
 	/** The ID of the album to create. */
 	protected String id;
+	protected Sone sone;
 
 	@Override
 	public AlbumBuilder randomId() {
@@ -47,6 +49,11 @@ public abstract class AbstractAlbumBuilder implements AlbumBuilder {
 		return this;
 	}
 
+	public AlbumBuilder by(Sone sone) {
+		this.sone = sone;
+		return this;
+	}
+
 	//
 	// PROTECTED METHODS
 	//
@@ -59,6 +66,7 @@ public abstract class AbstractAlbumBuilder implements AlbumBuilder {
 	 */
 	protected void validate() throws IllegalStateException {
 		checkState((randomId && (id == null)) || (!randomId && (id != null)), "exactly one of random ID or custom ID must be set");
+		checkState(sone != null, "Sone must not be null");
 	}
 
 }