X-Git-Url: https://git.pterodactylus.net/?a=blobdiff_plain;f=src%2Ftest%2Fjava%2Fnet%2Fpterodactylus%2Fsone%2Fdata%2FMocks.java;h=cbe98444f92d36cedddbe281d3b73b5b4556d092;hb=868b16cb87b3981d8affdfd73eb753ddc83cd3e8;hp=88baf22b39a3d360d8d160cd41867e77d00f305e;hpb=5bee7fb48ebff210d2b68f1939acbdb4ea10d1c1;p=Sone.git

diff --git a/src/test/java/net/pterodactylus/sone/data/Mocks.java b/src/test/java/net/pterodactylus/sone/data/Mocks.java
index 88baf22..cbe9844 100644
--- a/src/test/java/net/pterodactylus/sone/data/Mocks.java
+++ b/src/test/java/net/pterodactylus/sone/data/Mocks.java
@@ -39,6 +39,7 @@ import net.pterodactylus.sone.data.impl.DefaultPostReplyBuilder;
 import net.pterodactylus.sone.database.Database;
 import net.pterodactylus.sone.database.PostReplyBuilder;
 
+import com.google.common.base.Function;
 import com.google.common.base.Optional;
 import com.google.common.collect.FluentIterable;
 import com.google.common.collect.Multimap;
@@ -63,6 +64,12 @@ public class Mocks {
 	public Mocks() {
 		database = mockDatabase();
 		core = mockCore(database);
+		when(database.getSone()).thenReturn(new Function<String, Optional<Sone>>() {
+			@Override
+			public Optional<Sone> apply(String soneId) {
+				return (soneId == null) ? Optional.<Sone>absent() : fromNullable(sones.get(soneId));
+			}
+		});
 		when(core.getLocalSones()).then(new Answer<Collection<Sone>>() {
 			@Override
 			public Collection<Sone> answer(InvocationOnMock invocation) throws Throwable {