geode-gfsh/src/test/java/org/apache/geode/management/internal/cli/commands/AlterAsyncEventQueueCommandTest.java

/*
 * 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.internal.cli.commands;

import static org.mockito.ArgumentMatchers.any;
import static org.mockito.Mockito.doCallRealMethod;
import static org.mockito.Mockito.doNothing;
import static org.mockito.Mockito.doReturn;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.spy;

import java.util.HashSet;
import java.util.Set;

import org.junit.Before;
import org.junit.ClassRule;
import org.junit.Test;

import org.apache.geode.cache.Region;
import org.apache.geode.cache.configuration.CacheConfig;
import org.apache.geode.distributed.internal.InternalConfigurationPersistenceService;
import org.apache.geode.internal.cache.AbstractRegion;
import org.apache.geode.internal.config.JAXBService;
import org.apache.geode.test.junit.rules.GfshParserRule;

public class AlterAsyncEventQueueCommandTest {

  @ClassRule
  public static GfshParserRule gfsh = new GfshParserRule();

  private AlterAsyncEventQueueCommand command;
  private InternalConfigurationPersistenceService service;
  private Region configRegion;
  private Set<String> groupSet = new HashSet<>();

  @Before
  public void setUp() {
    command = spy(AlterAsyncEventQueueCommand.class);
    service =
        spy(new InternalConfigurationPersistenceService(JAXBService.create(CacheConfig.class)));
    configRegion = mock(AbstractRegion.class);

    doReturn(service).when(command).getConfigurationPersistenceService();

    groupSet.add("group1");
    groupSet.add("group2");
    doReturn(groupSet).when(service).getGroups();

    CacheConfig config = new CacheConfig();
    CacheConfig.AsyncEventQueue aeq1 = new CacheConfig.AsyncEventQueue();
    aeq1.setId("queue1");

    config.getAsyncEventQueues().add(aeq1);
    doReturn(config).when(service).getCacheConfig("group1");
    doReturn(new CacheConfig()).when(service).getCacheConfig("group2");
    doReturn(config).when(service).getCacheConfig("group1", true);
    doReturn(new CacheConfig()).when(service).getCacheConfig("group2", true);

    doReturn(true).when(service).lockSharedConfiguration();
    doNothing().when(service).unlockSharedConfiguration();
    doReturn(null).when(service).getConfiguration(any());
    doReturn(configRegion).when(service).getConfigurationRegion();
    doCallRealMethod().when(service).updateCacheConfig(any(), any());
  }

  @Test
  public void mandatoryOption() {
    gfsh.executeAndAssertThat(command, "alter async-event-queue").statusIsError()
        .containsOutput("Invalid command");
  }

  @Test
  public void noOptionToModify() {
    gfsh.executeAndAssertThat(command, "alter async-event-queue --id=test").statusIsError()
        .containsOutput("need to specify at least one option to modify.");
  }

  @Test
  public void emptyConfiguration() {
    gfsh.executeAndAssertThat(command, "alter async-event-queue --id=test --batch-size=100")
        .statusIsError().containsOutput("Can not find an async event queue");

  }

  @Test
  public void emptyConfiguration_ifExists() {
    gfsh.executeAndAssertThat(command,
        "alter async-event-queue --id=test --batch-size=100 --if-exists").statusIsSuccess()
        .containsOutput("Skipping: Can not find an async event queue with id");

  }

  @Test
  public void cluster_config_service_not_available() {
    doReturn(null).when(command).getConfigurationPersistenceService();
    gfsh.executeAndAssertThat(command, "alter async-event-queue --id=test --batch-size=100")
        .statusIsError().containsOutput("Cluster Configuration Service is not available");
  }

  @Test
  public void queueIdNotFoundInTheMap() {
    gfsh.executeAndAssertThat(command,
        "alter async-event-queue --batch-size=100 --id=queue")
        .statusIsError().containsOutput("Can not find an async event queue");

  }

  @Test
  public void queueIdFoundInTheMap_updateBatchSize() {
    gfsh.executeAndAssertThat(command, "alter async-event-queue --batch-size=100 --id=queue1")
        .statusIsSuccess()
        .containsOutput("Cluster configuration for group 'group1' is updated")
        .containsOutput("Cluster configuration for group 'group2' is not updated")
        .containsOutput("Please restart the servers to apply any changed configuration");
  }

  @Test
  public void queueIdFoundInTheMap_updateBatchTimeInterval() {
    gfsh.executeAndAssertThat(command,
        "alter async-event-queue --batch-time-interval=100 --id=queue1")
        .statusIsSuccess()
        .containsOutput("Cluster configuration for group 'group1' is updated")
        .containsOutput("Cluster configuration for group 'group2' is not updated")
        .containsOutput("Please restart the servers to apply any changed configuration");

    gfsh.executeAndAssertThat(command,
        "alter async-event-queue --batch-time-interval=100 --id=queue1").statusIsSuccess()
        .containsOutput("Please restart the servers");
  }

  @Test
  public void queueIdFoundInTheMap_updateMaxMemory() {
    gfsh.executeAndAssertThat(command, "alter async-event-queue --max-queue-memory=100 --id=queue1")
        .statusIsSuccess()
        .containsOutput("Cluster configuration for group 'group1' is updated")
        .containsOutput("Cluster configuration for group 'group2' is not updated")
        .containsOutput("Please restart the servers to apply any changed configuration");
  }
}