/*
 * #%L
 * =====================================================
 *   _____                _     ____  _   _       _   _
 *  |_   _|_ __ _   _ ___| |_  / __ \| | | | ___ | | | |
 *    | | | '__| | | / __| __|/ / _` | |_| |/ __|| |_| |
 *    | | | |  | |_| \__ \ |_| | (_| |  _  |\__ \|  _  |
 *    |_| |_|   \__,_|___/\__|\ \__,_|_| |_||___/|_| |_|
 *                             \____/
 * 
 * =====================================================
 * 
 * Hochschule Hannover
 * (University of Applied Sciences and Arts, Hannover)
 * Faculty IV, Dept. of Computer Science
 * Ricklinger Stadtweg 118, 30459 Hannover, Germany
 * 
 * Email: trust@f4-i.fh-hannover.de
 * Website: http://trust.f4.hs-hannover.de/
 * 
 * This file is part of visitmeta-dataservice, version 0.6.0,
 * implemented by the Trust@HsH research group at the Hochschule Hannover.
 * %%
 * Copyright (C) 2012 - 2016 Trust@HsH
 * %%
 * 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.
 * #L%
 */
package de.hshannover.f4.trust.visitmeta.ifmap.util;

import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.when;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;

import de.hshannover.f4.trust.ifmapj.messages.PollResult;
import de.hshannover.f4.trust.ifmapj.messages.SearchResult;

/**
 * 
 * @author Marcel Reichenbach
 *
 */
public class PollResultMock implements IfmapMock<PollResult> {

	private PollResult mPollResult_mock;

	private List<SearchResult> mSearchResults;

	public PollResultMock() {
		mPollResult_mock = mock(PollResult.class);
		mSearchResults = new ArrayList<SearchResult>();

		when(mPollResult_mock.getResults()).thenReturn(mSearchResults);
	}

	public PollResultMock(SearchResult... searchResults) {
		this();

		mSearchResults = Arrays.asList(searchResults);
		when(mPollResult_mock.getResults()).thenReturn(mSearchResults);
	}

	public PollResultMock(List<SearchResultMock> searchResults) {
		this();

		addSearchResultMock(searchResults);
	}

	public PollResultMock(SearchResultMock... searchResults) {
		this(Arrays.asList(searchResults));
	}

	public void addSearchResult(SearchResult searchResult) {
		mSearchResults.add(searchResult);
		when(mPollResult_mock.getResults()).thenReturn(mSearchResults);
	}

	public void addSearchResultMock(List<SearchResultMock> searchResultMock) {
		for (SearchResultMock mock : searchResultMock) {
			mSearchResults.add(mock.getMock());
		}

		when(mPollResult_mock.getResults()).thenReturn(mSearchResults);
	}

	@Override
	public PollResult getMock() {
		return mPollResult_mock;
	}

}