/*
 * ===============================LICENSE_START======================================
 *  dcae-analytics
 * ================================================================================
 *    Copyright © 2017 AT&T 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.openecomp.dcae.apod.analytics.dmaap;

import org.openecomp.dcae.apod.analytics.dmaap.domain.config.DMaaPMRPublisherConfig;
import org.openecomp.dcae.apod.analytics.dmaap.domain.config.DMaaPMRSubscriberConfig;
import org.openecomp.dcae.apod.analytics.test.BaseDCAEAnalyticsUnitTest;

import java.util.List;

import static com.google.common.collect.ImmutableList.of;

/**
 * @author Rajiv Singla . Creation Date: 10/14/2016.
 */
public abstract class BaseAnalyticsDMaaPUnitTest extends BaseDCAEAnalyticsUnitTest {

    // Unit Test Settings
    protected static final String HOST_NAME = "testHostName";
    protected static final Integer PORT_NUMBER = 8080;
    protected static final String TOPIC_NAME = "testTopicName";
    protected static final String USERNAME = "testUserName";
    protected static final String PASSWORD = "testPassword";
    protected static final String HTTP_PROTOCOL = "https";
    protected static final String CONTENT_TYPE = "application/json";

    protected static final int PUBLISHER_MAX_BATCH_QUEUE_SIZE = 200;
    protected static final int PUBLISHER_MAX_RECOVERY_QUEUE_SIZE = 2000;

    protected static final String SUBSCRIBER_CONSUMER_ID = "123";
    protected static final String SUBSCRIBER_CONSUMER_GROUP_NAME = "testGonsumerName-" + SUBSCRIBER_CONSUMER_ID;
    protected static final int SUBSCRIBER_TIMEOUT_MS = 2000;
    protected static final int SUBSCRIBER_MESSAGE_LIMIT = 20;

    /**
     * Creates Sample Publisher settings for unit testing purposes
     *
     * @return sample publisher settings for testing
     */
    protected static DMaaPMRPublisherConfig getPublisherConfig() {
        return new DMaaPMRPublisherConfig.Builder(HOST_NAME, TOPIC_NAME)
                .setPortNumber(PORT_NUMBER)
                .setProtocol(HTTP_PROTOCOL)
                .setUserName(USERNAME)
                .setUserPassword(PASSWORD)
                .setContentType(CONTENT_TYPE)
                .setMaxRecoveryQueueSize(PUBLISHER_MAX_RECOVERY_QUEUE_SIZE)
                .setMaxBatchSize(PUBLISHER_MAX_BATCH_QUEUE_SIZE).build();
    }

    /**
     * Creates Sample Subscriber settings for unit testing purposes
     *
     * @return sample subscriber settings for testing
     */
    protected static DMaaPMRSubscriberConfig getSubscriberConfig(String consumerId, String consumerGroup) {
        return new DMaaPMRSubscriberConfig.Builder(HOST_NAME, TOPIC_NAME)
                .setPortNumber(PORT_NUMBER)
                .setUserName(USERNAME)
                .setUserPassword(PASSWORD)
                .setProtocol(HTTP_PROTOCOL)
                .setContentType(CONTENT_TYPE)
                .setConsumerGroup(consumerGroup != null ? consumerGroup : SUBSCRIBER_CONSUMER_GROUP_NAME)
                .setConsumerId(consumerId != null ? consumerId : SUBSCRIBER_CONSUMER_ID)
                .setTimeoutMS(SUBSCRIBER_TIMEOUT_MS)
                .setMessageLimit(SUBSCRIBER_MESSAGE_LIMIT).build();
    }

    /**
     * Creates two sample message for publishing
     *
     * @return sample publish message list
     */
    protected static List<String> getTwoSampleMessages() {
        String message1 = "{ \"message\" : \"Test Message1\"}";
        String message2 = "{ \"message\" : \"Test Message2\"}";
        return of(message1, message2);
    }


}