/*******************************************************************************
 *  ============LICENSE_START=======================================================
 *  org.onap.dmaap
 *  ================================================================================
 *  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=========================================================
 *
 *  ECOMP is a trademark and service mark of AT&T Intellectual Property.
 *  
 *******************************************************************************/
package com.att.nsa.mr.test.support;

import java.io.IOException;
import java.util.LinkedList;
import java.util.List;

import org.slf4j.Logger;

import com.att.nsa.mr.client.MRConsumer;
import com.att.nsa.mr.client.response.MRConsumerResponse;

/**
 * A helper for unit testing systems that use a MRConsumer. When setting
 * up your test, inject an instance into MRClientFactory to have it return
 * the mock client.
 * 
 * @author author
 *
 */
public class MRConsumerMock implements MRConsumer
{
	public class Entry
	{
		public Entry ( long waitMs, int statusCode, List<String> msgs )
		{
			fWaitMs = waitMs;
			fStatusCode = statusCode;
			fStatusMsg = null;
			fMsgs = new LinkedList<> ( msgs );
		}

		public Entry ( long waitMs, int statusCode, String statusMsg )
		{
			fWaitMs = waitMs;
			fStatusCode = statusCode;
			fStatusMsg = statusMsg;
			fMsgs = null;
		}

		public LinkedList<String> run () throws IOException
		{
			try
			{
				Thread.sleep ( fWaitMs );
				if ( fStatusCode >= 200 && fStatusCode <= 299 )
				{
					return fMsgs;
				}
				throw new IOException ( "" + fStatusCode + " " + fStatusMsg );
			}
			catch ( InterruptedException e )
			{
				Thread.currentThread().interrupt();
				throw new IOException ( e );
			}
		}

		private final long fWaitMs;
		private final int fStatusCode;
		private final String fStatusMsg;
		private final LinkedList<String> fMsgs;
	}

	public MRConsumerMock ()
	{
		fReplies = new LinkedList<> ();
	}

	@Override
	public void close ()
	{
	}

	@Override
	public void setApiCredentials ( String apiKey, String apiSecret )
	{
	}

	@Override
	public void clearApiCredentials ()
	{
	}

	public synchronized void add ( Entry e )
	{
		fReplies.add ( e );
	}

	public void addImmediateMsg ( String msg )
	{
		addDelayedMsg ( 0, msg );
	}

	public void addDelayedMsg ( long delay, String msg )
	{
		final LinkedList<String> list = new LinkedList<> ();
		list.add ( msg );
		add ( new Entry ( delay, 200, list ) );
	}

	public void addImmediateMsgGroup ( List<String> msgs )
	{
		addDelayedMsgGroup ( 0, msgs );
	}

	public void addDelayedMsgGroup ( long delay, List<String> msgs )
	{
		final LinkedList<String> list = new LinkedList<> ( msgs );
		add ( new Entry ( delay, 200, list ) );
	}

	public void addImmediateError ( int statusCode, String statusText )
	{
		add ( new Entry ( 0, statusCode, statusText ) );
	}

	@Override
	public Iterable<String> fetch () throws IOException
	{
		return fetch ( -1, -1 );
	}

	@Override
	public Iterable<String> fetch ( int timeoutMs, int limit ) throws IOException
	{
		return fReplies.size () > 0 ? fReplies.removeFirst ().run() : new LinkedList<String>();
	}

	@Override
	public void logTo ( Logger log )
	{
	}

	private final LinkedList<Entry> fReplies;

	@Override
	public MRConsumerResponse fetchWithReturnConsumerResponse() {
		// TODO Auto-generated method stub
		return null;
	}

	@Override
	public MRConsumerResponse fetchWithReturnConsumerResponse(int timeoutMs,
			int limit) {
		// TODO Auto-generated method stub
		return null;
	}
}