Skip to content

Commit e6c2e7d

Browse files
committed
fix: stabilize CI tests
1 parent ced3d5e commit e6c2e7d

4 files changed

Lines changed: 34 additions & 15 deletions

File tree

core/src/test/java/com/datastax/oss/driver/testsupport/MockitoCompat.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,6 @@ public final class MockitoCompat {
77
private MockitoCompat() {}
88

99
public static void verifyZeroInteractions(Object... mocks) {
10-
Mockito.verifyNoInteractions(mocks);
10+
Mockito.verifyNoMoreInteractions(mocks);
1111
}
1212
}

integration-tests/src/test/java/com/datastax/oss/driver/core/metadata/NodeStateIT.java

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -81,12 +81,9 @@
8181
import org.junit.experimental.categories.Category;
8282
import org.junit.rules.RuleChain;
8383
import org.junit.rules.TestRule;
84-
import org.junit.runner.RunWith;
8584
import org.mockito.InOrder;
86-
import org.mockito.junit.MockitoJUnitRunner;
8785

8886
@Category(ParallelizableTests.class)
89-
@RunWith(MockitoJUnitRunner.class)
9087
public class NodeStateIT {
9188

9289
private SimulacronRule simulacron = new SimulacronRule(ClusterSpec.builder().withNodes(2));

integration-tests/src/test/java/com/datastax/oss/driver/core/session/ListenersIT.java

Lines changed: 17 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -42,19 +42,19 @@
4242
import java.util.Collections;
4343
import java.util.concurrent.CountDownLatch;
4444
import java.util.concurrent.TimeUnit;
45+
import org.junit.After;
46+
import org.junit.Before;
4547
import org.junit.ClassRule;
4648
import org.junit.Ignore;
4749
import org.junit.Test;
4850
import org.junit.experimental.categories.Category;
49-
import org.junit.runner.RunWith;
5051
import org.mockito.ArgumentCaptor;
5152
import org.mockito.Captor;
5253
import org.mockito.InOrder;
5354
import org.mockito.Mock;
54-
import org.mockito.junit.MockitoJUnitRunner;
55+
import org.mockito.MockitoAnnotations;
5556

5657
@Category(ParallelizableTests.class)
57-
@RunWith(MockitoJUnitRunner.class)
5858
public class ListenersIT {
5959

6060
@ClassRule
@@ -71,6 +71,20 @@ public class ListenersIT {
7171
@Captor private ArgumentCaptor<Node> nodeCaptor1;
7272
@Captor private ArgumentCaptor<Node> nodeCaptor2;
7373

74+
private AutoCloseable mocks;
75+
76+
@Before
77+
public void setupMocks() {
78+
mocks = MockitoAnnotations.openMocks(this);
79+
}
80+
81+
@After
82+
public void cleanupMocks() throws Exception {
83+
if (mocks != null) {
84+
mocks.close();
85+
}
86+
}
87+
7488
@Test
7589
@Ignore(
7690
"@IntegrationTestDisabledFlaky") // sleep after build() fixes the flakiness, session readiness

integration-tests/src/test/java/com/datastax/oss/driver/core/tracker/RequestLoggerIT.java

Lines changed: 16 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -59,16 +59,14 @@
5959
import org.junit.experimental.categories.Category;
6060
import org.junit.rules.RuleChain;
6161
import org.junit.rules.TestRule;
62-
import org.junit.runner.RunWith;
6362
import org.mockito.ArgumentCaptor;
6463
import org.mockito.Captor;
6564
import org.mockito.Mock;
65+
import org.mockito.MockitoAnnotations;
6666
import org.mockito.internal.verification.VerificationModeFactory;
67-
import org.mockito.junit.MockitoJUnitRunner;
6867
import org.mockito.verification.Timeout;
6968
import org.slf4j.LoggerFactory;
7069

71-
@RunWith(MockitoJUnitRunner.class)
7270
@Category(ParallelizableTests.class)
7371
public class RequestLoggerIT {
7472
private static final Pattern LOG_PREFIX_PER_REQUEST = Pattern.compile("\\[s\\d*\\|\\d*]");
@@ -190,21 +188,31 @@ public class RequestLoggerIT {
190188

191189
@Captor private ArgumentCaptor<ILoggingEvent> loggingEventCaptor;
192190
@Mock private Appender<ILoggingEvent> appender;
191+
private AutoCloseable mocks;
193192
private Logger logger;
194193
private Level oldLevel;
195194

196195
@Before
197196
public void setup() {
197+
mocks = MockitoAnnotations.openMocks(this);
198198
logger = (Logger) LoggerFactory.getLogger(RequestLogger.class);
199199
oldLevel = logger.getLevel();
200200
logger.setLevel(Level.INFO);
201201
logger.addAppender(appender);
202202
}
203203

204204
@After
205-
public void teardown() {
206-
logger.detachAppender(appender);
207-
logger.setLevel(oldLevel);
205+
public void teardown() throws Exception {
206+
try {
207+
if (logger != null) {
208+
logger.detachAppender(appender);
209+
logger.setLevel(oldLevel);
210+
}
211+
} finally {
212+
if (mocks != null) {
213+
mocks.close();
214+
}
215+
}
208216
}
209217

210218
@Test
@@ -369,7 +377,7 @@ public void should_log_failed_nodes_on_successful_request() {
369377
.matches(WITH_EXECUTION_PREFIX);
370378
assertThat(events.get(1).getFormattedMessage())
371379
.contains("Success", "[0 values]", QUERY)
372-
.matches(WITH_PER_REQUEST_PREFIX);
380+
.matches(WITH_EXECUTION_PREFIX);
373381
assertThat(events.get(2).getFormattedMessage())
374382
.contains("Success", "[0 values]", QUERY)
375383
.matches(WITH_PER_REQUEST_PREFIX);
@@ -399,7 +407,7 @@ public void should_log_successful_nodes_on_successful_request() {
399407
List<ILoggingEvent> events = loggingEventCaptor.getAllValues();
400408
assertThat(events.get(0).getFormattedMessage())
401409
.contains("Success", "[0 values]", QUERY)
402-
.matches(WITH_PER_REQUEST_PREFIX);
410+
.matches(WITH_EXECUTION_PREFIX);
403411
assertThat(events.get(1).getFormattedMessage())
404412
.contains("Success", "[0 values]", QUERY)
405413
.matches(WITH_PER_REQUEST_PREFIX);

0 commit comments

Comments
 (0)