Use the memory database to create all albums.

diff --git a/src/main/java/net/pterodactylus/sone/core/Core.java b/src/main/java/net/pterodactylus/sone/core/Core.java
index 8ce908b..0f0b6d3 100644 (file)
--- a/src/main/java/net/pterodactylus/sone/core/Core.java
+++ b/src/main/java/net/pterodactylus/sone/core/Core.java
@@ -52,7 +52,6 @@ import net.pterodactylus.sone.core.event.SoneLockedEvent;
 import net.pterodactylus.sone.core.event.SoneRemovedEvent;
 import net.pterodactylus.sone.core.event.SoneUnlockedEvent;
 import net.pterodactylus.sone.data.Album;
 import net.pterodactylus.sone.core.event.SoneRemovedEvent;
 import net.pterodactylus.sone.core.event.SoneUnlockedEvent;
 import net.pterodactylus.sone.data.Album;

-import net.pterodactylus.sone.data.AlbumImpl;

 import net.pterodactylus.sone.data.Client;
 import net.pterodactylus.sone.data.Image;
 import net.pterodactylus.sone.data.Post;
 import net.pterodactylus.sone.data.Client;
 import net.pterodactylus.sone.data.Image;
 import net.pterodactylus.sone.data.Post;


@@ -641,7 +640,7 @@ public class Core extends AbstractService implements SoneProvider, PostProvider,
                if (!create) {
                        return null;
                }
                if (!create) {
                        return null;
                }

-               Album newAlbum = new AlbumImpl(albumId);
+               Album newAlbum = database.newAlbumBuilder().withId(albumId).build();

                database.storeAlbum(newAlbum);
                return newAlbum;
        }
                database.storeAlbum(newAlbum);
                return newAlbum;
        }


@@ -1601,7 +1600,7 @@ public class Core extends AbstractService implements SoneProvider, PostProvider,
         * @return The new album
         */
        public Album createAlbum(Sone sone, Album parent) {
         * @return The new album
         */
        public Album createAlbum(Sone sone, Album parent) {

-               AlbumImpl album = new AlbumImpl();
+               Album album = database.newAlbumBuilder().randomId().build();

                database.storeAlbum(album);
                album.setSone(sone);
                parent.addAlbum(album);
                database.storeAlbum(album);
                album.setSone(sone);
                parent.addAlbum(album);