From: David ‘Bombe’ Roden <bombe@pterodactylus.net>
Date: Fri, 15 Nov 2013 23:20:37 +0000 (+0100)
Subject: Add unit test for PluginConnector.
X-Git-Url: https://git.pterodactylus.net/?p=Sone.git;a=commitdiff_plain;h=7ed204c5c223a86e00bd55433696dbf30b63841a

Add unit test for PluginConnector.
---

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 <http://www.gnu.org/licenses/>.
+ */
+
+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 <a href="mailto:bombe@pterodactylus.net">David ‘Bombe’ Roden</a>
+ */
+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<ReceivedReplyEvent> matches(final PluginConnector pluginConnector, final String pluginName, final String identitier, final SimpleFieldSet fieldSet, final Bucket data) {
+		return new TypeSafeMatcher<ReceivedReplyEvent>() {
+			@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) {
+			}
+		};
+	}
+
+}