| /* |
| * Licensed to the Apache Software Foundation (ASF) under one or more contributor license |
| * agreements. See the NOTICE file distributed with this work for additional information regarding |
| * copyright ownership. The ASF licenses this file to You under the Apache License, Version 2.0 (the |
| * "License"); you may not use this file except in compliance with the License. You may obtain a |
| * copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, software distributed under the License |
| * is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express |
| * or implied. See the License for the specific language governing permissions and limitations under |
| * the License. |
| */ |
| |
| package org.apache.geode.management.client; |
| |
| import static org.apache.geode.test.junit.assertions.ClusterManagementListResultAssert.assertManagementListResult; |
| import static org.apache.geode.test.junit.assertions.ClusterManagementRealizationResultAssert.assertManagementResult; |
| import static org.assertj.core.api.Assertions.assertThat; |
| import static org.assertj.core.api.Assertions.assertThatThrownBy; |
| |
| import java.util.List; |
| |
| import org.junit.After; |
| import org.junit.Before; |
| import org.junit.Rule; |
| import org.junit.Test; |
| import org.junit.runner.RunWith; |
| import org.springframework.beans.factory.annotation.Autowired; |
| import org.springframework.security.test.context.support.WithMockUser; |
| import org.springframework.test.annotation.DirtiesContext; |
| import org.springframework.test.context.ContextConfiguration; |
| import org.springframework.test.context.junit4.SpringRunner; |
| import org.springframework.test.context.web.WebAppConfiguration; |
| import org.springframework.web.client.RestTemplate; |
| import org.springframework.web.context.WebApplicationContext; |
| |
| import org.apache.geode.management.api.ClusterManagementListResult; |
| import org.apache.geode.management.api.ClusterManagementRealizationResult; |
| import org.apache.geode.management.api.ClusterManagementResult; |
| import org.apache.geode.management.api.ClusterManagementService; |
| import org.apache.geode.management.api.RealizationResult; |
| import org.apache.geode.management.api.RestTemplateClusterManagementServiceTransport; |
| import org.apache.geode.management.cluster.client.ClusterManagementServiceBuilder; |
| import org.apache.geode.management.configuration.Region; |
| import org.apache.geode.management.configuration.RegionType; |
| import org.apache.geode.management.internal.rest.LocatorWebContext; |
| import org.apache.geode.management.internal.rest.PlainLocatorContextLoader; |
| import org.apache.geode.management.runtime.RuntimeRegionInfo; |
| import org.apache.geode.test.dunit.rules.ClusterStartupRule; |
| |
| @RunWith(SpringRunner.class) |
| @ContextConfiguration(locations = {"classpath*:WEB-INF/management-servlet.xml"}, |
| loader = PlainLocatorContextLoader.class) |
| @WebAppConfiguration |
| @DirtiesContext(classMode = DirtiesContext.ClassMode.AFTER_CLASS) |
| public class ClientClusterManagementServiceDUnitTest { |
| |
| @Autowired |
| private WebApplicationContext webApplicationContext; |
| |
| @Rule |
| public ClusterStartupRule cluster = new ClusterStartupRule(3); |
| |
| private ClusterManagementService client; |
| |
| @Before |
| public void before() { |
| cluster.setSkipLocalDistributedSystemCleanup(true); |
| LocatorWebContext webContext = new LocatorWebContext(webApplicationContext); |
| |
| client = new ClusterManagementServiceBuilder().setTransport( |
| new RestTemplateClusterManagementServiceTransport( |
| new RestTemplate(webContext.getRequestFactory()))) |
| .build(); |
| |
| cluster.startServerVM(1, "group1", webContext.getLocator().getPort()); |
| cluster.startServerVM(2, "group2", webContext.getLocator().getPort()); |
| cluster.startServerVM(3, "group1,group2", webContext.getLocator().getPort()); |
| } |
| |
| @After |
| public void deleteAllRegions() { |
| List<Region> regions = client.list(new Region()) |
| .getConfigResult(); |
| |
| regions.forEach(r -> { |
| r.setGroup(null); |
| client.delete(r); |
| }); |
| |
| List<Region> moreRegions = client.list(new Region()) |
| .getConfigResult(); |
| assertThat(moreRegions).isEmpty(); |
| } |
| |
| @Test |
| @WithMockUser |
| public void createAndDeleteRegion() { |
| Region region = new Region(); |
| region.setName("customer"); |
| region.setType(RegionType.REPLICATE); |
| |
| ClusterManagementRealizationResult createResult = client.create(region); |
| assertManagementResult(createResult).hasStatusCode(ClusterManagementResult.StatusCode.OK); |
| |
| ClusterManagementRealizationResult deleteResult = client.delete(region); |
| assertManagementResult(deleteResult) |
| .hasStatusCode(ClusterManagementResult.StatusCode.OK); |
| |
| ClusterManagementListResult<Region, RuntimeRegionInfo> listResult = |
| client.list(new Region()); |
| assertManagementListResult(listResult) |
| .hasStatusCode(ClusterManagementResult.StatusCode.OK) |
| .hasConfigurations() |
| .isEmpty(); |
| } |
| |
| @Test |
| @WithMockUser |
| public void createAndDeleteRegionOnGroup() { |
| Region region = new Region(); |
| region.setName("customer"); |
| region.setGroup("group1"); |
| region.setType(RegionType.REPLICATE); |
| |
| ClusterManagementRealizationResult createResult = client.create(region); |
| assertManagementResult(createResult).hasStatusCode(ClusterManagementResult.StatusCode.OK); |
| |
| region.setGroup(null); |
| ClusterManagementRealizationResult deleteResult = client.delete(region); |
| assertManagementResult(deleteResult) |
| .hasStatusCode(ClusterManagementResult.StatusCode.OK); |
| |
| ClusterManagementListResult<Region, RuntimeRegionInfo> listResult = |
| client.list(new Region()); |
| assertManagementListResult(listResult) |
| .hasStatusCode(ClusterManagementResult.StatusCode.OK) |
| .hasConfigurations() |
| .isEmpty(); |
| } |
| |
| @Test |
| @WithMockUser |
| public void createSameRegionOnDifferentGroup() { |
| Region region = new Region(); |
| region.setName("test"); |
| region.setGroup("group1"); |
| region.setType(RegionType.REPLICATE); |
| |
| ClusterManagementRealizationResult result = client.create(region); |
| assertManagementResult(result).isSuccessful().hasMemberStatus() |
| .extracting(RealizationResult::getMemberName) |
| .containsExactlyInAnyOrder("server-1", |
| "server-3"); |
| |
| // creating the same region on group2 will not be successful because they have a common member |
| region.setGroup("group2"); |
| assertThatThrownBy(() -> client.create(region)) |
| .hasMessageContaining( |
| "ENTITY_EXISTS: Region 'test' already exists on member(s) server-3."); |
| } |
| |
| @Test |
| @WithMockUser |
| public void deleteRegionCreatedOnMultipleGroups() { |
| Region region = new Region(); |
| region.setName("test"); |
| region.setGroup("group1"); |
| region.setType(RegionType.REPLICATE); |
| |
| ClusterManagementRealizationResult result = client.create(region); |
| assertManagementResult(result).isSuccessful().hasMemberStatus() |
| .extracting(RealizationResult::getMemberName) |
| .containsExactlyInAnyOrder("server-1", |
| "server-3"); |
| |
| // creating the same region on group2 will not be successful because they have a common member |
| region.setGroup("group2"); |
| assertThatThrownBy(() -> client.create(region)).hasMessageContaining("ENTITY_EXISTS"); |
| |
| region.setGroup(null); |
| ClusterManagementRealizationResult deleteResult = client.delete(region); |
| |
| assertManagementResult(deleteResult).isSuccessful(); |
| |
| List<RuntimeRegionInfo> listResult = client.list(new Region()) |
| .getRuntimeResult(); |
| |
| assertThat(listResult).hasSize(0); |
| } |
| |
| |
| @Test |
| @WithMockUser |
| public void deleteRegionCreatedOnPartialGroups() { |
| Region region = new Region(); |
| region.setName("test"); |
| region.setGroup("group1"); |
| region.setType(RegionType.REPLICATE); |
| |
| ClusterManagementRealizationResult result = client.create(region); |
| assertManagementResult(result).isSuccessful().hasMemberStatus() |
| .extracting(RealizationResult::getMemberName) |
| .containsExactlyInAnyOrder("server-1", "server-3"); |
| |
| assertThatThrownBy(() -> client.delete(region)) |
| .hasMessageContaining("ILLEGAL_ARGUMENT: Group is an invalid option when deleting region"); |
| } |
| |
| @Test |
| @WithMockUser |
| public void deleteUnknownRegion() { |
| Region region = new Region(); |
| region.setName("unknown"); |
| |
| assertThatThrownBy(() -> client.delete(region)).hasMessageContaining("ENTITY_NOT_FOUND"); |
| } |
| |
| @Test |
| @WithMockUser |
| public void deleteRegionOnSpecificGroup() { |
| Region region = new Region(); |
| region.setName("region1"); |
| region.setGroup("group1"); |
| region.setType(RegionType.REPLICATE); |
| |
| ClusterManagementRealizationResult result = client.create(region); |
| assertManagementResult(result).isSuccessful().hasMemberStatus() |
| .extracting(RealizationResult::getMemberName) |
| .containsExactlyInAnyOrder("server-1", |
| "server-3"); |
| |
| region.setName("region2"); |
| region.setGroup("group2"); |
| result = client.create(region); |
| assertManagementResult(result).isSuccessful().hasMemberStatus() |
| .extracting(RealizationResult::getMemberName) |
| .containsExactlyInAnyOrder("server-2", |
| "server-3"); |
| |
| region.setName("region2"); |
| region.setGroup(null); |
| result = client.delete(region); |
| assertManagementResult(result) |
| .hasStatusCode(ClusterManagementResult.StatusCode.OK); |
| |
| ClusterManagementListResult<Region, RuntimeRegionInfo> listResult = |
| client.list(new Region()); |
| assertManagementListResult(listResult) |
| .hasStatusCode(ClusterManagementResult.StatusCode.OK) |
| .hasConfigurations() |
| .extracting(Region::getId) |
| .containsExactly("region1"); |
| } |
| |
| @Test |
| @WithMockUser |
| public void sanityCheck() { |
| assertThat(client.isConnected()).isTrue(); |
| } |
| } |