X-Git-Url: https://git.pterodactylus.net/?a=blobdiff_plain;f=src%2Ftest%2Fjava%2Fnet%2Fpterodactylus%2Fsone%2Ffcp%2FLockSoneCommandTest.java;h=01f9994f9891c7cfa614644c1bc4ce533c7f42f4;hb=2f995a29d8731dc3b3d20e78fd9382be542292ff;hp=3789d08b7a0f7076ae38fc4b68246c5e520351b1;hpb=e351d4aca7c80b6a9e7ddc9775995ce829e51031;p=Sone.git

diff --git a/src/test/java/net/pterodactylus/sone/fcp/LockSoneCommandTest.java b/src/test/java/net/pterodactylus/sone/fcp/LockSoneCommandTest.java
index 3789d08..01f9994 100644
--- a/src/test/java/net/pterodactylus/sone/fcp/LockSoneCommandTest.java
+++ b/src/test/java/net/pterodactylus/sone/fcp/LockSoneCommandTest.java
@@ -17,10 +17,10 @@
 
 package net.pterodactylus.sone.fcp;
 
+import static com.google.common.base.Optional.of;
 import static org.hamcrest.CoreMatchers.is;
 import static org.hamcrest.CoreMatchers.notNullValue;
 import static org.hamcrest.MatcherAssert.assertThat;
-import static org.mockito.Matchers.anyBoolean;
 import static org.mockito.Matchers.eq;
 import static org.mockito.Mockito.mock;
 import static org.mockito.Mockito.verify;
@@ -47,9 +47,10 @@ public class LockSoneCommandTest {
 	public void testLockingALocalSone() throws FcpException {
 		Sone localSone = mock(Sone.class);
 		when(localSone.getId()).thenReturn("LocalSone");
+		when(localSone.isLocal()).thenReturn(true);
 		Core core = mock(Core.class);
-		when(core.getSone(eq("LocalSone"), anyBoolean())).thenReturn(localSone);
-		when(core.getLocalSone(eq("LocalSone"), anyBoolean())).thenReturn(localSone);
+		when(core.getSone(eq("LocalSone"))).thenReturn(of(localSone));
+		when(core.getLocalSone(eq("LocalSone"))).thenReturn(of(localSone));
 		SimpleFieldSet fields = new SimpleFieldSetBuilder().put("Sone", "LocalSone").get();
 
 		LockSoneCommand lockSoneCommand = new LockSoneCommand(core);
@@ -66,7 +67,7 @@ public class LockSoneCommandTest {
 	public void testLockingARemoteSone() throws FcpException {
 		Sone removeSone = mock(Sone.class);
 		Core core = mock(Core.class);
-		when(core.getSone(eq("RemoteSone"), anyBoolean())).thenReturn(removeSone);
+		when(core.getSone(eq("RemoteSone"))).thenReturn(of(removeSone));
 		SimpleFieldSet fields = new SimpleFieldSetBuilder().put("Sone", "RemoteSone").get();
 
 		LockSoneCommand lockSoneCommand = new LockSoneCommand(core);