From 7ed204c5c223a86e00bd55433696dbf30b63841a Mon Sep 17 00:00:00 2001 From: =?utf8?q?David=20=E2=80=98Bombe=E2=80=99=20Roden?= Date: Sat, 16 Nov 2013 00:20:37 +0100 Subject: [PATCH] Add unit test for PluginConnector. --- .../sone/freenet/plugin/PluginConnectorTest.java | 109 +++++++++++++++++++++ 1 file changed, 109 insertions(+) create mode 100644 src/test/java/net/pterodactylus/sone/freenet/plugin/PluginConnectorTest.java diff --git a/src/test/java/net/pterodactylus/sone/freenet/plugin/PluginConnectorTest.java b/src/test/java/net/pterodactylus/sone/freenet/plugin/PluginConnectorTest.java new file mode 100644 index 0000000..7784bdc --- /dev/null +++ b/src/test/java/net/pterodactylus/sone/freenet/plugin/PluginConnectorTest.java @@ -0,0 +1,109 @@ +/* + * Sone - PluginConnectorTest.java - Copyright © 2013 David Roden + * + * This program is free software: you can redistribute it and/or modify + * it under the terms of the GNU General Public License as published by + * the Free Software Foundation, either version 3 of the License, or + * (at your option) any later version. + * + * This program is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU General Public License for more details. + * + * You should have received a copy of the GNU General Public License + * along with this program. If not, see . + */ + +package net.pterodactylus.sone.freenet.plugin; + +import static org.mockito.Matchers.argThat; +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.Matchers; +import net.pterodactylus.sone.freenet.SimpleFieldSetBuilder; +import net.pterodactylus.sone.freenet.StringBucket; +import net.pterodactylus.sone.freenet.plugin.event.ReceivedReplyEvent; + +import freenet.pluginmanager.PluginNotFoundException; +import freenet.pluginmanager.PluginRespirator; +import freenet.pluginmanager.PluginTalker; +import freenet.support.SimpleFieldSet; +import freenet.support.api.Bucket; + +import com.google.common.eventbus.EventBus; +import org.hamcrest.Description; +import org.hamcrest.Matcher; +import org.hamcrest.TypeSafeMatcher; +import org.junit.Test; + +/** + * Unit test for {@link PluginConnector}. + * + * @author David ‘Bombe’ Roden + */ +public class PluginConnectorTest { + + private static final String PLUGIN_NAME = "org.example.plugin"; + private static final String IDENTIFIER = "Some-Identifier-17"; + + private final EventBus eventBus = mock(EventBus.class); + private final PluginRespirator pluginRespirator = mock(PluginRespirator.class); + private final PluginConnector pluginConnector = new PluginConnector(eventBus, pluginRespirator); + + @Test + public void canSendRequest() throws PluginException, PluginNotFoundException { + SimpleFieldSet fieldSet = new SimpleFieldSetBuilder().put("Length", "35").get(); + Bucket data = new StringBucket("Test"); + PluginTalker pluginTalker = mock(PluginTalker.class); + when(pluginRespirator.getPluginTalker(pluginConnector, PLUGIN_NAME, IDENTIFIER)).thenReturn(pluginTalker); + pluginConnector.sendRequest(PLUGIN_NAME, IDENTIFIER, fieldSet, data); + verify(pluginTalker).send(argThat(Matchers.matches(fieldSet)), eq(data)); + } + + @Test(expected = PluginException.class) + public void unknownPluginNameCausesAnError() throws PluginNotFoundException, PluginException { + when(pluginRespirator.getPluginTalker(pluginConnector, PLUGIN_NAME, IDENTIFIER)).thenThrow(PluginNotFoundException.class); + pluginConnector.sendRequest(PLUGIN_NAME, IDENTIFIER, null, null); + } + + @Test + public void replyIsPostedToEventBus() { + SimpleFieldSet fieldSet = new SimpleFieldSetBuilder().put("Length", "35").get(); + Bucket data = new StringBucket("Test"); + pluginConnector.onReply(PLUGIN_NAME, IDENTIFIER, fieldSet, data); + verify(eventBus).post(argThat(matches(pluginConnector, PLUGIN_NAME, IDENTIFIER, fieldSet, data))); + } + + private static Matcher matches(final PluginConnector pluginConnector, final String pluginName, final String identitier, final SimpleFieldSet fieldSet, final Bucket data) { + return new TypeSafeMatcher() { + @Override + protected boolean matchesSafely(ReceivedReplyEvent receivedReplyEvent) { + if (receivedReplyEvent.pluginConnector() != pluginConnector) { + return false; + } + if (!receivedReplyEvent.pluginName().equals(pluginName)) { + return false; + } + if (!receivedReplyEvent.identifier().equals(identitier)) { + return false; + } + if (!Matchers.matches(fieldSet).matches(receivedReplyEvent.fieldSet())) { + return false; + } + if (data != receivedReplyEvent.data()) { + return false; + } + return true; + } + + @Override + public void describeTo(Description description) { + } + }; + } + +} -- 2.7.4