Removed delays from unit tests
Change-Id: I10679f09d4c533bad26b37419ca2ac8f821b08f6
diff --git a/apps/ofagent/src/test/java/org/onosproject/ofagent/impl/OFAgentManagerTest.java b/apps/ofagent/src/test/java/org/onosproject/ofagent/impl/OFAgentManagerTest.java
index 52b06f4..6e98181 100644
--- a/apps/ofagent/src/test/java/org/onosproject/ofagent/impl/OFAgentManagerTest.java
+++ b/apps/ofagent/src/test/java/org/onosproject/ofagent/impl/OFAgentManagerTest.java
@@ -17,6 +17,7 @@
import com.google.common.collect.Lists;
import com.google.common.collect.Sets;
+import com.google.common.util.concurrent.MoreExecutors;
import org.junit.After;
import org.junit.Before;
import org.junit.Test;
@@ -112,6 +113,7 @@
ofAgentStore = new DistributedOFAgentStore();
TestUtils.setField(ofAgentStore, "coreService", createMock(CoreService.class));
TestUtils.setField(ofAgentStore, "storageService", new TestStorageService());
+ TestUtils.setField(ofAgentStore, "eventExecutor", MoreExecutors.newDirectExecutorService());
ofAgentStore.activate();
expect(mockCoreService.registerApplication(anyObject()))
@@ -243,15 +245,13 @@
}
private void validateEvents(Enum... types) {
- TestTools.assertAfter(100, () -> {
- int i = 0;
- assertEquals("Number of events does not match", types.length, testListener.events.size());
- for (Event event : testListener.events) {
- assertEquals("Incorrect event received", types[i], event.type());
- i++;
- }
- testListener.events.clear();
- });
+ int i = 0;
+ assertEquals("Number of events does not match", types.length, testListener.events.size());
+ for (Event event : testListener.events) {
+ assertEquals("Incorrect event received", types[i], event.type());
+ i++;
+ }
+ testListener.events.clear();
}
private static class TestOFAgentListener implements OFAgentListener {