From 016178045f7082c0a4b47fb690047d3be0b697d0 Mon Sep 17 00:00:00 2001
From: akwizgran <michael@briarproject.org>
Date: Thu, 19 Dec 2019 17:44:24 +0000
Subject: [PATCH] Make integration tests less sensitive to sync order.

---
 .../briar/test/BriarIntegrationTest.java           | 14 +++++++++-----
 1 file changed, 9 insertions(+), 5 deletions(-)

diff --git a/briar-core/src/test/java/org/briarproject/briar/test/BriarIntegrationTest.java b/briar-core/src/test/java/org/briarproject/briar/test/BriarIntegrationTest.java
index c900410d2..266d3dabe 100644
--- a/briar-core/src/test/java/org/briarproject/briar/test/BriarIntegrationTest.java
+++ b/briar-core/src/test/java/org/briarproject/briar/test/BriarIntegrationTest.java
@@ -130,7 +130,8 @@ public abstract class BriarIntegrationTest<C extends BriarIntegrationTestCompone
 	protected C c0, c1, c2;
 
 	private final Semaphore messageSemaphore = new Semaphore(0);
-	private final AtomicInteger messageCounter = new AtomicInteger(0);
+	private final AtomicInteger deliveryCounter = new AtomicInteger(0);
+	private final AtomicInteger validationCounter = new AtomicInteger(0);
 	private final AtomicInteger ackCounter = new AtomicInteger(0);
 	private final File testDir = TestUtils.getTestDirectory();
 	private final String AUTHOR0 = "Author 0";
@@ -212,14 +213,14 @@ public abstract class BriarIntegrationTest<C extends BriarIntegrationTestCompone
 					if (event.getState() == DELIVERED) {
 						LOG.info("Delivered new message "
 								+ event.getMessageId());
-						messageCounter.addAndGet(1);
+						deliveryCounter.addAndGet(1);
 						loadAndLogMessage(event.getMessageId());
 						deliveryWaiter.resume();
 					} else if (event.getState() == INVALID ||
 							event.getState() == PENDING) {
 						LOG.info("Validated new " + event.getState().name() +
 								" message " + event.getMessageId());
-						messageCounter.addAndGet(1);
+						validationCounter.addAndGet(1);
 						loadAndLogMessage(event.getMessageId());
 						validationWaiter.resume();
 					}
@@ -380,10 +381,13 @@ public abstract class BriarIntegrationTest<C extends BriarIntegrationTestCompone
 
 		if (valid) {
 			deliveryWaiter.await(TIMEOUT, num);
+			assertEquals("Messages delivered", num,
+					deliveryCounter.getAndSet(0));
 		} else {
 			validationWaiter.await(TIMEOUT, num);
+			assertEquals("Messages validated", num,
+					validationCounter.getAndSet(0));
 		}
-		assertEquals("Messages delivered", num, messageCounter.getAndSet(0));
 		try {
 			messageSemaphore.tryAcquire(num, TIMEOUT, MILLISECONDS);
 		} catch (InterruptedException e) {
@@ -421,7 +425,7 @@ public abstract class BriarIntegrationTest<C extends BriarIntegrationTestCompone
 
 		ackWaiter.await(TIMEOUT, num);
 		assertEquals("ACKs delivered", num, ackCounter.getAndSet(0));
-		assertEquals("No messages delivered", 0, messageCounter.get());
+		assertEquals("No messages delivered", 0, deliveryCounter.get());
 		try {
 			messageSemaphore.tryAcquire(num, TIMEOUT, MILLISECONDS);
 		} catch (InterruptedException e) {