Distinguish between local and “normal” Sones in FCP handler.
[Sone.git] / src / test / java / net / pterodactylus / sone / fcp / LockSoneCommandTest.java
index 5932c31..7132637 100644 (file)
@@ -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;
@@ -68,6 +69,7 @@ public class LockSoneCommandTest {
        public void testLockingARemoteSone() throws FcpException {
                Sone remoteSone = mock(Sone.class);
                Core core = mock(Core.class);
+               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();