|
| 1 | +/* |
| 2 | + * SPDX-License-Identifier: Apache-2.0 |
| 3 | + * |
| 4 | + * The OpenSearch Contributors require contributions made to |
| 5 | + * this file be licensed under the Apache-2.0 license or a |
| 6 | + * compatible open source license. |
| 7 | + */ |
| 8 | + |
| 9 | +package org.opensearch.remotemigration; |
| 10 | + |
| 11 | +import org.opensearch.action.admin.cluster.repositories.get.GetRepositoriesRequest; |
| 12 | +import org.opensearch.action.admin.cluster.repositories.get.GetRepositoriesResponse; |
| 13 | +import org.opensearch.action.admin.cluster.settings.ClusterUpdateSettingsRequest; |
| 14 | +import org.opensearch.client.Client; |
| 15 | +import org.opensearch.common.settings.Settings; |
| 16 | +import org.opensearch.test.OpenSearchIntegTestCase; |
| 17 | + |
| 18 | +import java.util.List; |
| 19 | + |
| 20 | +import static org.opensearch.node.remotestore.RemoteStoreNodeService.MIGRATION_DIRECTION_SETTING; |
| 21 | +import static org.opensearch.node.remotestore.RemoteStoreNodeService.REMOTE_STORE_COMPATIBILITY_MODE_SETTING; |
| 22 | +import static org.opensearch.remotestore.RemoteStoreBaseIntegTestCase.remoteStoreClusterSettings; |
| 23 | +import static org.opensearch.test.hamcrest.OpenSearchAssertions.assertAcked; |
| 24 | + |
| 25 | +@OpenSearchIntegTestCase.ClusterScope(scope = OpenSearchIntegTestCase.Scope.TEST, numDataNodes = 0, autoManageMasterNodes = false) |
| 26 | +public class RemoteStoreMigrationTestCase extends MigrationBaseTestCase { |
| 27 | + public void testMixedModeAddRemoteNodes() throws Exception { |
| 28 | + internalCluster().setBootstrapClusterManagerNodeIndex(0); |
| 29 | + List<String> cmNodes = internalCluster().startNodes(1); |
| 30 | + Client client = internalCluster().client(cmNodes.get(0)); |
| 31 | + ClusterUpdateSettingsRequest updateSettingsRequest = new ClusterUpdateSettingsRequest(); |
| 32 | + updateSettingsRequest.persistentSettings(Settings.builder().put(REMOTE_STORE_COMPATIBILITY_MODE_SETTING.getKey(), "mixed")); |
| 33 | + assertAcked(client().admin().cluster().updateSettings(updateSettingsRequest).actionGet()); |
| 34 | + |
| 35 | + // add remote node in mixed mode cluster |
| 36 | + addRemote = true; |
| 37 | + internalCluster().startNode(); |
| 38 | + internalCluster().startNode(); |
| 39 | + internalCluster().validateClusterFormed(); |
| 40 | + |
| 41 | + // assert repo gets registered |
| 42 | + GetRepositoriesRequest gr = new GetRepositoriesRequest(new String[] { REPOSITORY_NAME }); |
| 43 | + GetRepositoriesResponse getRepositoriesResponse = client.admin().cluster().getRepositories(gr).actionGet(); |
| 44 | + assertEquals(1, getRepositoriesResponse.repositories().size()); |
| 45 | + |
| 46 | + // add docrep mode in mixed mode cluster |
| 47 | + addRemote = true; |
| 48 | + internalCluster().startNode(); |
| 49 | + assertBusy(() -> { |
| 50 | + assertEquals(client.admin().cluster().prepareClusterStats().get().getNodes().size(), internalCluster().getNodeNames().length); |
| 51 | + }); |
| 52 | + |
| 53 | + // add incompatible remote node in remote mixed cluster |
| 54 | + Settings.Builder badSettings = Settings.builder() |
| 55 | + .put(remoteStoreClusterSettings(REPOSITORY_NAME, segmentRepoPath, "REPOSITORY_2_NAME", translogRepoPath)) |
| 56 | + .put("discovery.initial_state_timeout", "500ms"); |
| 57 | + String badNode = internalCluster().startNode(badSettings); |
| 58 | + assertTrue(client.admin().cluster().prepareClusterStats().get().getNodes().size() < internalCluster().getNodeNames().length); |
| 59 | + internalCluster().stopRandomNode(settings -> settings.get("node.name").equals(badNode)); |
| 60 | + } |
| 61 | + |
| 62 | + public void testMigrationDirections() { |
| 63 | + ClusterUpdateSettingsRequest updateSettingsRequest = new ClusterUpdateSettingsRequest(); |
| 64 | + // add remote node in docrep cluster |
| 65 | + updateSettingsRequest.persistentSettings(Settings.builder().put(MIGRATION_DIRECTION_SETTING.getKey(), "docrep")); |
| 66 | + assertAcked(client().admin().cluster().updateSettings(updateSettingsRequest).actionGet()); |
| 67 | + |
| 68 | + updateSettingsRequest.persistentSettings(Settings.builder().put(MIGRATION_DIRECTION_SETTING.getKey(), "remote_store")); |
| 69 | + assertAcked(client().admin().cluster().updateSettings(updateSettingsRequest).actionGet()); |
| 70 | + |
| 71 | + updateSettingsRequest.persistentSettings(Settings.builder().put(MIGRATION_DIRECTION_SETTING.getKey(), "random")); |
| 72 | + assertThrows(IllegalArgumentException.class, () -> client().admin().cluster().updateSettings(updateSettingsRequest).actionGet()); |
| 73 | + } |
| 74 | +} |
0 commit comments