diff --git a/bramble-api/src/test/java/org/briarproject/bramble/test/BrambleMockTestCase.java b/bramble-api/src/test/java/org/briarproject/bramble/test/BrambleMockTestCase.java index fb18022f5..c2ee23723 100644 --- a/bramble-api/src/test/java/org/briarproject/bramble/test/BrambleMockTestCase.java +++ b/bramble-api/src/test/java/org/briarproject/bramble/test/BrambleMockTestCase.java @@ -1,12 +1,17 @@ package org.briarproject.bramble.test; import org.jmock.Mockery; +import org.jmock.lib.concurrent.Synchroniser; import org.junit.After; public abstract class BrambleMockTestCase extends BrambleTestCase { protected final Mockery context = new Mockery(); + public BrambleMockTestCase() { + context.setThreadingPolicy(new Synchroniser()); + } + @After public void checkExpectations() { context.assertIsSatisfied(); diff --git a/bramble-core/src/test/java/org/briarproject/bramble/plugin/PluginManagerImplTest.java b/bramble-core/src/test/java/org/briarproject/bramble/plugin/PluginManagerImplTest.java index faca469b1..89f50b67e 100644 --- a/bramble-core/src/test/java/org/briarproject/bramble/plugin/PluginManagerImplTest.java +++ b/bramble-core/src/test/java/org/briarproject/bramble/plugin/PluginManagerImplTest.java @@ -12,10 +12,8 @@ import org.briarproject.bramble.api.plugin.simplex.SimplexPlugin; import org.briarproject.bramble.api.plugin.simplex.SimplexPluginFactory; import org.briarproject.bramble.api.properties.TransportPropertyManager; import org.briarproject.bramble.api.settings.SettingsManager; -import org.briarproject.bramble.test.BrambleTestCase; +import org.briarproject.bramble.test.BrambleMockTestCase; import org.jmock.Expectations; -import org.jmock.Mockery; -import org.jmock.lib.concurrent.Synchroniser; import org.junit.Test; import java.util.Arrays; @@ -24,13 +22,10 @@ import java.util.concurrent.Executors; import static org.briarproject.bramble.test.TestUtils.getTransportId; -public class PluginManagerImplTest extends BrambleTestCase { +public class PluginManagerImplTest extends BrambleMockTestCase { @Test public void testStartAndStop() throws Exception { - Mockery context = new Mockery() {{ - setThreadingPolicy(new Synchroniser()); - }}; Executor ioExecutor = Executors.newSingleThreadExecutor(); EventBus eventBus = context.mock(EventBus.class); PluginConfig pluginConfig = context.mock(PluginConfig.class); @@ -116,7 +111,5 @@ public class PluginManagerImplTest extends BrambleTestCase { // Two plugins should be started and stopped p.startService(); p.stopService(); - - context.assertIsSatisfied(); } }