/*-
 * ============LICENSE_START=======================================================
 * TestSOManager
 * ================================================================================
 * Copyright (C) 2018 Ericsson. All rights reserved.
 * ================================================================================
 * Modifications Copyright (C) 2018-2019 AT&T Intellectual Property. All rights reserved.
 * Modifications Copyright (C) 2019 Nordix Foundation.
 * ================================================================================
 *
 * 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.
 * ============LICENSE_END=========================================================
 */

package org.onap.policy.so;

import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertNotNull;
import static org.junit.Assert.assertNull;
import static org.junit.Assert.assertTrue;

import java.io.IOException;
import java.net.URI;
import java.util.UUID;
import java.util.concurrent.Future;
import org.apache.http.client.methods.CloseableHttpResponse;
import org.apache.http.client.methods.HttpGet;
import org.apache.http.impl.client.CloseableHttpClient;
import org.apache.http.impl.client.HttpClients;
import org.apache.http.util.EntityUtils;
import org.glassfish.grizzly.http.server.HttpServer;
import org.glassfish.jersey.grizzly2.httpserver.GrizzlyHttpServerFactory;
import org.glassfish.jersey.server.ResourceConfig;
import org.junit.AfterClass;
import org.junit.BeforeClass;
import org.junit.Test;
import org.onap.policy.so.SoManager.SoCallback;

public class SoManagerTest implements SoCallback {
    private static final String LOCALHOST_99999999 = "http:/localhost:99999999";
    private static final String CITIZEN = "citizen";
    private static final String RETURN_ONGING202 = "ReturnOnging202";
    private static final String RETURN_ONGING200 = "ReturnOnging200";
    private static final String RETURN_FAILED = "ReturnFailed";
    private static final String RETURN_COMPLETED = "ReturnCompleted";
    private static final String RETURN_BAD_JSON = "ReturnBadJson";
    private static final String RETURN_BAD_AFTER_WAIT = "ReturnBadAfterWait";
    private static final String ONGOING = "ONGOING";
    private static final String FAILED = "FAILED";
    private static final String COMPLETE = "COMPLETE";
    private static final String DATE1 = "2018-03-23 16:31";
    private static final String SERVICE_INSTANTIATION_V7 = "/serviceInstantiation/v7";
    private static final String BASE_URI = "http://localhost:46553/TestSOManager";
    private static final String BASE_SO_URI = BASE_URI + "/SO";
    private static HttpServer server;

    /**
     * Set up test class.
     */
    @BeforeClass
    public static void setUp() throws IOException {
        final ResourceConfig rc = new ResourceConfig(SoDummyServer.class);
        //Grizzly by default doesn't allow payload for HTTP methods (ex: DELETE), for which HTTP spec doesn't
        // explicitly state that.
        //allow it before starting the server
        server = GrizzlyHttpServerFactory.createHttpServer(URI.create(BASE_URI), rc, false);
        server.getServerConfiguration().setAllowPayloadForUndefinedHttpMethods(true);
        server.start();
    }

    @AfterClass
    public static void tearDown() {
        server.shutdown();
    }

    @Test
    public void testGrizzlyServer() throws IOException {
        try (CloseableHttpClient httpclient = HttpClients.createDefault()) {
            HttpGet httpGet = new HttpGet("http://localhost:46553/TestSOManager/SO/Stats");
            CloseableHttpResponse response = httpclient.execute(httpGet);

            String returnBody = EntityUtils.toString(response.getEntity(), "UTF-8");
            assertTrue(returnBody.matches("^\\{\"GET\": [0-9]*,\"STAT\": [0-9]*,\"POST\": [0-9]*,\"PUT\": [0-9]*,"
                    + "\"DELETE\": [0-9]*\\}$"));
        }
    }

    @Test
    public void testServiceInstantiation() {
        SoManager manager = new SoManager(null, null, null);
        assertNotNull(manager);
        manager.setRestGetTimeout(100);

        SoResponse response = manager.createModuleInstance(LOCALHOST_99999999, BASE_SO_URI, "sean",
                CITIZEN, null);
        assertNull(response);

        response = manager.createModuleInstance(BASE_SO_URI + SERVICE_INSTANTIATION_V7, BASE_SO_URI, "sean",
                        CITIZEN, null);
        assertNull(response);

        response = manager.createModuleInstance(BASE_SO_URI + SERVICE_INSTANTIATION_V7, BASE_SO_URI, "sean",
                        CITIZEN, new SoRequest());
        assertNull(response);

        SoRequest request = new SoRequest();
        request.setRequestId(UUID.randomUUID());
        request.setRequestScope("Test");
        request.setRequestType(RETURN_BAD_JSON);
        request.setStartTime(DATE1);
        request.setRequestStatus(new SoRequestStatus());
        request.getRequestStatus().setRequestState(ONGOING);

        response = manager.createModuleInstance(BASE_SO_URI + SERVICE_INSTANTIATION_V7, BASE_SO_URI, "sean",
                        CITIZEN, request);
        assertNull(response);

        request.setRequestType(RETURN_COMPLETED);
        response = manager.createModuleInstance(BASE_SO_URI + SERVICE_INSTANTIATION_V7, BASE_SO_URI, "sean",
                        CITIZEN, request);
        assertNotNull(response);
        assertEquals(COMPLETE, response.getRequest().getRequestStatus().getRequestState());

        request.setRequestType(RETURN_FAILED);
        response = manager.createModuleInstance(BASE_SO_URI + SERVICE_INSTANTIATION_V7, BASE_SO_URI, "sean",
                        CITIZEN, request);
        assertNotNull(response);
        assertEquals(FAILED, response.getRequest().getRequestStatus().getRequestState());

        // Use scope to set the number of iterations we'll wait for

        request.setRequestType(RETURN_ONGING200);
        request.setRequestScope("10");
        response = manager.createModuleInstance(BASE_SO_URI + SERVICE_INSTANTIATION_V7, BASE_SO_URI, "sean",
                        CITIZEN, request);
        assertNotNull(response);
        assertNotNull(response.getRequest());
        assertEquals(COMPLETE, response.getRequest().getRequestStatus().getRequestState());

        request.setRequestType(RETURN_ONGING202);
        request.setRequestScope("20");
        response = manager.createModuleInstance(BASE_SO_URI + SERVICE_INSTANTIATION_V7, BASE_SO_URI, "sean",
                        CITIZEN, request);
        assertNotNull(response);
        assertNotNull(response.getRequest());
        assertEquals(COMPLETE, response.getRequest().getRequestStatus().getRequestState());

        // Test timeout after 20 attempts for a response
        request.setRequestType(RETURN_ONGING202);
        request.setRequestScope("21");
        response = manager.createModuleInstance(BASE_SO_URI + SERVICE_INSTANTIATION_V7, BASE_SO_URI, "sean",
                        CITIZEN, request);
        assertNull(response);

        // Test bad response after 3 attempts for a response
        request.setRequestType(RETURN_BAD_AFTER_WAIT);
        request.setRequestScope("3");
        response = manager.createModuleInstance(BASE_SO_URI + SERVICE_INSTANTIATION_V7, BASE_SO_URI, "sean",
                        CITIZEN, request);
        assertNull(response);
    }

    @Test
    public void testVfModuleCreation() throws Exception {
        SoManager manager = new SoManager(LOCALHOST_99999999, "sean", CITIZEN);
        assertNotNull(manager);
        manager.setRestGetTimeout(100);

        SoRequest soRequest = new SoRequest();
        soRequest.setOperationType(SoOperationType.SCALE_OUT);
        Future<SoResponse> asyncRestCallFuture = manager.asyncSoRestCall(UUID.randomUUID().toString(), this,
                        UUID.randomUUID().toString(), UUID.randomUUID().toString(), soRequest);
        SoResponse response = asyncRestCallFuture.get();
        assertEquals(999, response.getHttpResponseCode());

        manager = new SoManager(BASE_SO_URI, "sean", CITIZEN);
        manager.setRestGetTimeout(100);
        asyncRestCallFuture = manager.asyncSoRestCall(UUID.randomUUID().toString(), this, UUID.randomUUID().toString(),
                        UUID.randomUUID().toString(), soRequest);
        response = asyncRestCallFuture.get();
        assertEquals(999, response.getHttpResponseCode());

        SoRequest request = new SoRequest();
        request.setRequestId(UUID.randomUUID());
        request.setRequestScope("Test");
        request.setRequestType(RETURN_BAD_JSON);
        request.setStartTime(DATE1);
        request.setRequestStatus(new SoRequestStatus());
        request.getRequestStatus().setRequestState(ONGOING);
        request.setOperationType(SoOperationType.SCALE_OUT);

        asyncRestCallFuture = manager.asyncSoRestCall(UUID.randomUUID().toString(), this, UUID.randomUUID().toString(),
                        UUID.randomUUID().toString(), request);
        response = asyncRestCallFuture.get();
        assertEquals(999, response.getHttpResponseCode());

        request.setRequestType(RETURN_COMPLETED);

        asyncRestCallFuture = manager.asyncSoRestCall(UUID.randomUUID().toString(), this, UUID.randomUUID().toString(),
                        UUID.randomUUID().toString(), request);
        response = asyncRestCallFuture.get();
        assertEquals(COMPLETE, response.getRequest().getRequestStatus().getRequestState());

        request.setRequestType(RETURN_FAILED);
        asyncRestCallFuture = manager.asyncSoRestCall(UUID.randomUUID().toString(), this, UUID.randomUUID().toString(),
                        UUID.randomUUID().toString(), request);
        response = asyncRestCallFuture.get();
        assertEquals(FAILED, response.getRequest().getRequestStatus().getRequestState());

        // Use scope to set the number of iterations we'll wait for

        request.setRequestType(RETURN_ONGING200);
        request.setRequestScope("10");
        asyncRestCallFuture = manager.asyncSoRestCall(UUID.randomUUID().toString(), this, UUID.randomUUID().toString(),
                        UUID.randomUUID().toString(), request);
        response = asyncRestCallFuture.get();
        assertNotNull(response.getRequest());
        assertEquals(COMPLETE, response.getRequest().getRequestStatus().getRequestState());

        request.setRequestType(RETURN_ONGING202);
        request.setRequestScope("20");
        asyncRestCallFuture = manager.asyncSoRestCall(UUID.randomUUID().toString(), this, UUID.randomUUID().toString(),
                        UUID.randomUUID().toString(), request);
        response = asyncRestCallFuture.get();
        assertNotNull(response.getRequest());
        assertEquals(COMPLETE, response.getRequest().getRequestStatus().getRequestState());

        // Test timeout after 20 attempts for a response
        request.setRequestType(RETURN_ONGING202);
        request.setRequestScope("21");
        asyncRestCallFuture = manager.asyncSoRestCall(UUID.randomUUID().toString(), this, UUID.randomUUID().toString(),
                        UUID.randomUUID().toString(), request);
        response = asyncRestCallFuture.get();
        assertEquals(999, response.getHttpResponseCode());

        // Test bad response after 3 attempts for a response
        request.setRequestType(RETURN_BAD_AFTER_WAIT);
        request.setRequestScope("3");
        asyncRestCallFuture = manager.asyncSoRestCall(UUID.randomUUID().toString(), this, UUID.randomUUID().toString(),
                        UUID.randomUUID().toString(), request);
        response = asyncRestCallFuture.get();
        assertEquals(999, response.getHttpResponseCode());
    }

    @Test
    public void testVfModuleDeletion() throws Exception {
        SoManager manager = new SoManager(LOCALHOST_99999999, "sean", CITIZEN);
        assertNotNull(manager);
        manager.setRestGetTimeout(100);

        SoRequest soRequest = new SoRequest();
        soRequest.setOperationType(SoOperationType.DELETE_VF_MODULE);
        Future<SoResponse> asyncRestCallFuture = manager.asyncSoRestCall(UUID.randomUUID().toString(), this,
                UUID.randomUUID().toString(), UUID.randomUUID().toString(), UUID.randomUUID().toString(), soRequest);
        SoResponse response = asyncRestCallFuture.get();
        assertEquals(999, response.getHttpResponseCode());

        manager = new SoManager(BASE_SO_URI, "sean", CITIZEN);
        manager.setRestGetTimeout(100);

        asyncRestCallFuture = manager.asyncSoRestCall(UUID.randomUUID().toString(), this, UUID.randomUUID().toString(),
                UUID.randomUUID().toString(), UUID.randomUUID().toString(), soRequest);
        response = asyncRestCallFuture.get();
        assertEquals(999, response.getHttpResponseCode());

        SoRequest request = new SoRequest();
        request.setRequestId(UUID.randomUUID());
        request.setRequestScope("Test");
        request.setRequestType(RETURN_BAD_JSON);
        request.setStartTime(DATE1);
        request.setRequestStatus(new SoRequestStatus());
        request.getRequestStatus().setRequestState(ONGOING);
        request.setOperationType(SoOperationType.DELETE_VF_MODULE);

        asyncRestCallFuture = manager.asyncSoRestCall(UUID.randomUUID().toString(), this, UUID.randomUUID().toString(),
                UUID.randomUUID().toString(), UUID.randomUUID().toString(), request);
        response = asyncRestCallFuture.get();
        assertEquals(999, response.getHttpResponseCode());

        request.setRequestType(RETURN_COMPLETED);

        asyncRestCallFuture = manager.asyncSoRestCall(UUID.randomUUID().toString(), this, UUID.randomUUID().toString(),
                UUID.randomUUID().toString(), UUID.randomUUID().toString(), request);
        response = asyncRestCallFuture.get();
        assertEquals(COMPLETE, response.getRequest().getRequestStatus().getRequestState());

        request.setRequestType(RETURN_FAILED);
        asyncRestCallFuture = manager.asyncSoRestCall(UUID.randomUUID().toString(), this, UUID.randomUUID().toString(),
                UUID.randomUUID().toString(), UUID.randomUUID().toString(), request);
        response = asyncRestCallFuture.get();
        assertEquals(FAILED, response.getRequest().getRequestStatus().getRequestState());

        // Use scope to set the number of iterations we'll wait for

        request.setRequestType(RETURN_ONGING200);
        request.setRequestScope("10");
        asyncRestCallFuture = manager.asyncSoRestCall(UUID.randomUUID().toString(), this, UUID.randomUUID().toString(),
                UUID.randomUUID().toString(), UUID.randomUUID().toString(), request);
        response = asyncRestCallFuture.get();
        assertNotNull(response.getRequest());
        assertEquals(COMPLETE, response.getRequest().getRequestStatus().getRequestState());

        request.setRequestType(RETURN_ONGING202);
        request.setRequestScope("20");
        asyncRestCallFuture = manager.asyncSoRestCall(UUID.randomUUID().toString(), this, UUID.randomUUID().toString(),
                UUID.randomUUID().toString(), UUID.randomUUID().toString(), request);
        response = asyncRestCallFuture.get();
        assertNotNull(response.getRequest());
        assertEquals(COMPLETE, response.getRequest().getRequestStatus().getRequestState());

        // Test timeout after 20 attempts for a response
        request.setRequestType(RETURN_ONGING202);
        request.setRequestScope("21");
        asyncRestCallFuture = manager.asyncSoRestCall(UUID.randomUUID().toString(), this, UUID.randomUUID().toString(),
                UUID.randomUUID().toString(), UUID.randomUUID().toString(), request);
        response = asyncRestCallFuture.get();
        assertEquals(999, response.getHttpResponseCode());

        // Test bad response after 3 attempts for a response
        request.setRequestType(RETURN_BAD_AFTER_WAIT);
        request.setRequestScope("3");
        asyncRestCallFuture = manager.asyncSoRestCall(UUID.randomUUID().toString(), this, UUID.randomUUID().toString(),
                UUID.randomUUID().toString(), UUID.randomUUID().toString(), request);
        response = asyncRestCallFuture.get();
        assertEquals(999, response.getHttpResponseCode());
    }

    @Override
    public void onSoResponseWrapper(SoResponseWrapper wrapper) {
        //
        // Nothing really needed to do
        //
    }
}