killbill-memoizeit

cleanup in TestBeatrixListener

11/22/2017 3:38:11 PM

Details

diff --git a/beatrix/src/test/java/org/killbill/billing/beatrix/extbus/TestBeatrixListener.java b/beatrix/src/test/java/org/killbill/billing/beatrix/extbus/TestBeatrixListener.java
index 4998082..ffa9563 100644
--- a/beatrix/src/test/java/org/killbill/billing/beatrix/extbus/TestBeatrixListener.java
+++ b/beatrix/src/test/java/org/killbill/billing/beatrix/extbus/TestBeatrixListener.java
@@ -63,7 +63,6 @@ import org.killbill.bus.api.BusEvent;
 import org.killbill.bus.api.PersistentBus;
 import org.killbill.bus.api.PersistentBus.EventBusException;
 import org.mockito.ArgumentCaptor;
-import org.testng.annotations.AfterMethod;
 import org.testng.annotations.BeforeMethod;
 import org.testng.annotations.Test;
 
@@ -75,7 +74,6 @@ import static org.mockito.Matchers.anyObject;
 import static org.mockito.Mockito.doThrow;
 import static org.mockito.Mockito.mock;
 import static org.mockito.Mockito.never;
-import static org.mockito.Mockito.reset;
 import static org.mockito.Mockito.times;
 import static org.mockito.Mockito.verify;
 import static org.mockito.Mockito.when;
@@ -108,13 +106,15 @@ public class TestBeatrixListener {
     private static final String BROADCAST_EVENT_JSON = "{\"key\":\"value\"}";
 
     private BeatrixListener beatrixListener;
-    private PersistentBus externalBus = mock(PersistentBus.class);
-    private InternalCallContextFactory internalCallContextFactory = mock(InternalCallContextFactory.class);
+    private PersistentBus externalBus;
+    private InternalCallContextFactory internalCallContextFactory;
     private TenantContext tenantContext;
     private ObjectMapper objectMapper;
 
     @BeforeMethod(groups = "fast")
     public void setUp() throws Exception {
+        externalBus = mock(PersistentBus.class);
+        internalCallContextFactory = mock(InternalCallContextFactory.class);
         beatrixListener = new BeatrixListener(externalBus, internalCallContextFactory);
 
         objectMapper = mock(ObjectMapper.class);
@@ -140,11 +140,6 @@ public class TestBeatrixListener {
         when(internalCallContextFactory.createTenantContext(internalContext)).thenReturn(tenantContext);
     }
 
-    @AfterMethod(groups = "fast")
-    public void tearDown() throws Exception {
-        reset(externalBus);
-    }
-
     @Test(groups = "fast")
     public void testAccountCreate() throws Exception {
         AccountCreationInternalEvent event = mock(AccountCreationInternalEvent.class);