X-Git-Url: https://git.pterodactylus.net/?a=blobdiff_plain;f=src%2Ftest%2Fjava%2Fnet%2Fpterodactylus%2Fsone%2Fdata%2FMocks.java;h=19626864526db6378e510804d1facd85cb365295;hb=22d68072fb1770cdc4262ede974bd0a6202d7062;hp=ff52581ace2381cd91f616544e4d79524b11fbc2;hpb=8dbd99ec310a150de00e38231a3739a83f106707;p=Sone.git diff --git a/src/test/java/net/pterodactylus/sone/data/Mocks.java b/src/test/java/net/pterodactylus/sone/data/Mocks.java index ff52581..1962686 100644 --- a/src/test/java/net/pterodactylus/sone/data/Mocks.java +++ b/src/test/java/net/pterodactylus/sone/data/Mocks.java @@ -24,32 +24,40 @@ import static com.google.common.collect.ArrayListMultimap.create; import static com.google.common.collect.Maps.newHashMap; import static com.google.common.collect.Ordering.from; import static java.util.Collections.emptySet; +import static org.mockito.Matchers.any; import static org.mockito.Matchers.anyString; import static org.mockito.Matchers.eq; import static org.mockito.Mockito.doAnswer; import static org.mockito.Mockito.mock; import static org.mockito.Mockito.when; -import java.net.URI; -import java.net.URISyntaxException; import java.util.Collection; import java.util.List; import java.util.Map; import java.util.Set; import net.pterodactylus.sone.core.Core; +import net.pterodactylus.sone.core.Options; +import net.pterodactylus.sone.core.Options.DefaultOption; +import net.pterodactylus.sone.core.Preferences; import net.pterodactylus.sone.data.impl.DefaultPostBuilder; import net.pterodactylus.sone.data.impl.DefaultPostReplyBuilder; import net.pterodactylus.sone.database.Database; +import net.pterodactylus.sone.database.PostBuilder.PostCreated; import net.pterodactylus.sone.database.PostReplyBuilder; +import net.pterodactylus.sone.database.PostReplyBuilder.PostReplyCreated; +import net.pterodactylus.sone.freenet.wot.OwnIdentity; +import net.pterodactylus.sone.utils.IntegerRangePredicate; import net.pterodactylus.sone.web.WebInterface; import net.pterodactylus.sone.web.page.FreenetRequest; -import freenet.clients.http.HTTPRequestImpl; +import freenet.clients.http.SessionManager.Session; +import freenet.clients.http.ToadletContext; import freenet.support.api.HTTPRequest; import com.google.common.base.Function; import com.google.common.base.Optional; +import com.google.common.base.Predicates; import com.google.common.collect.FluentIterable; import com.google.common.collect.HashMultimap; import com.google.common.collect.Multimap; @@ -68,6 +76,8 @@ public class Mocks { private final Multimap sonePosts = create(); private final Map sones = newHashMap(); + private Optional currentSone = absent(); + private final Map posts = newHashMap(); private final Multimap postReplies = create(); private final Multimap directedPosts = create(); private final SetMultimap postLikingSones = HashMultimap.create(); @@ -86,18 +96,53 @@ public class Mocks { return (soneId == null) ? Optional.absent() : fromNullable(sones.get(soneId)); } }); + Answer returnCurrentSone = new Answer() { + @Override + public Sone answer(InvocationOnMock invocation) throws Throwable { + return currentSone.orNull(); + } + }; + when(webInterface.getCurrentSone(any(ToadletContext.class))).then(returnCurrentSone); + when(webInterface.getCurrentSone(any(Session.class))).then(returnCurrentSone); when(core.getSones()).then(new Answer>() { @Override public Collection answer(InvocationOnMock invocation) throws Throwable { return sones.values(); } }); + when(core.getLocalSone(anyString())).then(new Answer>() { + @Override + public Optional answer(InvocationOnMock invocation) throws Throwable { + Sone localSone = sones.get(invocation.getArguments()[0]); + return ((localSone == null) || (!localSone.isLocal())) ? Optional.absent() : of(localSone); + } + }); when(core.getLocalSones()).then(new Answer>() { @Override public Collection answer(InvocationOnMock invocation) throws Throwable { return FluentIterable.from(sones.values()).filter(Sone.LOCAL_SONE_FILTER).toList(); } }); + when(core.postCreated()).thenReturn(Optional.of(new PostCreated() { + @Override + public void postCreated(Post post) { + posts.put(post.getId(), post); + sonePosts.put(post.getSone(), post); + } + })); + when(core.postReplyCreated()).then(new Answer>() { + @Override + public Optional answer(InvocationOnMock invocation) throws Throwable { + return Optional.of(new PostReplyCreated() { + @Override + public void postReplyCreated(PostReply postReply) { + postReplies.put(postReply.getPost().get(), postReply); + } + }); + } + }); + Options options = createOptions(); + when(core.getPreferences()).thenReturn(new Preferences(options)); when(database.getDirectedPosts(anyString())).then(new Answer>() { @Override public Collection answer(InvocationOnMock invocation) throws Throwable { @@ -106,6 +151,22 @@ public class Mocks { }); } + private Options createOptions() { + Options options = new Options(); + options.addIntegerOption("InsertionDelay", new DefaultOption(60, new IntegerRangePredicate(0, Integer.MAX_VALUE))); + options.addIntegerOption("PostsPerPage", new DefaultOption(10, new IntegerRangePredicate(1, Integer.MAX_VALUE))); + options.addIntegerOption("ImagesPerPage", new DefaultOption(9, new IntegerRangePredicate(1, Integer.MAX_VALUE))); + options.addIntegerOption("CharactersPerPost", new DefaultOption(400, Predicates.or(new IntegerRangePredicate(50, Integer.MAX_VALUE), Predicates.equalTo(-1)))); + options.addIntegerOption("PostCutOffLength", new DefaultOption(200, Predicates.or(new IntegerRangePredicate(50, Integer.MAX_VALUE), Predicates.equalTo(-1)))); + options.addBooleanOption("RequireFullAccess", new DefaultOption(false)); + options.addIntegerOption("PositiveTrust", new DefaultOption(75, new IntegerRangePredicate(0, 100))); + options.addIntegerOption("NegativeTrust", new DefaultOption(-25, new IntegerRangePredicate(-100, 100))); + options.addStringOption("TrustComment", new DefaultOption("Set from Sone Web Interface")); + options.addBooleanOption("ActivateFcpInterface", new DefaultOption(false)); + options.addIntegerOption("FcpFullAccessRequired", new DefaultOption(2)); + return options; + } + private static Core mockCore(Database database) { Core core = mock(Core.class); when(core.getDatabase()).thenReturn(database); @@ -113,10 +174,15 @@ public class Mocks { return core; } - private static Database mockDatabase() { + private Database mockDatabase() { Database database = mock(Database.class); when(database.getSone(anyString())).thenReturn(Optional.absent()); - when(database.getPost(anyString())).thenReturn(Optional.absent()); + when(database.getPost(anyString())).then(new Answer>() { + @Override + public Optional answer(InvocationOnMock invocation) throws Throwable { + return fromNullable(posts.get(invocation.getArguments()[0])); + } + }); when(database.getPostReply(anyString())).thenReturn(Optional.absent()); return database; } @@ -139,8 +205,10 @@ public class Mocks { return new PostReplyMocker(replyId, sone); } - public FreenetRequest mockRequest(String path) throws URISyntaxException { - HTTPRequest httpRequest = new HTTPRequestImpl(new URI(path), "GET"); + public FreenetRequest mockRequest(String path) { + HTTPRequest httpRequest = mock(HTTPRequest.class); + when(httpRequest.getMethod()).thenReturn("GET"); + when(httpRequest.getPath()).thenReturn(path); FreenetRequest request = mock(FreenetRequest.class); when(request.getHttpRequest()).thenReturn(httpRequest); return request; @@ -150,7 +218,9 @@ public class Mocks { private final Sone mockedSone = mock(Sone.class); private final String id; + private String insertUrI; private boolean local; + private boolean current; private Optional name = absent(); private long time; private Profile profile = new Profile(mockedSone); @@ -165,6 +235,16 @@ public class Mocks { return this; } + public SoneMocker insertUri(String insertUri) { + this.insertUrI = insertUri; + return this; + } + + public SoneMocker current() { + current = true; + return this; + } + public SoneMocker withName(String name) { this.name = fromNullable(name); return this; @@ -193,6 +273,14 @@ public class Mocks { public Sone create() { when(mockedSone.getId()).thenReturn(id); when(mockedSone.isLocal()).thenReturn(local); + if (local) { + OwnIdentity ownIdentity = mock(OwnIdentity.class); + when(ownIdentity.getInsertUri()).thenReturn(insertUrI); + when(mockedSone.getIdentity()).thenReturn(ownIdentity); + } + if (current) { + currentSone = of(mockedSone); + } if (name.isPresent()) { when(mockedSone.getName()).thenReturn(name.get()); }