| /* |
| * 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.distributed.internal; |
| |
| import static org.assertj.core.api.Assertions.assertThat; |
| import static org.assertj.core.api.Assertions.assertThatCode; |
| import static org.assertj.core.api.Assertions.assertThatThrownBy; |
| import static org.mockito.Mockito.when; |
| |
| import java.io.File; |
| import java.io.IOException; |
| import java.net.InetAddress; |
| import java.nio.charset.Charset; |
| import java.nio.file.Path; |
| import java.util.Properties; |
| |
| import org.apache.commons.io.FileUtils; |
| import org.junit.After; |
| import org.junit.Before; |
| import org.junit.Ignore; |
| import org.junit.Rule; |
| import org.junit.Test; |
| import org.junit.contrib.java.lang.system.RestoreSystemProperties; |
| import org.junit.rules.TemporaryFolder; |
| import org.mockito.Mock; |
| import org.mockito.junit.MockitoJUnit; |
| import org.mockito.junit.MockitoRule; |
| import org.mockito.quality.Strictness; |
| |
| import org.apache.geode.distributed.Locator; |
| import org.apache.geode.distributed.internal.membership.gms.membership.GMSJoinLeave; |
| import org.apache.geode.internal.security.SecurableCommunicationChannel; |
| import org.apache.geode.logging.internal.LoggingSession; |
| |
| public class InternalLocatorIntegrationTest { |
| |
| @Rule |
| public RestoreSystemProperties restoreSystemProperties = new RestoreSystemProperties(); |
| |
| @Rule |
| public MockitoRule mockitoRule = MockitoJUnit.rule().strictness(Strictness.STRICT_STUBS); |
| |
| @Rule |
| public TemporaryFolder temporaryFolder = new TemporaryFolder(); |
| |
| private int port; |
| @Mock |
| private LoggingSession loggingSession; |
| private File logFile; |
| private InetAddress bindAddress; |
| private String hostnameForClients; |
| @Mock |
| private Properties distributedSystemProperties; |
| @Mock |
| private DistributionConfigImpl distributionConfig; |
| private Path workingDirectory; |
| private InternalLocator internalLocator; |
| |
| @Before |
| public void setUp() throws IOException { |
| // set a property to tell membership to create a new cluster |
| System.setProperty(GMSJoinLeave.BYPASS_DISCOVERY_PROPERTY, "true"); |
| port = 0; |
| hostnameForClients = ""; |
| bindAddress = null; |
| |
| logFile = temporaryFolder.newFile("logfile.log"); |
| workingDirectory = temporaryFolder.getRoot().toPath(); |
| |
| if (Locator.hasLocator()) { |
| Locator.getLocator().stop(); |
| } |
| } |
| |
| @After |
| public void tearDown() { |
| if (internalLocator != null) { |
| internalLocator.stop(); |
| } |
| } |
| |
| @Test |
| public void startedLocatorDoesNotAttemptReconnect() throws IOException, InterruptedException { |
| // start a locator that's not part of a cluster |
| final boolean joinCluster = false; |
| internalLocator = InternalLocator.startLocator(port, logFile, null, |
| null, bindAddress, joinCluster, |
| distributedSystemProperties, hostnameForClients, workingDirectory); |
| port = internalLocator.getPort(); |
| // the locator shouldn't attempt a reconnect because it's not part of a cluster |
| internalLocator.stoppedForReconnect = true; |
| assertThat(internalLocator.attemptReconnect()).isFalse(); |
| String output = FileUtils.readFileToString(logFile, Charset.defaultCharset()); |
| assertThat(output).isNotEmpty(); |
| assertThat(output).contains(InternalLocator.IGNORING_RECONNECT_REQUEST); |
| } |
| |
| |
| |
| @Test |
| public void constructs() { |
| when(distributionConfig.getLogFile()).thenReturn(logFile); |
| when(distributionConfig.getLocators()).thenReturn(""); |
| when(distributionConfig.getSecurableCommunicationChannels()).thenReturn( |
| new SecurableCommunicationChannel[0]); |
| |
| assertThatCode(() -> { |
| internalLocator = |
| new InternalLocator(port, loggingSession, logFile, null, null, |
| bindAddress, hostnameForClients, distributedSystemProperties, distributionConfig, |
| workingDirectory); |
| }).doesNotThrowAnyException(); |
| } |
| |
| @Test |
| public void restartingClusterConfigurationDoesNotThrowException() throws IOException { |
| internalLocator = InternalLocator.startLocator(port, logFile, null, |
| null, bindAddress, true, |
| distributedSystemProperties, hostnameForClients, workingDirectory); |
| port = internalLocator.getPort(); |
| internalLocator.stop(true, true, false); |
| assertThat(InternalLocator.getLocator()).isNull(); |
| // try starting a cluster configuration service when a reconnected locator doesn't exist |
| assertThatCode(() -> { |
| internalLocator.startClusterManagementService(); |
| }).doesNotThrowAnyException(); |
| } |
| |
| @Test |
| public void startedLocatorIsRunning() throws IOException { |
| internalLocator = InternalLocator.startLocator(port, logFile, null, |
| null, bindAddress, true, |
| distributedSystemProperties, hostnameForClients, workingDirectory); |
| port = internalLocator.getPort(); |
| |
| assertThat(internalLocator.isStopped()).isFalse(); |
| } |
| |
| @Test |
| public void startedLocatorHasLocator() throws IOException { |
| internalLocator = InternalLocator.startLocator(port, logFile, null, |
| null, bindAddress, true, |
| distributedSystemProperties, hostnameForClients, workingDirectory); |
| port = internalLocator.getPort(); |
| |
| assertThat(InternalLocator.hasLocator()).isTrue(); |
| } |
| |
| @Test |
| public void stoppedLocatorIsStopped() throws IOException { |
| internalLocator = InternalLocator.startLocator(port, null, null, |
| null, bindAddress, true, |
| distributedSystemProperties, hostnameForClients, workingDirectory); |
| port = internalLocator.getPort(); |
| |
| internalLocator.stop(); |
| |
| assertThat(internalLocator.isStopped()).isTrue(); |
| } |
| |
| @Test |
| public void stoppedLocatorDoesNotHaveLocator() throws IOException { |
| internalLocator = InternalLocator.startLocator(port, null, null, |
| null, bindAddress, true, |
| distributedSystemProperties, hostnameForClients, workingDirectory); |
| port = internalLocator.getPort(); |
| |
| internalLocator.stop(); |
| |
| assertThat(InternalLocator.hasLocator()).isFalse(); |
| } |
| |
| @Test |
| @Ignore("GEODE-7762 this test fails repeatedly in stress tests") |
| public void startLocatorFail() throws Exception { |
| Properties properties = new Properties(); |
| // use this property to induce a NPE when calling |
| // InternalLocator.startConfigurationPersistenceService |
| // so this would demonstrate that we would throw the exception when we encounter an error when |
| // calling InternalLocator.startConfigurationPersistenceService |
| properties.put("load-cluster-configuration-from-dir", "true"); |
| assertThatThrownBy(() -> InternalLocator.startLocator(port, null, null, |
| null, bindAddress, true, |
| properties, hostnameForClients, workingDirectory)).isInstanceOf(RuntimeException.class); |
| |
| assertThat(InternalLocator.hasLocator()).isFalse(); |
| } |
| } |