From: Debora Wöpcke <deboradub@gmail.com>
Date: Sat, 1 Aug 2020 05:14:37 +0000 (+0200)
Subject: Add access to the NewKnownGood field in a SubscribedUSKUpdate message.
X-Git-Url: https://git.pterodactylus.net/?a=commitdiff_plain;h=5fc70db494113ea28557b08a614b74f08c781613;p=jFCPlib.git

Add access to the NewKnownGood field in a SubscribedUSKUpdate message.
---

diff --git a/src/main/java/net/pterodactylus/fcp/SubscribedUSKUpdate.java b/src/main/java/net/pterodactylus/fcp/SubscribedUSKUpdate.java
index d39e8db..4b1e75b 100644
--- a/src/main/java/net/pterodactylus/fcp/SubscribedUSKUpdate.java
+++ b/src/main/java/net/pterodactylus/fcp/SubscribedUSKUpdate.java
@@ -67,4 +67,13 @@ public class SubscribedUSKUpdate extends BaseMessage implements Identifiable {
 		return getField("URI");
 	}
 
+	/**
+	 * Returns whether the request returns a new known good.
+	 *
+	 * @return <code>true</code> if it does, <code>false</code> if it does not.
+	 */
+	public boolean isNewKnownGood() {
+		return Boolean.valueOf(getField("NewKnownGood"));
+	}
+
 }
diff --git a/src/test/java/net/pterodactylus/fcp/SubscribedUSKUpdateMessageTest.java b/src/test/java/net/pterodactylus/fcp/SubscribedUSKUpdateMessageTest.java
new file mode 100644
index 0000000..c0dee37
--- /dev/null
+++ b/src/test/java/net/pterodactylus/fcp/SubscribedUSKUpdateMessageTest.java
@@ -0,0 +1,61 @@
+/*
+ * jFCPlib - PeerNoteTest.java - Copyright © 2020 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.fcp;
+
+import static org.hamcrest.MatcherAssert.assertThat;
+import static org.hamcrest.Matchers.is;
+
+import org.junit.Test;
+
+public class SubscribedUSKUpdateMessageTest {
+
+	@Test
+	public void testWithoutNewKnownGood() {
+		FcpMessage receivedMessage = new FcpMessage("SubscribedUSKUpdate");
+		String URIValue = "CHK@anURI";
+		receivedMessage.setField("URI", URIValue);
+
+		SubscribedUSKUpdate subscribedUSKUpdate = new SubscribedUSKUpdate(receivedMessage);
+
+		assertThat(subscribedUSKUpdate.getURI(), is(URIValue));
+		assertThat(subscribedUSKUpdate.isNewKnownGood(), is(false));
+	}
+
+	@Test
+	public void testWithNewKnownGoodFalse() {
+		checkWithNewKnownGood("false", false);
+	}
+
+	@Test
+	public void testWithNewKnownGoodTrue() {
+		checkWithNewKnownGood("true", true);
+	}
+
+	private void checkWithNewKnownGood(String text, boolean value) {
+		FcpMessage receivedMessage = new FcpMessage("SubscribedUSKUpdate");
+		String URIValue = "CHK@anURI";
+		receivedMessage.setField("URI", URIValue);
+		receivedMessage.setField("NewKnownGood", text);
+
+		SubscribedUSKUpdate subscribedUSKUpdate = new SubscribedUSKUpdate(receivedMessage);
+
+		assertThat(subscribedUSKUpdate.getURI(), is(URIValue));
+		assertThat(subscribedUSKUpdate.isNewKnownGood(), is(value));
+	}
+
+}