honeymoose
/
OpenSearch
mirror of https://github.com/honeymoose/OpenSearch.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Fix timing issue in test 

Original commit: elastic/x-pack-elasticsearch@7fe221ef15
This commit is contained in:
David Kyle 2017-12-21 09:19:14 +00:00
parent 8c99a4dc78
commit 3f6478e798
1 changed files with 26 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

35
plugin/src/test/java/org/elasticsearch/xpack/ml/integration/JobProviderIT.java
View File

@ -169,10 +169,13 @@ public class JobProviderIT extends XPackSingleNodeTestCase {
indexCalendars(calendars);
CountDownLatch latch = new CountDownLatch(1);
AtomicReference<Exception> exceptionHolder = new AtomicReference<>();
final AtomicReference<Exception> exceptionHolder = new AtomicReference<>();
jobProvider.removeJobFromCalendars("bar", ActionListener.wrap(
r -> latch.countDown(),
exceptionHolder::set));
e -> {
exceptionHolder.set(e);
latch.countDown();
}));
latch.await();
if (exceptionHolder.get() != null) {
@ -189,10 +192,12 @@ public class JobProviderIT extends XPackSingleNodeTestCase {
assertThat(catFoo.getJobIds(), contains("cat", "foo"));
CountDownLatch latch2 = new CountDownLatch(1);
exceptionHolder = new AtomicReference<>();
jobProvider.removeJobFromCalendars("cat", ActionListener.wrap(
r -> latch2.countDown(),
exceptionHolder::set));
e -> {
exceptionHolder.set(e);
latch2.countDown();
}));
latch2.await();
if (exceptionHolder.get() != null) {
@ -219,10 +224,13 @@ public class JobProviderIT extends XPackSingleNodeTestCase {
}
jobProvider.calendars(query, ActionListener.wrap(
r -> {
latch.countDown();
result.set(r);
latch.countDown();
},
exceptionHolder::set));
e -> {
exceptionHolder.set(e);
latch.countDown();
}));
latch.await();
if (exceptionHolder.get() != null) {
@ -237,7 +245,10 @@ public class JobProviderIT extends XPackSingleNodeTestCase {
AtomicReference<Exception> exceptionHolder = new AtomicReference<>();
jobProvider.updateCalendar(calendarId, idsToAdd, idsToRemove,
r -> latch.countDown(),
exceptionHolder::set);
e -> {
exceptionHolder.set(e);
latch.countDown();
});
latch.await();
if (exceptionHolder.get() != null) {
@ -253,8 +264,14 @@ public class JobProviderIT extends XPackSingleNodeTestCase {
AtomicReference<Exception> exceptionHolder = new AtomicReference<>();
AtomicReference<Calendar> calendarHolder = new AtomicReference<>();
jobProvider.calendar(calendarId, ActionListener.wrap(
c -> { latch.countDown(); calendarHolder.set(c); },
exceptionHolder::set)
c -> {
calendarHolder.set(c);
latch.countDown();
},
e -> {
exceptionHolder.set(e);
latch.countDown();
})
);
latch.await();