nizhikov commented on code in PR #11634: URL: https://github.com/apache/ignite/pull/11634#discussion_r1837830595
########## modules/core/src/test/java/org/apache/ignite/internal/processors/service/IgniteServiceDeploymentFailureTest.java: ########## @@ -0,0 +1,267 @@ +/* + * 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.internal.processors.service; + +import java.util.UUID; +import java.util.concurrent.atomic.AtomicInteger; +import java.util.function.Predicate; + +import org.apache.ignite.Ignition; +import org.apache.ignite.cluster.ClusterNode; +import org.apache.ignite.configuration.IgniteConfiguration; +import org.apache.ignite.failure.StopNodeFailureHandler; +import org.apache.ignite.internal.IgniteEx; +import org.apache.ignite.lang.IgnitePredicate; +import org.apache.ignite.services.Service; +import org.apache.ignite.services.ServiceConfiguration; +import org.apache.ignite.services.ServiceDeploymentException; +import org.apache.ignite.services.ServiceDescriptor; +import org.apache.ignite.testframework.junits.common.GridCommonAbstractTest; +import org.jetbrains.annotations.NotNull; +import org.junit.Test; + +import static org.apache.ignite.testframework.GridTestUtils.assertThrowsWithCause; +import static org.apache.ignite.testframework.GridTestUtils.waitForCondition; + +/** + * Test class for verifying the behavior of Ignite service deployment when + * specified service classes are not found or when service initialization fails. + */ +public class IgniteServiceDeploymentFailureTest extends GridCommonAbstractTest { + /** */ + private static final String NOOP_SERVICE_CLS_NAME = "org.apache.ignite.tests.p2p.NoopService"; + + /** */ + public static final String NODE_FILTER_CLS_NAME = "org.apache.ignite.tests.p2p.ExcludeNodeFilter"; + + /** */ + private static final int SERVER_NODES_CNT = 5; + + /** */ + private static final int CLIENT_NODES_CNT = 4; + + /** */ + private static final int TIMEOUT = 10_000; + + /** */ + private static ClassLoader extClsLdr; + + /** {@inheritDoc} */ + @Override protected IgniteConfiguration getConfiguration(String igniteInstanceName) throws Exception { + IgniteConfiguration cfg = super.getConfiguration(igniteInstanceName); + + cfg.setPeerClassLoadingEnabled(false); + cfg.setFailureHandler(new StopNodeFailureHandler()); + + return cfg; + } + + /** {@inheritDoc} */ + @Override protected void beforeTest() throws Exception { + super.beforeTest(); + + extClsLdr = getExternalClassLoader(); + } + + /** {@inheritDoc} */ + @Override protected void afterTest() throws Exception { + stopAllGrids(); + + super.afterTest(); + + extClsLdr = null; + } + + /** + * Tests that deploying a service with a missing class causes a ServiceDeploymentException. + * + * @throws Exception If failed. + */ + @Test + public void testFailWhenClassNotFound() throws Exception { + final String srvcName = "TestDeploymentService"; + + IgniteEx srv = startGrid(getConfiguration("server")); + IgniteEx cli = startClientGrid(1); + + ServiceConfiguration svcCfg = new ServiceConfiguration() + .setName(srvcName) + .setService(((Class<Service>)extClsLdr.loadClass(NOOP_SERVICE_CLS_NAME)).getDeclaredConstructor().newInstance()) + .setNodeFilter(((Class<IgnitePredicate<ClusterNode>>)extClsLdr.loadClass(NODE_FILTER_CLS_NAME)) + .getConstructor(UUID.class) + .newInstance(cli.configuration().getNodeId())) + .setTotalCount(1); + + assertThrowsWithCause(() -> cli.services().deploy(svcCfg), ServiceDeploymentException.class); + + assertTrue(waitForCondition(() -> noDescriptorInClusterForService(srvcName), TIMEOUT)); + } + + /** + * Tests that service descriptors are clear after an attempt of deploying a service which throws an exception + * during initialization. Dynamic configuration for services is used. + * + * @throws Exception If failed. + */ + @Test + public void testFailedServiceDescriptorsDynamicConfiguration() throws Exception { + final int noopSrvcTotalCnt = 20; + + final int noopSrvcMaxPerNodeCnt0 = 2; + final int noopSrvcTotalCnt0 = noopSrvcMaxPerNodeCnt0 * SERVER_NODES_CNT; + + final int noopSrvcMaxPerNodeCnt1 = 4; + + final int throwSrvcTotalCnt = 10; + final int throwSrvcMaxPerNodeCnt = 2; + + startGrids(SERVER_NODES_CNT); + + IgniteEx client = startClientGrid(SERVER_NODES_CNT); + + for (int i = 1; i < CLIENT_NODES_CNT; i++) + startClientGrid(SERVER_NODES_CNT + i); + + ServiceConfiguration noopSrvcCfg = new ServiceConfiguration() + .setName(NoopService.class.getSimpleName()) + .setService(new NoopService()) + .setTotalCount(noopSrvcTotalCnt) + .setMaxPerNodeCount(noopSrvcMaxPerNodeCnt0); + + // Deploying NoopService - should be completed successfully. + client.services().deploy(noopSrvcCfg); + + // Check that the expected number of instances has been deployed + assertEquals(noopSrvcTotalCnt0, + totalInstancesCount(client, NoopService.class.getSimpleName())); + + // Deploy InitThrowingService that throws an exception in init - should not be deployed. + assertThrowsWithCause(() -> client.services().deploy(new ServiceConfiguration() + .setName(InitThrowingService.class.getSimpleName()) + .setService(new InitThrowingService()) + .setTotalCount(throwSrvcTotalCnt) + .setMaxPerNodeCount(throwSrvcMaxPerNodeCnt)), + ServiceDeploymentException.class); + + // Wait until the descriptors are updated on all nodes and check that there are no descriptors of an undeployed service. + assertTrue(waitForCondition(() -> noDescriptorInClusterForService(InitThrowingService.class.getSimpleName()), TIMEOUT)); + + // Check that NoopService that is deployed successfully previously is presented in descriptors. + assertEquals(noopSrvcTotalCnt0, totalInstancesCount(client, NoopService.class.getSimpleName())); Review Comment: ```suggestion assertEquals( "Deployed service must be presented in cluster", noopSrvcMaxPerNodeCnt0 * SERVER_NODES_CNT, totalInstancesCount(client, NoopService.class.getSimpleName()) ); ``` -- This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. To unsubscribe, e-mail: notifications-unsubscr...@ignite.apache.org For queries about this service, please contact Infrastructure at: us...@infra.apache.org
