X-Git-Url: https://git.pterodactylus.net/?a=blobdiff_plain;f=src%2Ftest%2Fjava%2Fnet%2Fpterodactylus%2Fsone%2Fcore%2FWebOfTrustUpdaterTest.java;h=ff646b05e44e6939131e2b30aa7a1228b09aa0de;hb=fd2ae87f8dbc9fa0214dbbd9a6b803f81846a23c;hp=40230ab17d7e1019a1ef45f8b343f4b960289df0;hpb=521f08bf196e8d74e8ab4ca1be4aace02ac56379;p=Sone.git

diff --git a/src/test/java/net/pterodactylus/sone/core/WebOfTrustUpdaterTest.java b/src/test/java/net/pterodactylus/sone/core/WebOfTrustUpdaterTest.java
index 40230ab..ff646b0 100644
--- a/src/test/java/net/pterodactylus/sone/core/WebOfTrustUpdaterTest.java
+++ b/src/test/java/net/pterodactylus/sone/core/WebOfTrustUpdaterTest.java
@@ -1,27 +1,39 @@
 package net.pterodactylus.sone.core;
 
 import static java.lang.Thread.sleep;
+import static java.util.concurrent.TimeUnit.SECONDS;
 import static org.hamcrest.MatcherAssert.assertThat;
 import static org.hamcrest.Matchers.containsString;
 import static org.hamcrest.Matchers.is;
 import static org.hamcrest.Matchers.not;
 import static org.mockito.Matchers.eq;
+import static org.mockito.Mockito.doAnswer;
 import static org.mockito.Mockito.doThrow;
 import static org.mockito.Mockito.mock;
 import static org.mockito.Mockito.never;
+import static org.mockito.Mockito.times;
 import static org.mockito.Mockito.verify;
 import static org.mockito.Mockito.when;
 
+import java.util.concurrent.CountDownLatch;
+import java.util.concurrent.atomic.AtomicInteger;
+
 import net.pterodactylus.sone.core.WebOfTrustUpdater.AddContextJob;
 import net.pterodactylus.sone.core.WebOfTrustUpdater.RemoveContextJob;
 import net.pterodactylus.sone.core.WebOfTrustUpdater.SetPropertyJob;
+import net.pterodactylus.sone.core.WebOfTrustUpdater.SetTrustJob;
 import net.pterodactylus.sone.core.WebOfTrustUpdater.WebOfTrustContextUpdateJob;
 import net.pterodactylus.sone.core.WebOfTrustUpdater.WebOfTrustUpdateJob;
 import net.pterodactylus.sone.freenet.plugin.PluginException;
+import net.pterodactylus.sone.freenet.wot.Identity;
 import net.pterodactylus.sone.freenet.wot.OwnIdentity;
+import net.pterodactylus.sone.freenet.wot.Trust;
 import net.pterodactylus.sone.freenet.wot.WebOfTrustConnector;
+import net.pterodactylus.sone.freenet.wot.WebOfTrustException;
 
 import org.junit.Test;
+import org.mockito.invocation.InvocationOnMock;
+import org.mockito.stubbing.Answer;
 
 /**
  * Unit test for {@link WebOfTrustUpdater} and its subclasses.
@@ -31,6 +43,10 @@ import org.junit.Test;
 public class WebOfTrustUpdaterTest {
 
 	private static final String CONTEXT = "test-context";
+	private static final Integer SCORE = 50;
+	private static final Integer OTHER_SCORE = 25;
+	private static final String TRUST_COMMENT = "set in a test";
+	private static final String PROPERTY_NAME = "test-property";
 	private final WebOfTrustConnector webOfTrustConnector = mock(WebOfTrustConnector.class);
 	private final WebOfTrustUpdater webOfTrustUpdater = new WebOfTrustUpdater(webOfTrustConnector);
 	private final OwnIdentity ownIdentity = when(mock(OwnIdentity.class).getId()).thenReturn("own-identity-id").getMock();
@@ -39,6 +55,7 @@ public class WebOfTrustUpdaterTest {
 	private final WebOfTrustContextUpdateJob contextUpdateJob = webOfTrustUpdater.new WebOfTrustContextUpdateJob(ownIdentity, CONTEXT);
 	private final AddContextJob addContextJob = webOfTrustUpdater.new AddContextJob(ownIdentity, CONTEXT);
 	private final RemoveContextJob removeContextJob = webOfTrustUpdater.new RemoveContextJob(ownIdentity, CONTEXT);
+	private final Identity trustee = when(mock(Identity.class).getId()).thenReturn("trustee-id").getMock();
 
 	private WebOfTrustUpdateJob createWebOfTrustUpdateJob(final boolean success) {
 		return webOfTrustUpdater.new WebOfTrustUpdateJob() {
@@ -221,4 +238,242 @@ public class WebOfTrustUpdaterTest {
 		assertThat(firstSetPropertyJob, not(is(secondSetPropertyJob)));
 	}
 
+	@Test
+	public void setTrustJobSetsTrust() throws PluginException {
+		SetTrustJob setTrustJob = webOfTrustUpdater.new SetTrustJob(ownIdentity, trustee, SCORE, TRUST_COMMENT);
+		setTrustJob.run();
+		verify(webOfTrustConnector).setTrust(eq(ownIdentity), eq(trustee), eq(SCORE), eq(TRUST_COMMENT));
+		verify(trustee).setTrust(eq(ownIdentity), eq(new Trust(SCORE, null, 0)));
+		assertThat(setTrustJob.waitForCompletion(), is(true));
+	}
+
+	@Test
+	public void settingNullTrustRemovesTrust() throws WebOfTrustException {
+		SetTrustJob setTrustJob = webOfTrustUpdater.new SetTrustJob(ownIdentity, trustee, null, TRUST_COMMENT);
+		setTrustJob.run();
+		verify(webOfTrustConnector).removeTrust(eq(ownIdentity), eq(trustee));
+		verify(trustee).removeTrust(eq(ownIdentity));
+		assertThat(setTrustJob.waitForCompletion(), is(true));
+	}
+
+	@Test
+	public void exceptionWhileSettingTrustIsCaught() throws PluginException {
+		doThrow(PluginException.class).when(webOfTrustConnector).setTrust(eq(ownIdentity), eq(trustee), eq(SCORE), eq(TRUST_COMMENT));
+		SetTrustJob setTrustJob = webOfTrustUpdater.new SetTrustJob(ownIdentity, trustee, SCORE, TRUST_COMMENT);
+		setTrustJob.run();
+		verify(webOfTrustConnector).setTrust(eq(ownIdentity), eq(trustee), eq(SCORE), eq(TRUST_COMMENT));
+		verify(trustee, never()).setTrust(eq(ownIdentity), eq(new Trust(SCORE, null, 0)));
+		assertThat(setTrustJob.waitForCompletion(), is(false));
+	}
+
+	@Test
+	public void setTrustJobsWithDifferentClassesAreNotEqual() {
+		SetTrustJob firstSetTrustJob = webOfTrustUpdater.new SetTrustJob(ownIdentity, trustee, SCORE, TRUST_COMMENT);
+		SetTrustJob secondSetTrustJob = webOfTrustUpdater.new SetTrustJob(ownIdentity, trustee, SCORE, TRUST_COMMENT) {
+		};
+		assertThat(firstSetTrustJob, not(is(secondSetTrustJob)));
+		assertThat(secondSetTrustJob, not(is(firstSetTrustJob)));
+	}
+
+	@Test
+	public void setTrustJobsWithDifferentTrustersAreNotEqual() {
+		SetTrustJob firstSetTrustJob = webOfTrustUpdater.new SetTrustJob(ownIdentity, trustee, SCORE, TRUST_COMMENT);
+		SetTrustJob secondSetTrustJob = webOfTrustUpdater.new SetTrustJob(mock(OwnIdentity.class), trustee, SCORE, TRUST_COMMENT);
+		assertThat(firstSetTrustJob, not(is(secondSetTrustJob)));
+		assertThat(secondSetTrustJob, not(is(firstSetTrustJob)));
+	}
+
+	@Test
+	public void setTrustJobsWithDifferentTrusteesAreNotEqual() {
+		SetTrustJob firstSetTrustJob = webOfTrustUpdater.new SetTrustJob(ownIdentity, trustee, SCORE, TRUST_COMMENT);
+		SetTrustJob secondSetTrustJob = webOfTrustUpdater.new SetTrustJob(ownIdentity, mock(Identity.class), SCORE, TRUST_COMMENT);
+		assertThat(firstSetTrustJob, not(is(secondSetTrustJob)));
+		assertThat(secondSetTrustJob, not(is(firstSetTrustJob)));
+	}
+
+	@Test
+	public void setTrustJobsWithDifferentScoreAreEqual() {
+		SetTrustJob firstSetTrustJob = webOfTrustUpdater.new SetTrustJob(ownIdentity, trustee, SCORE, TRUST_COMMENT);
+		SetTrustJob secondSetTrustJob = webOfTrustUpdater.new SetTrustJob(ownIdentity, trustee, OTHER_SCORE, TRUST_COMMENT);
+		assertThat(firstSetTrustJob, is(secondSetTrustJob));
+		assertThat(secondSetTrustJob, is(firstSetTrustJob));
+		assertThat(firstSetTrustJob.hashCode(), is(secondSetTrustJob.hashCode()));
+	}
+
+	@Test
+	public void setTrustJobDoesNotEqualNull() {
+		SetTrustJob setTrustJob = webOfTrustUpdater.new SetTrustJob(ownIdentity, trustee, SCORE, TRUST_COMMENT);
+		assertThat(setTrustJob, not(is((Object) null)));
+	}
+
+	@Test
+	public void toStringOfSetTrustJobContainsIdsOfTrusterAndTrustee() {
+		SetTrustJob setTrustJob = webOfTrustUpdater.new SetTrustJob(ownIdentity, trustee, SCORE, TRUST_COMMENT);
+		assertThat(setTrustJob.toString(), containsString(ownIdentity.getId()));
+		assertThat(setTrustJob.toString(), containsString(trustee.getId()));
+	}
+
+	@Test
+	public void webOfTrustUpdaterStopsWhenItShould() {
+		webOfTrustUpdater.stop();
+		webOfTrustUpdater.serviceRun();
+	}
+
+	@Test
+	public void webOfTrustUpdaterStopsAfterItWasStarted() {
+		webOfTrustUpdater.start();
+		webOfTrustUpdater.stop();
+	}
+
+	@Test
+	public void removePropertyRemovesProperty() throws InterruptedException, PluginException {
+		final CountDownLatch wotCallTriggered = new CountDownLatch(1);
+		doAnswer(new Answer<Void>() {
+			@Override
+			public Void answer(InvocationOnMock invocation) throws Throwable {
+				wotCallTriggered.countDown();
+				return null;
+			}
+		}).when(webOfTrustConnector).removeProperty(eq(ownIdentity), eq(PROPERTY_NAME));
+		webOfTrustUpdater.removeProperty(ownIdentity, PROPERTY_NAME);
+		webOfTrustUpdater.start();
+		assertThat(wotCallTriggered.await(1, SECONDS), is(true));
+	}
+
+	@Test
+	public void multipleCallsToSetPropertyAreCollapsed() throws InterruptedException, PluginException {
+		final CountDownLatch wotCallTriggered = new CountDownLatch(1);
+		doAnswer(new Answer<Void>() {
+			@Override
+			public Void answer(InvocationOnMock invocation) throws Throwable {
+				wotCallTriggered.countDown();
+				return null;
+			}
+		}).when(webOfTrustConnector).removeProperty(eq(ownIdentity), eq(PROPERTY_NAME));
+		webOfTrustUpdater.removeProperty(ownIdentity, PROPERTY_NAME);
+		webOfTrustUpdater.removeProperty(ownIdentity, PROPERTY_NAME);
+		webOfTrustUpdater.start();
+		assertThat(wotCallTriggered.await(1, SECONDS), is(true));
+		verify(webOfTrustConnector).removeProperty(eq(ownIdentity), eq(PROPERTY_NAME));
+	}
+
+	@Test
+	public void addContextWaitWaitsForTheContextToBeAdded() {
+		webOfTrustUpdater.start();
+		assertThat(webOfTrustUpdater.addContextWait(ownIdentity, CONTEXT), is(true));
+		verify(ownIdentity).addContext(eq(CONTEXT));
+	}
+
+	@Test
+	public void multipleCallsToAddContextAreCollapsed() throws InterruptedException, PluginException {
+		final AtomicInteger errorCount = new AtomicInteger();
+		final CountDownLatch addContextsFinished = new CountDownLatch(2);
+		for (int i = 1; i <= 2; i++) {
+			/* this is so fucking volatile. */
+			if (i > 1) {
+				sleep(200);
+			}
+			new Thread(new Runnable() {
+				public void run() {
+					if (!webOfTrustUpdater.addContextWait(ownIdentity, CONTEXT)) {
+						errorCount.incrementAndGet();
+					}
+					addContextsFinished.countDown();
+				}
+			}).start();
+		}
+		webOfTrustUpdater.start();
+		assertThat(addContextsFinished.await(1, SECONDS), is(true));
+		verify(ownIdentity).addContext(eq(CONTEXT));
+		assertThat(errorCount.get(), is(0));
+	}
+
+	@Test
+	public void removeContextRemovesAContext() throws InterruptedException, PluginException {
+		webOfTrustUpdater.start();
+		final CountDownLatch removeContextTrigger = new CountDownLatch(1);
+		doAnswer(new Answer<Void>() {
+			@Override
+			public Void answer(InvocationOnMock invocation) throws Throwable {
+				removeContextTrigger.countDown();
+				return null;
+			}
+		}).when(ownIdentity).removeContext(eq(CONTEXT));
+		webOfTrustUpdater.removeContext(ownIdentity, CONTEXT);
+		removeContextTrigger.await(1, SECONDS);
+		verify(webOfTrustConnector).removeContext(eq(ownIdentity), eq(CONTEXT));
+		verify(ownIdentity).removeContext(eq(CONTEXT));
+	}
+
+	@Test
+	public void removeContextRequestsAreCoalesced() throws InterruptedException, PluginException {
+		final CountDownLatch contextRemovedTrigger = new CountDownLatch(1);
+		doAnswer(new Answer<Void>() {
+			@Override
+			public Void answer(InvocationOnMock invocation) throws Throwable {
+				contextRemovedTrigger.countDown();
+				return null;
+			}
+		}).when(ownIdentity).removeContext(eq(CONTEXT));
+		for (int i = 1; i <= 2; i++) {
+			/* this is so fucking volatile. */
+			if (i > 1) {
+				sleep(200);
+			}
+			new Thread(new Runnable() {
+				public void run() {
+					webOfTrustUpdater.removeContext(ownIdentity, CONTEXT);
+				}
+			}).start();
+		}
+		webOfTrustUpdater.start();
+		assertThat(contextRemovedTrigger.await(1, SECONDS), is(true));
+		verify(webOfTrustConnector).removeContext(eq(ownIdentity), eq(CONTEXT));
+		verify(ownIdentity).removeContext(eq(CONTEXT));
+	}
+
+	@Test
+	public void setTrustSetsTrust() throws InterruptedException, PluginException {
+		final CountDownLatch trustSetTrigger =new CountDownLatch(1);
+		doAnswer(new Answer<Void>() {
+			@Override
+			public Void answer(InvocationOnMock invocation) throws Throwable {
+				trustSetTrigger.countDown();
+				return null;
+			}
+		}).when(trustee).setTrust(eq(ownIdentity), eq(new Trust(SCORE, null, 0)));
+		webOfTrustUpdater.start();
+		webOfTrustUpdater.setTrust(ownIdentity, trustee, SCORE, TRUST_COMMENT);
+		assertThat(trustSetTrigger.await(1, SECONDS), is(true));
+		verify(trustee).setTrust(eq(ownIdentity), eq(new Trust(SCORE, null, 0)));
+		verify(webOfTrustConnector).setTrust(eq(ownIdentity), eq(trustee), eq(SCORE), eq(TRUST_COMMENT));
+	}
+
+	@Test
+	public void setTrustRequestsAreCoalesced() throws InterruptedException, PluginException {
+		final CountDownLatch trustSetTrigger = new CountDownLatch(1);
+		doAnswer(new Answer<Void>() {
+			@Override
+			public Void answer(InvocationOnMock invocation) throws Throwable {
+				trustSetTrigger.countDown();
+				return null;
+			}
+		}).when(trustee).setTrust(eq(ownIdentity), eq(new Trust(SCORE, null, 0)));
+		for (int i = 1; i <= 2; i++) {
+			/* this is so fucking volatile. */
+			if (i > 1) {
+				sleep(200);
+			}
+			new Thread(new Runnable() {
+				public void run() {
+					webOfTrustUpdater.setTrust(ownIdentity, trustee, SCORE, TRUST_COMMENT);
+				}
+			}).start();
+		}
+		webOfTrustUpdater.start();
+		assertThat(trustSetTrigger.await(1, SECONDS), is(true));
+		verify(trustee).setTrust(eq(ownIdentity), eq(new Trust(SCORE, null, 0)));
+		verify(webOfTrustConnector).setTrust(eq(ownIdentity), eq(trustee), eq(SCORE), eq(TRUST_COMMENT));
+	}
+
 }