modules/core/src/test/java/org/apache/ignite/platform/PlatformServiceCallThinTask.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.ignite.platform;

import org.apache.ignite.client.ClientException;
import org.apache.ignite.client.IgniteClient;
import org.apache.ignite.compute.ComputeJobAdapter;
import org.apache.ignite.internal.processors.platform.services.PlatformService;
import org.apache.ignite.internal.util.typedef.internal.U;
import org.apache.ignite.services.ServiceCallContext;
import org.apache.ignite.testframework.GridTestUtils;

/**
 * Basic task to calling {@link PlatformService} from java thin client.
 */
public class PlatformServiceCallThinTask extends AbstractPlatformServiceCallTask {
    /** {@inheritDoc} */
    @Override ComputeJobAdapter createJob(String svcName) {
        return new PlatformServiceCallThinJob(svcName);
    }

    /** */
    static class PlatformServiceCallThinJob extends PlatformServiceCallTask.PlatformServiceCallJob {
        /** Thin client. */
        IgniteClient client;

        /**
         * @param srvcName Service name.
         */
        PlatformServiceCallThinJob(String srvcName) {
            super(srvcName);
        }

        /** {@inheritDoc} */
        @Override TestPlatformService serviceProxy() {
            ServiceCallContext callCtx = ServiceCallContext.builder().put("attr", "value").build();

            return client.services().serviceProxy(srvcName, TestPlatformService.class, callCtx);
        }

        /** {@inheritDoc} */
        @Override void runTest() {
            client = startClient();

            try {
                super.runTest();
            }
            finally {
                U.close(client, ignite.log().getLogger(getClass()));
            }
        }

        /** */
        @Override protected void checkErrorMethod(TestPlatformService srv) {
            // For thin client only top level error message is passed from server to client, so we should override
            // method for error check.
            GridTestUtils.assertThrowsAnyCause(null, () -> {
                srv.errorMethod();
                return null;
            }, ClientException.class, "Failed to invoke platform service");
        }
    }
}