Move method to remove image from Album to Image.

[Sone.git] / src / main / java / net / pterodactylus / sone / database / memory / MemoryImage.java

diff --git a/src/main/java/net/pterodactylus/sone/database/memory/MemoryImage.java b/src/main/java/net/pterodactylus/sone/database/memory/MemoryImage.java
index a694720..338da0a 100644 (file)
--- a/src/main/java/net/pterodactylus/sone/database/memory/MemoryImage.java
+++ b/src/main/java/net/pterodactylus/sone/database/memory/MemoryImage.java
@@ -50,4 +50,9 @@ public class MemoryImage extends AbstractImage {
                return memoryDatabase.getAlbum(albumId).get();
        }
 
+       @Override
+       public void remove() throws IllegalStateException {
+               memoryDatabase.removeImage(this);
+       }
+
 }