path: root/plugins/plugins-event/plugins-event-carrier/plugins-event-carrier-restrequestor/src/test/java/org/onap/policy/apex/plugins/event/carrier/restrequestor/RestRequestorTest.java

/*-
 * ============LICENSE_START=======================================================
 *  Copyright (C) 2016-2018 Ericsson. All rights reserved.
 *  Modifications Copyright (C) 2019-2020,2023-2024 Nordix Foundation.
 *  Modifications Copyright (C) 2021 Bell Canada. All rights reserved.
 * ================================================================================
 * Licensed 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.
 *
 * SPDX-License-Identifier: Apache-2.0
 * ============LICENSE_END=========================================================
 */

package org.onap.policy.apex.plugins.event.carrier.restrequestor;

import static org.assertj.core.api.Assertions.assertThat;
import static org.awaitility.Awaitility.await;
import static org.junit.Assert.assertEquals;

import com.google.gson.Gson;
import jakarta.ws.rs.client.Client;
import jakarta.ws.rs.client.ClientBuilder;
import jakarta.ws.rs.core.Response;
import java.io.ByteArrayOutputStream;
import java.io.PrintStream;
import java.util.Map;
import java.util.concurrent.TimeUnit;
import org.junit.After;
import org.junit.AfterClass;
import org.junit.Before;
import org.junit.BeforeClass;
import org.junit.Test;
import org.onap.policy.apex.core.infrastructure.messaging.MessagingException;
import org.onap.policy.apex.model.basicmodel.concepts.ApexException;
import org.onap.policy.apex.service.engine.main.ApexMain;
import org.onap.policy.common.endpoints.http.server.HttpServletServer;
import org.onap.policy.common.endpoints.http.server.HttpServletServerFactoryInstance;
import org.onap.policy.common.gson.GsonMessageBodyHandler;
import org.onap.policy.common.utils.network.NetworkUtil;

/**
 * The Class TestRestRequestor.
 */
public class RestRequestorTest {
    private static final int PORT = 32801;
    private static HttpServletServer server;

    private ByteArrayOutputStream outContent = new ByteArrayOutputStream();
    private ByteArrayOutputStream errContent = new ByteArrayOutputStream();

    private final PrintStream stdout = System.out;
    private final PrintStream stderr = System.err;

    /**
     * Sets the up.
     *
     * @throws Exception the exception
     */
    @BeforeClass
    public static void setUp() throws Exception {
        server = HttpServletServerFactoryInstance.getServerFactory().build(null, false, null, PORT, false,
            "/TestRESTRequestor", false, false);

        server.addServletClass(null, SupportRestRequestorEndpoint.class.getName());
        server.setSerializationProvider(GsonMessageBodyHandler.class.getName());

        server.start();

        if (!NetworkUtil.isTcpPortOpen("localHost", PORT, 60, 500L)) {
            throw new IllegalStateException("port " + PORT + " is still not in use");
        }
    }

    /**
     * Tear down.
     *
     */
    @AfterClass
    public static void tearDown() {
        if (server != null) {
            server.stop();
        }
    }

    /**
     * Before test.
     */
    @Before
    public void beforeTest() {
        SupportRestRequestorEndpoint.resetCounters();
        System.setOut(new PrintStream(outContent));
        System.setErr(new PrintStream(errContent));
    }

    /**
     * After test.
     */
    @After
    public void afterTest() {
        System.setOut(stdout);
        System.setErr(stderr);
    }

    /**
     * Test rest requestor get.
     *
     * @throws MessagingException the messaging exception
     * @throws Exception an exception
     */
    @Test
    public void testRestRequestorGet() throws Exception {
        final Client client = ClientBuilder.newClient();

        final String[] args = {"src/test/resources/prodcons/File2RESTRequest2FileGet.json"};
        final ApexMain apexMain = new ApexMain(args);
        await().atMost(2, TimeUnit.SECONDS).until(apexMain::isAlive);

        await().pollInterval(300, TimeUnit.MILLISECONDS).atMost(10, TimeUnit.SECONDS)
            .until(() -> getStatsFromServer(client, "GET") >= 50.0);

        apexMain.shutdown();
        await().atMost(2, TimeUnit.SECONDS).until(() -> !apexMain.isAlive());

        client.close();
    }

    /**
     * Test rest requestor get empty.
     *
     * @throws ApexException the apex exception
     */
    @Test
    public void testRestRequestorGetEmpty() throws ApexException {
        final Client client = ClientBuilder.newClient();

        final String[] args = {"src/test/resources/prodcons/File2RESTRequest2FileGetEmpty.json"};
        final ApexMain apexMain = new ApexMain(args);
        await().atMost(2, TimeUnit.SECONDS).until(apexMain::isAlive);

        Response response = null;

        // Wait for the required amount of events to be received or for 10 seconds
        double getsSoFar = 0.0;
        for (int i = 0; i < 40; i++) {
            response = client.target("http://localhost:32801/TestRESTRequestor/apex/event/Stats")
                .request("application/json").get();

            if (Response.Status.OK.getStatusCode() != response.getStatus()) {
                break;
            }

            final String responseString = response.readEntity(String.class);

            @SuppressWarnings("unchecked")
            final Map<String, Object> jsonMap = new Gson().fromJson(responseString, Map.class);
            getsSoFar = Double.parseDouble(jsonMap.get("GET").toString());

            if (getsSoFar >= 50.0) {
                break;
            }
        }

        apexMain.shutdown();
        await().atMost(2, TimeUnit.SECONDS).until(() -> !apexMain.isAlive());

        client.close();

        assertEquals(Response.Status.OK.getStatusCode(), response.getStatus());
    }

    /**
     * Test REST requestor put.
     *
     * @throws ApexException the apex exception
     */
    @Test
    public void testRestRequestorPut() throws ApexException {
        final Client client = ClientBuilder.newClient();

        final String[] args = {"src/test/resources/prodcons/File2RESTRequest2FilePut.json"};
        final ApexMain apexMain = new ApexMain(args);
        await().atMost(2, TimeUnit.SECONDS).until(apexMain::isAlive);

        await().pollInterval(300, TimeUnit.MILLISECONDS).atMost(10, TimeUnit.SECONDS)
            .until(() -> getStatsFromServer(client, "PUT") >= 50.0);

        apexMain.shutdown();
        await().atMost(2, TimeUnit.SECONDS).until(() -> !apexMain.isAlive());

        client.close();
    }

    /**
     * Test REST requestor post.
     *
     * @throws ApexException the apex exception
     */
    @Test
    public void testRestRequestorPost() throws ApexException {
        final Client client = ClientBuilder.newClient();

        final String[] args = {"src/test/resources/prodcons/File2RESTRequest2FilePost.json"};
        final ApexMain apexMain = new ApexMain(args);
        await().atMost(2, TimeUnit.SECONDS).until(apexMain::isAlive);

        await().pollInterval(300, TimeUnit.MILLISECONDS).atMost(10, TimeUnit.SECONDS)
            .until(() -> getStatsFromServer(client, "POST") >= 50.0);

        apexMain.shutdown();
        await().atMost(2, TimeUnit.SECONDS).until(() -> !apexMain.isAlive());

        client.close();
    }

    /**
     * Test REST requestor delete.
     *
     * @throws ApexException the apex exception
     */
    @Test
    public void testRestRequestorDelete() throws ApexException {
        final Client client = ClientBuilder.newClient();

        final String[] args = {"src/test/resources/prodcons/File2RESTRequest2FileDelete.json"};
        final ApexMain apexMain = new ApexMain(args);
        await().atMost(2, TimeUnit.SECONDS).until(apexMain::isAlive);

        // Wait for the required amount of events to be received
        await().pollInterval(300, TimeUnit.MILLISECONDS).atMost(10, TimeUnit.SECONDS)
            .until(() -> getStatsFromServer(client, "DELETE") >= 50.0);

        apexMain.shutdown();
        await().atMost(2, TimeUnit.SECONDS).until(() -> !apexMain.isAlive());

        client.close();
    }

    /**
     * Test REST requestor multi inputs.
     *
     * @throws ApexException the apex exception
     */
    @Test
    public void testRestRequestorMultiInputs() throws ApexException {
        final Client client = ClientBuilder.newClient();

        final String[] args = {"src/test/resources/prodcons/File2RESTRequest2FileGetMulti.json"};
        final ApexMain apexMain = new ApexMain(args);
        await().atMost(10, TimeUnit.SECONDS).until(apexMain::isAlive);

        await().pollInterval(300, TimeUnit.MILLISECONDS).atMost(10, TimeUnit.SECONDS)
            .until(() -> getStatsFromServer(client, "GET") >= 8.0);

        apexMain.shutdown();
        await().atMost(2, TimeUnit.SECONDS).until(() -> !apexMain.isAlive());

        client.close();
    }

    /**
     * Test REST requestor producer alone.
     *
     * @throws ApexException the apex exception
     */
    @Test
    public void testRestRequestorProducerAlone() throws ApexException {

        final String[] args = {"src/test/resources/prodcons/File2RESTRequest2FileGetProducerAlone.json"};

        ApexMain apexMain = new ApexMain(args);
        apexMain.shutdown();

        final String outString = outContent.toString();

        assertThat(outString).contains("REST Requestor producer (RestRequestorProducer) "
            + "must run in peered requestor mode with a REST Requestor consumer");
    }

    /**
     * Test REST requestor consumer alone.
     *
     * @throws ApexException the apex exception
     */
    @Test
    public void testRestRequestorConsumerAlone() throws ApexException {
        final String[] args = {"src/test/resources/prodcons/File2RESTRequest2FileGetConsumerAlone.json"};
        ApexMain apexMain = new ApexMain(args);
        apexMain.shutdown();
        final String outString = outContent.toString();
        assertThat(outString).contains("peer \"RestRequestorProducer for peered mode REQUESTOR "
            + "does not exist or is not defined with the same peered mode");
    }

    private double getStatsFromServer(final Client client, final String statToGet) {
        final Response response = client.target("http://localhost:32801/TestRESTRequestor/apex/event/Stats")
            .request("application/json").get();

        assertEquals(Response.Status.OK.getStatusCode(), response.getStatus());
        final String responseString = response.readEntity(String.class);

        @SuppressWarnings("unchecked")
        final Map<String, Object> jsonMap = new Gson().fromJson(responseString, Map.class);
        return Double.parseDouble(jsonMap.get(statToGet).toString());
    }
}