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.anyString;
import static org.mockito.Matchers.eq;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.verify;
import static org.mockito.Mockito.when;
import net.pterodactylus.sone.core.Core;
+import net.pterodactylus.sone.data.LocalSone;
import net.pterodactylus.sone.data.Sone;
import net.pterodactylus.sone.freenet.SimpleFieldSetBuilder;
import net.pterodactylus.sone.freenet.fcp.Command.Response;
import freenet.support.SimpleFieldSet;
+import com.google.common.base.Optional;
import org.junit.Test;
/**
@Test
public void testUnlockingALocalSone() throws FcpException {
- Sone localSone = mock(Sone.class);
+ LocalSone localSone = mock(LocalSone.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(Optional.<Sone>of(localSone));
+ when(core.getLocalSone(eq("LocalSone"))).thenReturn(Optional.of(localSone));
SimpleFieldSet fields = new SimpleFieldSetBuilder().put("Sone", "LocalSone").get();
UnlockSoneCommand unlockSoneCommand = new UnlockSoneCommand(core);
public void testUnlockingARemoteSone() throws FcpException {
Sone removeSone = mock(Sone.class);
Core core = mock(Core.class);
- when(core.getSone(eq("RemoteSone"), anyBoolean())).thenReturn(removeSone);
+ when(core.getLocalSone(anyString())).thenReturn(Optional.<LocalSone>absent());
+ when(core.getSone(eq("RemoteSone"))).thenReturn(Optional.of(removeSone));
SimpleFieldSet fields = new SimpleFieldSetBuilder().put("Sone", "RemoteSone").get();
UnlockSoneCommand unlockSoneCommand = new UnlockSoneCommand(core);