diff --git a/spring-data-rest-core/src/test/java/org/springframework/data/rest/AbstractJMockTests.java b/spring-data-rest-core/src/test/java/org/springframework/data/rest/AbstractJMockTests.java index 1e87fc0da..10f2463c0 100644 --- a/spring-data-rest-core/src/test/java/org/springframework/data/rest/AbstractJMockTests.java +++ b/spring-data-rest-core/src/test/java/org/springframework/data/rest/AbstractJMockTests.java @@ -1,19 +1,16 @@ package org.springframework.data.rest; -import org.jmock.integration.junit4.JMock; -import org.jmock.integration.junit4.JUnit4Mockery; +import org.jmock.integration.junit4.JUnitRuleMockery; import org.jmock.lib.legacy.ClassImposteriser; -import org.junit.runner.RunWith; /** * Abstract base classes for JUnit tests that use JMock. * * @author Jon Brisbin */ -@RunWith(JMock.class) public abstract class AbstractJMockTests { - protected JUnit4Mockery context = new JUnit4Mockery() {{ + protected JUnitRuleMockery context = new JUnitRuleMockery() {{ setImposteriser(ClassImposteriser.INSTANCE); }}; diff --git a/spring-data-rest-repository/src/test/java/org/springframework/data/rest/AbstractJMockTests.java b/spring-data-rest-repository/src/test/java/org/springframework/data/rest/AbstractJMockTests.java index 1e87fc0da..10f2463c0 100644 --- a/spring-data-rest-repository/src/test/java/org/springframework/data/rest/AbstractJMockTests.java +++ b/spring-data-rest-repository/src/test/java/org/springframework/data/rest/AbstractJMockTests.java @@ -1,19 +1,16 @@ package org.springframework.data.rest; -import org.jmock.integration.junit4.JMock; -import org.jmock.integration.junit4.JUnit4Mockery; +import org.jmock.integration.junit4.JUnitRuleMockery; import org.jmock.lib.legacy.ClassImposteriser; -import org.junit.runner.RunWith; /** * Abstract base classes for JUnit tests that use JMock. * * @author Jon Brisbin */ -@RunWith(JMock.class) public abstract class AbstractJMockTests { - protected JUnit4Mockery context = new JUnit4Mockery() {{ + protected JUnitRuleMockery context = new JUnitRuleMockery() {{ setImposteriser(ClassImposteriser.INSTANCE); }};