/*
 * ============LICENSE_START=======================================================
 * PNF-REGISTRATION-HANDLER
 * ================================================================================
 * Copyright (C) 2018 NOKIA Intellectual Property. 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.
 * ============LICENSE_END=========================================================
 */

package org.onap.pnfsimulator.message;

import static org.junit.jupiter.api.Assertions.assertEquals;
import static org.junit.jupiter.api.Assertions.assertThrows;
import static org.junit.jupiter.api.Assertions.assertTrue;
import static org.onap.pnfsimulator.message.MessageConstants.COMMON_EVENT_HEADER;
import static org.onap.pnfsimulator.message.MessageConstants.EVENT;
import static org.onap.pnfsimulator.message.MessageConstants.NOTIFICATION_FIELDS;
import static org.onap.pnfsimulator.message.MessageConstants.PNF_REGISTRATION_FIELDS;
import java.util.Optional;
import org.json.JSONObject;
import org.junit.jupiter.api.BeforeAll;
import org.junit.jupiter.api.Test;

public class MessageProviderTest {

    private static final String testParamsPnfRegistration =
        "{\"pnfKey1\": \"pnfVal1\",\"pnfKey2\": \"pnfVal2\",\"pnfKey3\": \"pnfVal3\",\"pnfKey4\": \"pnfVal4\"}";

    private static final String testParamsNotification =
        "{\"notKey1\": \"notVal1\",\"notKey2\": \"notVal2\",\"notKey3\": \"notVal3\",\"notKey4\": \"notVal4\"}";

    private static MessageProvider messageProvider;

    @BeforeAll
    public static void setup() {
        messageProvider = new MessageProvider();
    }

    @Test
    public void createMessage_should_throw_when_given_empty_arguments() {
        assertThrows(IllegalArgumentException.class,
            () -> messageProvider.createMessage(new JSONObject(), Optional.empty(), Optional.empty()),
            "Params object cannot be null");
    }

    @Test
    public void createMessage_should_create_constant_message_when_no_params_specified() {
        JSONObject message = messageProvider.createMessage(new JSONObject(), Optional.ofNullable(new JSONObject()),
            Optional.ofNullable(new JSONObject()));
        JSONObject event = message.getJSONObject(EVENT);

        JSONObject commonEventHeader = event.getJSONObject(COMMON_EVENT_HEADER);
        JSONObject pnfRegistrationFields = event.getJSONObject(PNF_REGISTRATION_FIELDS);
        JSONObject notificationFields = event.getJSONObject(NOTIFICATION_FIELDS);

        JSONObject expectedCommonEventHeader = JSONObjectFactory.generateConstantCommonEventHeader();
        JSONObject expectedPnfRegistrationFields = JSONObjectFactory.generatePnfRegistrationFields();
        JSONObject expectedNotificationFields = JSONObjectFactory.generateNotificationFields();

        expectedCommonEventHeader
            .toMap()
            .forEach((key, val) -> assertTrue(commonEventHeader.has(key),
                () -> String.format("Key %s is not present", key)));

        expectedPnfRegistrationFields
            .toMap()
            .forEach((key, val) -> assertTrue(pnfRegistrationFields.has(key),
                () -> String.format("Key %s is not present", key)));

        expectedNotificationFields
            .toMap()
            .forEach((key, val) -> assertTrue(notificationFields.has(key),
                () -> String.format("Key %s is not present", key)));
    }

    @Test
    public void createMessage_should_throw_exception_when_params_specified_as_empty() {
        assertThrows(IllegalArgumentException.class,
            () -> messageProvider.createMessage(new JSONObject(), Optional.empty(),
                Optional.empty()));
    }

    @Test
    public void createMessage_should_add_specified_params_to_valid_subobjects_with_event_pnf_registration() {
        JSONObject message = messageProvider
            .createMessage(new JSONObject(), Optional.of(new JSONObject(testParamsPnfRegistration)), Optional.empty());
        JSONObject event = message.getJSONObject(EVENT);

        JSONObject commonEventHeader = event.getJSONObject(COMMON_EVENT_HEADER);
        assertEquals(13, commonEventHeader.keySet().size());

        JSONObject pnfRegistrationFields = event.getJSONObject(PNF_REGISTRATION_FIELDS);
        assertEquals("pnfVal1", pnfRegistrationFields.getString("pnfKey1"));
        assertEquals("pnfVal2", pnfRegistrationFields.getString("pnfKey2"));
    }

    @Test
    public void createMessage_should_add_specified_params_to_valid_subobjects_with_event_notification() {
        JSONObject message = messageProvider
            .createMessage(new JSONObject(), Optional.empty(), Optional.of(new JSONObject(testParamsNotification)));
        JSONObject event = message.getJSONObject(EVENT);

        JSONObject commonEventHeader = event.getJSONObject(COMMON_EVENT_HEADER);
        assertEquals(12, commonEventHeader.keySet().size());

        JSONObject notificationFields = event.getJSONObject(NOTIFICATION_FIELDS);
        assertEquals("notVal1", notificationFields.getString("notKey1"));
        assertEquals("notVal2", notificationFields.getString("notKey2"));
    }

}