/**
 * 
 */
package org.gcube.accounting.analytics;

import java.util.HashSet;
import java.util.Set;

import org.gcube.accounting.datamodel.BasicUsageRecord;
import org.gcube.accounting.datamodel.usagerecords.JobUsageRecord;
import org.gcube.accounting.datamodel.usagerecords.PortletUsageRecord;
import org.gcube.accounting.datamodel.usagerecords.ServiceUsageRecord;
import org.gcube.accounting.datamodel.usagerecords.StorageUsageRecord;
import org.gcube.accounting.datamodel.usagerecords.TaskUsageRecord;
import org.gcube.documentstore.records.Record;
import org.junit.Assert;
import org.junit.Test;

/**
 * @author Luca Frosini (ISTI - CNR) http://www.lucafrosini.com/
 */
public class ResourceRecordQueryTest {

	public class TestUsageRecord extends BasicUsageRecord {

		/**
		 * Generated Serial Version UID
		 */
		private static final long serialVersionUID = 1939161386352514727L;

		@Override
		protected String giveMeRecordType() {
			return TestUsageRecord.class.getSimpleName();
		}
		
	}
	
	public static Set<Class<? extends Record>> getExpectedResourceRecordsTypes(){
		Set<Class<? extends Record>> expected = new HashSet<Class<? extends Record>>();
		expected.add(ServiceUsageRecord.class);
		expected.add(StorageUsageRecord.class);
		expected.add(JobUsageRecord.class);
		expected.add(TaskUsageRecord.class);
		expected.add(PortletUsageRecord.class);
		return expected;
	}
	
	@Test
	public void testGetResourceRecordsTypes(){
		Set<Class<? extends Record>> expected = getExpectedResourceRecordsTypes();
		Set<Class<? extends Record>> found = ResourceRecordQuery.getResourceRecordsTypes().keySet();
		Assert.assertTrue(expected.containsAll(found));
		Assert.assertTrue(found.containsAll(expected));
	}

	@Test
	public void testGetResourceRecordsTypesWithFakeClass(){
		Set<Class<? extends Record>> expected = getExpectedResourceRecordsTypes();
		expected.add(TestUsageRecord.class);
		Set<Class<? extends Record>> found = ResourceRecordQuery.getResourceRecordsTypes().keySet();
		Assert.assertTrue(expected.containsAll(found));
		Assert.assertFalse(found.containsAll(expected));
	}
	
}