Distinguish between local and “normal” Sones in FCP handler.

[Sone.git] / src / test / java / net / pterodactylus / sone / fcp / LockSoneCommandTest.java

diff --git a/src/test/java/net/pterodactylus/sone/fcp/LockSoneCommandTest.java b/src/test/java/net/pterodactylus/sone/fcp/LockSoneCommandTest.java
index 383b52a..7132637 100644 (file)
--- a/src/test/java/net/pterodactylus/sone/fcp/LockSoneCommandTest.java
+++ b/src/test/java/net/pterodactylus/sone/fcp/LockSoneCommandTest.java
@@ -20,6 +20,7 @@ package net.pterodactylus.sone.fcp;
 import static org.hamcrest.CoreMatchers.is;
 import static org.hamcrest.CoreMatchers.notNullValue;
 import static org.hamcrest.MatcherAssert.assertThat;
+import static org.mockito.Matchers.anyString;
 import static org.mockito.Matchers.eq;
 import static org.mockito.Mockito.mock;
 import static org.mockito.Mockito.verify;
@@ -66,9 +67,10 @@ public class LockSoneCommandTest {
 
        @Test(expected = FcpException.class)
        public void testLockingARemoteSone() throws FcpException {
-               Sone removeSone = mock(Sone.class);
+               Sone remoteSone = mock(Sone.class);
                Core core = mock(Core.class);
-               when(core.getSone(eq("RemoteSone"))).thenReturn(Optional.of(removeSone));
+               when(core.getLocalSone(anyString())).thenReturn(Optional.<LocalSone>absent());
+               when(core.getSone(eq("RemoteSone"))).thenReturn(Optional.of(remoteSone));
                SimpleFieldSet fields = new SimpleFieldSetBuilder().put("Sone", "RemoteSone").get();
 
                LockSoneCommand lockSoneCommand = new LockSoneCommand(core);