| /* |
| * 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.ignite.client; |
| |
| import static org.apache.ignite.internal.testframework.IgniteTestUtils.assertThrowsWithCause; |
| import static org.apache.ignite.internal.util.IgniteUtils.closeAll; |
| import static org.apache.ignite.lang.ErrorGroups.Client.CLUSTER_ID_MISMATCH_ERR; |
| import static org.hamcrest.MatcherAssert.assertThat; |
| import static org.junit.jupiter.api.Assertions.assertEquals; |
| import static org.junit.jupiter.api.Assertions.assertTrue; |
| |
| import java.util.UUID; |
| import org.apache.ignite.client.IgniteClient.Builder; |
| import org.apache.ignite.client.fakes.FakeIgnite; |
| import org.apache.ignite.internal.testframework.BaseIgniteAbstractTest; |
| import org.apache.ignite.internal.testframework.IgniteTestUtils; |
| import org.hamcrest.CoreMatchers; |
| import org.jetbrains.annotations.Nullable; |
| import org.junit.jupiter.api.AfterEach; |
| import org.junit.jupiter.api.BeforeEach; |
| import org.junit.jupiter.api.Test; |
| |
| /** |
| * Tests client behavior with multiple clusters. |
| */ |
| public class MultiClusterTest extends BaseIgniteAbstractTest { |
| private static final UUID clusterId1 = UUID.randomUUID(); |
| |
| private static final UUID clusterId2 = UUID.randomUUID(); |
| |
| private TestServer server1; |
| |
| private TestServer server2; |
| |
| @BeforeEach |
| void setUp() { |
| server1 = new TestServer(0, new FakeIgnite(), null, null, "s1", clusterId1, null, null); |
| server2 = new TestServer(0, new FakeIgnite(), null, null, "s2", clusterId2, null, null); |
| } |
| |
| @AfterEach |
| void tearDown() throws Exception { |
| closeAll(server1, server2); |
| } |
| |
| @Test |
| public void testClientDropsConnectionOnClusterIdMismatch() |
| throws Exception { |
| TestLoggerFactory loggerFactory = new TestLoggerFactory("client"); |
| |
| Builder builder = IgniteClient.builder() |
| .addresses("127.0.0.1:" + server1.port(), "127.0.0.1:" + server2.port()) |
| .loggerFactory(loggerFactory); |
| |
| try (var client = builder.build()) { |
| assertTrue(IgniteTestUtils.waitForCondition(() -> getFailedConnectionEntry(loggerFactory) != null, 3000)); |
| |
| assertEquals(1, client.connections().size()); |
| |
| String err = getFailedConnectionEntry(loggerFactory); |
| assertThat(err, CoreMatchers.containsString("Cluster ID mismatch")); |
| } |
| } |
| |
| @Test |
| public void testReconnectToDifferentClusterFails() |
| throws Exception { |
| int port = server1.port(); |
| Builder builder = IgniteClient.builder() |
| .addresses("127.0.0.1:" + port); |
| |
| server2.close(); |
| |
| try (var client = builder.build()) { |
| client.tables().tables(); |
| |
| server1.close(); |
| server1 = new TestServer(0, new FakeIgnite(), null, null, "s1", clusterId2, null, port); |
| |
| IgniteClientConnectionException ex = (IgniteClientConnectionException) assertThrowsWithCause( |
| () -> client.tables().tables(), IgniteClientConnectionException.class, "Cluster ID mismatch"); |
| |
| assertEquals(CLUSTER_ID_MISMATCH_ERR, ex.code()); |
| } |
| } |
| |
| private static @Nullable String getFailedConnectionEntry(TestLoggerFactory loggerFactory) { |
| return loggerFactory.logger.entries().stream() |
| .filter(x -> x.contains("Failed to establish connection to")) |
| .findFirst() |
| .orElse(null); |
| } |
| } |