|
@@ -17,7 +17,6 @@ import org.elasticsearch.persistent.PersistentTasksCustomMetaData;
|
|
|
import org.elasticsearch.search.aggregations.Aggregations;
|
|
|
import org.elasticsearch.search.aggregations.bucket.composite.CompositeAggregation;
|
|
|
import org.elasticsearch.tasks.TaskId;
|
|
|
-import org.elasticsearch.tasks.TaskManager;
|
|
|
import org.elasticsearch.test.ESTestCase;
|
|
|
import org.elasticsearch.threadpool.TestThreadPool;
|
|
|
import org.elasticsearch.threadpool.ThreadPool;
|
|
@@ -68,10 +67,8 @@ public class RollupJobTaskTests extends ESTestCase {
|
|
|
Client client = mock(Client.class);
|
|
|
when(client.settings()).thenReturn(Settings.EMPTY);
|
|
|
SchedulerEngine schedulerEngine = new SchedulerEngine(SETTINGS, Clock.systemUTC());
|
|
|
- TaskId taskId = new TaskId("node", 123);
|
|
|
- RollupJobTask task = new RollupJobTask(1, "type", "action", taskId, job,
|
|
|
+ RollupJobTask task = new RollupJobTask(1, "type", "action", new TaskId("node", 123), job,
|
|
|
status, client, schedulerEngine, pool, Collections.emptyMap());
|
|
|
- task.init(null, mock(TaskManager.class), taskId.toString(), 123);
|
|
|
assertThat(((RollupJobStatus)task.getStatus()).getIndexerState(), equalTo(IndexerState.STOPPED));
|
|
|
assertThat(((RollupJobStatus)task.getStatus()).getPosition().size(), equalTo(1));
|
|
|
assertTrue(((RollupJobStatus)task.getStatus()).getPosition().containsKey("foo"));
|
|
@@ -83,10 +80,8 @@ public class RollupJobTaskTests extends ESTestCase {
|
|
|
Client client = mock(Client.class);
|
|
|
when(client.settings()).thenReturn(Settings.EMPTY);
|
|
|
SchedulerEngine schedulerEngine = new SchedulerEngine(SETTINGS, Clock.systemUTC());
|
|
|
- TaskId taskId = new TaskId("node", 123);
|
|
|
- RollupJobTask task = new RollupJobTask(1, "type", "action", taskId, job,
|
|
|
+ RollupJobTask task = new RollupJobTask(1, "type", "action", new TaskId("node", 123), job,
|
|
|
status, client, schedulerEngine, pool, Collections.emptyMap());
|
|
|
- task.init(null, mock(TaskManager.class), taskId.toString(), 123);
|
|
|
assertThat(((RollupJobStatus)task.getStatus()).getIndexerState(), equalTo(IndexerState.STOPPED));
|
|
|
assertThat(((RollupJobStatus)task.getStatus()).getPosition().size(), equalTo(1));
|
|
|
assertTrue(((RollupJobStatus)task.getStatus()).getPosition().containsKey("foo"));
|
|
@@ -98,10 +93,8 @@ public class RollupJobTaskTests extends ESTestCase {
|
|
|
Client client = mock(Client.class);
|
|
|
when(client.settings()).thenReturn(Settings.EMPTY);
|
|
|
SchedulerEngine schedulerEngine = new SchedulerEngine(SETTINGS, Clock.systemUTC());
|
|
|
- TaskId taskId = new TaskId("node", 123);
|
|
|
- RollupJobTask task = new RollupJobTask(1, "type", "action", taskId, job,
|
|
|
+ RollupJobTask task = new RollupJobTask(1, "type", "action", new TaskId("node", 123), job,
|
|
|
status, client, schedulerEngine, pool, Collections.emptyMap());
|
|
|
- task.init(null, mock(TaskManager.class), taskId.toString(), 123);
|
|
|
assertThat(((RollupJobStatus)task.getStatus()).getIndexerState(), equalTo(IndexerState.STOPPED));
|
|
|
assertThat(((RollupJobStatus)task.getStatus()).getPosition().size(), equalTo(1));
|
|
|
assertTrue(((RollupJobStatus)task.getStatus()).getPosition().containsKey("foo"));
|
|
@@ -113,10 +106,8 @@ public class RollupJobTaskTests extends ESTestCase {
|
|
|
Client client = mock(Client.class);
|
|
|
when(client.settings()).thenReturn(Settings.EMPTY);
|
|
|
SchedulerEngine schedulerEngine = new SchedulerEngine(SETTINGS, Clock.systemUTC());
|
|
|
- TaskId taskId = new TaskId("node", 123);
|
|
|
- RollupJobTask task = new RollupJobTask(1, "type", "action", taskId, job,
|
|
|
+ RollupJobTask task = new RollupJobTask(1, "type", "action", new TaskId("node", 123), job,
|
|
|
status, client, schedulerEngine, pool, Collections.emptyMap());
|
|
|
- task.init(null, mock(TaskManager.class), taskId.toString(), 123);
|
|
|
assertThat(((RollupJobStatus)task.getStatus()).getIndexerState(), equalTo(IndexerState.STARTED));
|
|
|
assertThat(((RollupJobStatus)task.getStatus()).getPosition().size(), equalTo(1));
|
|
|
assertTrue(((RollupJobStatus)task.getStatus()).getPosition().containsKey("foo"));
|
|
@@ -128,10 +119,8 @@ public class RollupJobTaskTests extends ESTestCase {
|
|
|
Client client = mock(Client.class);
|
|
|
when(client.settings()).thenReturn(Settings.EMPTY);
|
|
|
SchedulerEngine schedulerEngine = new SchedulerEngine(SETTINGS, Clock.systemUTC());
|
|
|
- TaskId taskId = new TaskId("node", 123);
|
|
|
- RollupJobTask task = new RollupJobTask(1, "type", "action", taskId, job,
|
|
|
+ RollupJobTask task = new RollupJobTask(1, "type", "action", new TaskId("node", 123), job,
|
|
|
status, client, schedulerEngine, pool, Collections.emptyMap());
|
|
|
- task.init(null, mock(TaskManager.class), taskId.toString(), 123);
|
|
|
assertThat(((RollupJobStatus)task.getStatus()).getIndexerState(), equalTo(IndexerState.STARTED));
|
|
|
assertThat(((RollupJobStatus)task.getStatus()).getPosition().size(), equalTo(1));
|
|
|
assertTrue(((RollupJobStatus)task.getStatus()).getPosition().containsKey("foo"));
|
|
@@ -143,10 +132,8 @@ public class RollupJobTaskTests extends ESTestCase {
|
|
|
Client client = mock(Client.class);
|
|
|
when(client.settings()).thenReturn(Settings.EMPTY);
|
|
|
SchedulerEngine schedulerEngine = new SchedulerEngine(SETTINGS, Clock.systemUTC());
|
|
|
- TaskId taskId = new TaskId("node", 123);
|
|
|
- RollupJobTask task = new RollupJobTask(1, "type", "action", taskId, job,
|
|
|
+ RollupJobTask task = new RollupJobTask(1, "type", "action", new TaskId("node", 123), job,
|
|
|
status, client, schedulerEngine, pool, Collections.emptyMap());
|
|
|
- task.init(null, mock(TaskManager.class), taskId.toString(), 123);
|
|
|
assertThat(((RollupJobStatus)task.getStatus()).getIndexerState(), equalTo(IndexerState.STARTED));
|
|
|
assertThat(((RollupJobStatus)task.getStatus()).getPosition().size(), equalTo(1));
|
|
|
assertTrue(((RollupJobStatus)task.getStatus()).getPosition().containsKey("foo"));
|
|
@@ -157,10 +144,8 @@ public class RollupJobTaskTests extends ESTestCase {
|
|
|
Client client = mock(Client.class);
|
|
|
when(client.settings()).thenReturn(Settings.EMPTY);
|
|
|
SchedulerEngine schedulerEngine = new SchedulerEngine(SETTINGS, Clock.systemUTC());
|
|
|
- TaskId taskId = new TaskId("node", 123);
|
|
|
- RollupJobTask task = new RollupJobTask(1, "type", "action", taskId, job,
|
|
|
+ RollupJobTask task = new RollupJobTask(1, "type", "action", new TaskId("node", 123), job,
|
|
|
null, client, schedulerEngine, pool, Collections.emptyMap());
|
|
|
- task.init(null, mock(TaskManager.class), taskId.toString(), 123);
|
|
|
assertThat(((RollupJobStatus)task.getStatus()).getIndexerState(), equalTo(IndexerState.STOPPED));
|
|
|
assertNull(((RollupJobStatus)task.getStatus()).getPosition());
|
|
|
}
|
|
@@ -171,10 +156,8 @@ public class RollupJobTaskTests extends ESTestCase {
|
|
|
Client client = mock(Client.class);
|
|
|
when(client.settings()).thenReturn(Settings.EMPTY);
|
|
|
SchedulerEngine schedulerEngine = new SchedulerEngine(SETTINGS, Clock.systemUTC());
|
|
|
- TaskId taskId = new TaskId("node", 123);
|
|
|
- RollupJobTask task = new RollupJobTask(1, "type", "action", taskId, job,
|
|
|
+ RollupJobTask task = new RollupJobTask(1, "type", "action", new TaskId("node", 123), job,
|
|
|
status, client, schedulerEngine, pool, Collections.emptyMap());
|
|
|
- task.init(null, mock(TaskManager.class), taskId.toString(), 123);
|
|
|
assertThat(((RollupJobStatus)task.getStatus()).getIndexerState(), equalTo(IndexerState.STARTED));
|
|
|
assertThat(((RollupJobStatus)task.getStatus()).getPosition().size(), equalTo(1));
|
|
|
assertTrue(((RollupJobStatus)task.getStatus()).getPosition().containsKey("foo"));
|
|
@@ -203,9 +186,8 @@ public class RollupJobTaskTests extends ESTestCase {
|
|
|
SchedulerEngine schedulerEngine = mock(SchedulerEngine.class);
|
|
|
|
|
|
AtomicInteger counter = new AtomicInteger(0);
|
|
|
- TaskId taskId = new TaskId("node", 123);
|
|
|
- RollupJobTask task = new RollupJobTask(1, "type", "action", taskId, job,
|
|
|
- null, client, schedulerEngine, pool, Collections.emptyMap()) {
|
|
|
+ RollupJobTask task = new RollupJobTask(1, "type", "action", new TaskId("node", 123), job,
|
|
|
+ null, client, schedulerEngine, pool, Collections.emptyMap()) {
|
|
|
@Override
|
|
|
public void updatePersistentTaskState(PersistentTaskState taskState,
|
|
|
ActionListener<PersistentTasksCustomMetaData.PersistentTask<?>> listener) {
|
|
@@ -223,7 +205,6 @@ public class RollupJobTaskTests extends ESTestCase {
|
|
|
counter.incrementAndGet();
|
|
|
}
|
|
|
};
|
|
|
- task.init(null, mock(TaskManager.class), taskId.toString(), 123);
|
|
|
assertThat(((RollupJobStatus)task.getStatus()).getIndexerState(), equalTo(IndexerState.STOPPED));
|
|
|
assertNull(((RollupJobStatus)task.getStatus()).getPosition());
|
|
|
|
|
@@ -282,8 +263,7 @@ public class RollupJobTaskTests extends ESTestCase {
|
|
|
Client client = mock(Client.class);
|
|
|
when(client.settings()).thenReturn(Settings.EMPTY);
|
|
|
SchedulerEngine schedulerEngine = mock(SchedulerEngine.class);
|
|
|
- TaskId taskId = new TaskId("node", 123);
|
|
|
- RollupJobTask task = new RollupJobTask(1, "type", "action", taskId, job,
|
|
|
+ RollupJobTask task = new RollupJobTask(1, "type", "action", new TaskId("node", 123), job,
|
|
|
status, client, schedulerEngine, pool, Collections.emptyMap()) {
|
|
|
@Override
|
|
|
public void updatePersistentTaskState(PersistentTaskState taskState,
|
|
@@ -294,7 +274,6 @@ public class RollupJobTaskTests extends ESTestCase {
|
|
|
new PersistentTasksCustomMetaData.Assignment("foo", "foo")));
|
|
|
}
|
|
|
};
|
|
|
- task.init(null, mock(TaskManager.class), taskId.toString(), 123);
|
|
|
assertThat(((RollupJobStatus)task.getStatus()).getIndexerState(), equalTo(IndexerState.STOPPED));
|
|
|
assertThat(((RollupJobStatus)task.getStatus()).getPosition().size(), equalTo(1));
|
|
|
assertTrue(((RollupJobStatus)task.getStatus()).getPosition().containsKey("foo"));
|
|
@@ -322,8 +301,7 @@ public class RollupJobTaskTests extends ESTestCase {
|
|
|
Client client = mock(Client.class);
|
|
|
when(client.settings()).thenReturn(Settings.EMPTY);
|
|
|
SchedulerEngine schedulerEngine = mock(SchedulerEngine.class);
|
|
|
- TaskId taskId = new TaskId("node", 123);
|
|
|
- RollupJobTask task = new RollupJobTask(1, "type", "action", taskId, job,
|
|
|
+ RollupJobTask task = new RollupJobTask(1, "type", "action", new TaskId("node", 123), job,
|
|
|
status, client, schedulerEngine, pool, Collections.emptyMap()) {
|
|
|
@Override
|
|
|
public void updatePersistentTaskState(PersistentTaskState taskState,
|
|
@@ -334,7 +312,6 @@ public class RollupJobTaskTests extends ESTestCase {
|
|
|
new PersistentTasksCustomMetaData.Assignment("foo", "foo")));
|
|
|
}
|
|
|
};
|
|
|
- task.init(null, mock(TaskManager.class), taskId.toString(), 123);
|
|
|
assertThat(((RollupJobStatus)task.getStatus()).getIndexerState(), equalTo(IndexerState.STOPPED));
|
|
|
assertThat(((RollupJobStatus)task.getStatus()).getPosition().size(), equalTo(1));
|
|
|
assertTrue(((RollupJobStatus)task.getStatus()).getPosition().containsKey("foo"));
|
|
@@ -365,8 +342,7 @@ public class RollupJobTaskTests extends ESTestCase {
|
|
|
when(client.settings()).thenReturn(Settings.EMPTY);
|
|
|
when(client.threadPool()).thenReturn(pool);
|
|
|
SchedulerEngine schedulerEngine = mock(SchedulerEngine.class);
|
|
|
- TaskId taskId = new TaskId("node", 123);
|
|
|
- RollupJobTask task = new RollupJobTask(1, "type", "action", taskId, job,
|
|
|
+ RollupJobTask task = new RollupJobTask(1, "type", "action", new TaskId("node", 123), job,
|
|
|
null, client, schedulerEngine, pool, Collections.emptyMap()) {
|
|
|
@Override
|
|
|
public void updatePersistentTaskState(PersistentTaskState taskState,
|
|
@@ -377,7 +353,6 @@ public class RollupJobTaskTests extends ESTestCase {
|
|
|
new PersistentTasksCustomMetaData.Assignment("foo", "foo")));
|
|
|
}
|
|
|
};
|
|
|
- task.init(null, mock(TaskManager.class), taskId.toString(), 123);
|
|
|
assertThat(((RollupJobStatus)task.getStatus()).getIndexerState(), equalTo(IndexerState.STOPPED));
|
|
|
assertNull(((RollupJobStatus)task.getStatus()).getPosition());
|
|
|
|
|
@@ -435,8 +410,7 @@ public class RollupJobTaskTests extends ESTestCase {
|
|
|
}).when(client).execute(anyObject(), anyObject(), anyObject());
|
|
|
|
|
|
SchedulerEngine schedulerEngine = mock(SchedulerEngine.class);
|
|
|
- TaskId taskId = new TaskId("node", 123);
|
|
|
- RollupJobTask task = new RollupJobTask(1, "type", "action", taskId, job,
|
|
|
+ RollupJobTask task = new RollupJobTask(1, "type", "action", new TaskId("node", 123), job,
|
|
|
null, client, schedulerEngine, pool, Collections.emptyMap()) {
|
|
|
@Override
|
|
|
public void updatePersistentTaskState(PersistentTaskState taskState,
|
|
@@ -453,7 +427,6 @@ public class RollupJobTaskTests extends ESTestCase {
|
|
|
|
|
|
}
|
|
|
};
|
|
|
- task.init(null, mock(TaskManager.class), taskId.toString(), 123);
|
|
|
assertThat(((RollupJobStatus)task.getStatus()).getIndexerState(), equalTo(IndexerState.STOPPED));
|
|
|
assertNull(((RollupJobStatus)task.getStatus()).getPosition());
|
|
|
|
|
@@ -521,8 +494,7 @@ public class RollupJobTaskTests extends ESTestCase {
|
|
|
}).when(client).execute(anyObject(), anyObject(), anyObject());
|
|
|
|
|
|
SchedulerEngine schedulerEngine = mock(SchedulerEngine.class);
|
|
|
- TaskId taskId = new TaskId("node", 123);
|
|
|
- RollupJobTask task = new RollupJobTask(1, "type", "action", taskId, job,
|
|
|
+ RollupJobTask task = new RollupJobTask(1, "type", "action", new TaskId("node", 123), job,
|
|
|
null, client, schedulerEngine, pool, Collections.emptyMap()) {
|
|
|
@Override
|
|
|
public void updatePersistentTaskState(PersistentTaskState taskState,
|
|
@@ -539,7 +511,6 @@ public class RollupJobTaskTests extends ESTestCase {
|
|
|
|
|
|
}
|
|
|
};
|
|
|
- task.init(null, mock(TaskManager.class), taskId.toString(), 123);
|
|
|
assertThat(((RollupJobStatus)task.getStatus()).getIndexerState(), equalTo(IndexerState.STOPPED));
|
|
|
assertNull(((RollupJobStatus)task.getStatus()).getPosition());
|
|
|
|
|
@@ -608,8 +579,7 @@ public class RollupJobTaskTests extends ESTestCase {
|
|
|
|
|
|
SchedulerEngine schedulerEngine = mock(SchedulerEngine.class);
|
|
|
RollupJobStatus status = new RollupJobStatus(IndexerState.STOPPED, null);
|
|
|
- TaskId taskId = new TaskId("node", 123);
|
|
|
- RollupJobTask task = new RollupJobTask(1, "type", "action", taskId, job,
|
|
|
+ RollupJobTask task = new RollupJobTask(1, "type", "action", new TaskId("node", 123), job,
|
|
|
status, client, schedulerEngine, pool, Collections.emptyMap()) {
|
|
|
@Override
|
|
|
public void updatePersistentTaskState(PersistentTaskState taskState,
|
|
@@ -626,7 +596,6 @@ public class RollupJobTaskTests extends ESTestCase {
|
|
|
|
|
|
}
|
|
|
};
|
|
|
- task.init(null, mock(TaskManager.class), taskId.toString(), 123);
|
|
|
assertThat(((RollupJobStatus)task.getStatus()).getIndexerState(), equalTo(IndexerState.STOPPED));
|
|
|
assertNull(((RollupJobStatus)task.getStatus()).getPosition());
|
|
|
|
|
@@ -661,10 +630,8 @@ public class RollupJobTaskTests extends ESTestCase {
|
|
|
Client client = mock(Client.class);
|
|
|
when(client.settings()).thenReturn(Settings.EMPTY);
|
|
|
SchedulerEngine schedulerEngine = new SchedulerEngine(SETTINGS, Clock.systemUTC());
|
|
|
- TaskId taskId = new TaskId("node", 123);
|
|
|
- RollupJobTask task = new RollupJobTask(1, "type", "action", taskId, job,
|
|
|
+ RollupJobTask task = new RollupJobTask(1, "type", "action", new TaskId("node", 123), job,
|
|
|
status, client, schedulerEngine, pool, Collections.emptyMap());
|
|
|
- task.init(null, mock(TaskManager.class), taskId.toString(), 123);
|
|
|
assertThat(((RollupJobStatus)task.getStatus()).getIndexerState(), equalTo(IndexerState.STOPPED));
|
|
|
|
|
|
CountDownLatch latch = new CountDownLatch(1);
|
|
@@ -691,8 +658,7 @@ public class RollupJobTaskTests extends ESTestCase {
|
|
|
SchedulerEngine schedulerEngine = mock(SchedulerEngine.class);
|
|
|
|
|
|
AtomicInteger counter = new AtomicInteger(0);
|
|
|
- TaskId taskId = new TaskId("node", 123);
|
|
|
- RollupJobTask task = new RollupJobTask(1, "type", "action", taskId, job,
|
|
|
+ RollupJobTask task = new RollupJobTask(1, "type", "action", new TaskId("node", 123), job,
|
|
|
null, client, schedulerEngine, pool, Collections.emptyMap()) {
|
|
|
@Override
|
|
|
public void updatePersistentTaskState(PersistentTaskState taskState,
|
|
@@ -714,7 +680,6 @@ public class RollupJobTaskTests extends ESTestCase {
|
|
|
|
|
|
}
|
|
|
};
|
|
|
- task.init(null, mock(TaskManager.class), taskId.toString(), 123);
|
|
|
assertThat(((RollupJobStatus)task.getStatus()).getIndexerState(), equalTo(IndexerState.STOPPED));
|
|
|
assertNull(((RollupJobStatus)task.getStatus()).getPosition());
|
|
|
|
|
@@ -779,15 +744,13 @@ public class RollupJobTaskTests extends ESTestCase {
|
|
|
// the task would end before stop could be called. But to help test out all pathways,
|
|
|
// just in case, we can override markAsCompleted so it's a no-op and test how stop
|
|
|
// handles the situation
|
|
|
- TaskId taskId = new TaskId("node", 123);
|
|
|
- RollupJobTask task = new RollupJobTask(1, "type", "action", taskId, job,
|
|
|
+ RollupJobTask task = new RollupJobTask(1, "type", "action", new TaskId("node", 123), job,
|
|
|
status, client, schedulerEngine, pool, Collections.emptyMap()) {
|
|
|
@Override
|
|
|
public void markAsCompleted() {
|
|
|
latch.countDown();
|
|
|
}
|
|
|
};
|
|
|
- task.init(null, mock(TaskManager.class), taskId.toString(), 123);
|
|
|
assertThat(((RollupJobStatus)task.getStatus()).getIndexerState(), equalTo(IndexerState.STOPPED));
|
|
|
|
|
|
task.onCancelled();
|