package gr.uoa.di.webui.search;

import gr.uoa.di.web.utils.search.query.QueryCollectionEnhancer;

import java.util.ArrayList;

import junit.framework.Assert;
import junit.framework.TestCase;

public class QueryCollectionEnhancementTest extends TestCase {

	private QueryCollectionEnhancer enhancer = new QueryCollectionEnhancer();
	private String query = "no matter what query";
	
	public void testEmptyCollectionList(){
		enhancer.setCollectionIds(new ArrayList<String>());
		Assert.assertEquals(query, enhancer.enhanceQuery(query));
	}

	public void testNonEmptyCollectionList(){
		enhancer.setCollectionIds(new ArrayList<String>());
		
		enhancer.getCollectionIds().add("12345");
		Assert.assertEquals(query + " AND ( (collection = \"12345\") )",enhancer.enhanceQuery(query));
		
		enhancer.getCollectionIds().add("67890");
		Assert.assertEquals
			(query + " AND ( (collection = \"12345\") OR (collection = \"67890\") )",
				enhancer.enhanceQuery(query));
		
	}
	
	
}