D-Net

package gr.uoa.di.webui.search;

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));

	}

}