Skip to content

Commit

Permalink
USE_SPY_TASK_MANAGER_SETTING -> SPY_TASK_MANAGER_SETTING
Browse files Browse the repository at this point in the history
  • Loading branch information
albertzaharovits committed Jun 7, 2024
1 parent 3e0a635 commit 93d88f6
Show file tree
Hide file tree
Showing 3 changed files with 5 additions and 5 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -41,7 +41,7 @@
import java.util.concurrent.TimeUnit;
import java.util.concurrent.atomic.AtomicLong;

import static org.elasticsearch.test.tasks.MockTaskManager.USE_SPY_TASK_MANAGER_SETTING;
import static org.elasticsearch.test.tasks.MockTaskManager.SPY_TASK_MANAGER_SETTING;
import static org.hamcrest.Matchers.equalTo;
import static org.hamcrest.Matchers.instanceOf;
import static org.mockito.ArgumentMatchers.anyString;
Expand Down Expand Up @@ -75,7 +75,7 @@ private MockTransportService startTransport(String id, List<DiscoveryNode> known

public void testRemoteTaskCancellationOnFailedResponse() throws Exception {
Settings.Builder remoteTransportSettingsBuilder = Settings.builder();
remoteTransportSettingsBuilder.put(USE_SPY_TASK_MANAGER_SETTING.getKey(), true);
remoteTransportSettingsBuilder.put(SPY_TASK_MANAGER_SETTING.getKey(), true);
try (
MockTransportService remoteTransport = RemoteClusterConnectionTests.startTransport(
"seed_node",
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -39,7 +39,7 @@ public class MockTaskManager extends TaskManager {
Property.NodeScope
);

public static final Setting<Boolean> USE_SPY_TASK_MANAGER_SETTING = Setting.boolSetting(
public static final Setting<Boolean> SPY_TASK_MANAGER_SETTING = Setting.boolSetting(
"tests.spy.taskmanager.enabled",
false,
Property.NodeScope
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -103,7 +103,7 @@ public class MockTransportService extends TransportService {
public static class TestPlugin extends Plugin {
@Override
public List<Setting<?>> getSettings() {
return List.of(MockTaskManager.USE_MOCK_TASK_MANAGER_SETTING, MockTaskManager.USE_SPY_TASK_MANAGER_SETTING);
return List.of(MockTaskManager.USE_MOCK_TASK_MANAGER_SETTING, MockTaskManager.SPY_TASK_MANAGER_SETTING);
}
}

Expand Down Expand Up @@ -312,7 +312,7 @@ private static TransportAddress[] extractTransportAddresses(TransportService tra
}

public static TaskManager createTaskManager(Settings settings, ThreadPool threadPool, Set<String> taskHeaders, Tracer tracer) {
if (MockTaskManager.USE_SPY_TASK_MANAGER_SETTING.get(settings)) {
if (MockTaskManager.SPY_TASK_MANAGER_SETTING.get(settings)) {
return spy(createMockTaskManager(settings, threadPool, taskHeaders, tracer));
} else {
return createMockTaskManager(settings, threadPool, taskHeaders, tracer);
Expand Down

0 comments on commit 93d88f6

Please sign in to comment.