X-Git-Url: https://git.pterodactylus.net/?p=Sone.git;a=blobdiff_plain;f=src%2Ftest%2Fjava%2Fnet%2Fpterodactylus%2Fsone%2Fdata%2FMocks.java;h=f56a1d4de7a4f40e72f60fd49547ededb8fbb825;hp=72a4e75e1d7d5bf48f2b5f704cfce954ca5559d3;hb=aa5513efdffab4a204d9bea43d4e7820aa92da73;hpb=b30df9db8b6400e00e431ce2a92370cda91615c7

diff --git a/src/test/java/net/pterodactylus/sone/data/Mocks.java b/src/test/java/net/pterodactylus/sone/data/Mocks.java
index 72a4e75..f56a1d4 100644
--- a/src/test/java/net/pterodactylus/sone/data/Mocks.java
+++ b/src/test/java/net/pterodactylus/sone/data/Mocks.java
@@ -101,6 +101,13 @@ public class Mocks {
 				return sones.values();
 			}
 		});
+		when(core.getLocalSone(anyString())).then(new Answer<Optional<Sone>>() {
+			@Override
+			public Optional<Sone> answer(InvocationOnMock invocation) throws Throwable {
+				Sone localSone = sones.get(invocation.getArguments()[0]);
+				return ((localSone == null) || (!localSone.isLocal())) ? Optional.<Sone>absent() : of(localSone);
+			}
+		});
 		when(core.getLocalSones()).then(new Answer<Collection<Sone>>() {
 			@Override
 			public Collection<Sone> answer(InvocationOnMock invocation) throws Throwable {