package org.gcube.dataharvest;

import java.util.Date;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

import org.gcube.dataharvest.datamodel.HarvestedData;
import org.gcube.dataharvest.harvester.sobigdata.TagMeMethodInvocationHarvester;
import org.gcube.dataharvest.utils.ContextTest;
import org.gcube.dataharvest.utils.DateUtils;
import org.gcube.dataharvest.utils.MeasureType;
import org.junit.Test;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

public class AccountingDataHarvesterPluginTest extends ContextTest {
	
	private static Logger logger = LoggerFactory.getLogger(AccountingDataHarvesterPluginTest.class);
	
	@Test
	public void test() {
		try {
			
			org.gcube.dataharvest.utils.Utils.setContext(ROOT);
			
			DataHarvestPluginDeclaration dataHarvestPluginDeclaration = new DataHarvestPluginDeclaration();
			AccountingDataHarvesterPlugin accountingDataHarvesterPlugin = new AccountingDataHarvesterPlugin(dataHarvestPluginDeclaration);
			
			Map<String,Object> inputs = new HashMap<>();
			
			
			MeasureType measureType = MeasureType.MONTHLY;
			
			inputs.put(AccountingDataHarvesterPlugin.MEASURE_TYPE_INPUT_PARAMETER, measureType.name());
			inputs.put(AccountingDataHarvesterPlugin.RERUN_INPUT_PARAMETER, true);
			inputs.put(AccountingDataHarvesterPlugin.DRY_RUN_INPUT_PARAMETER, true);
			
			/*
			Calendar from = DateUtils.getStartCalendar(2018, Calendar.APRIL, 1);
			String fromDate = DateUtils.LAUNCH_DATE_FORMAT.format(from.getTime());
			logger.trace("{} is {}", AccountingDataHarvesterPlugin.START_DATE_INPUT_PARAMETER, fromDate);
			inputs.put(AccountingDataHarvesterPlugin.START_DATE_INPUT_PARAMETER, fromDate);
			*/
			
			accountingDataHarvesterPlugin.launch(inputs);

			logger.info("End.");

		} catch (Exception e) {
			logger.error("", e);
		}
	}
	
	

	@Test
	public void testTagMe() {
		try {
			
			org.gcube.dataharvest.utils.Utils.setContext(TAGME);
			
			MeasureType measureType = MeasureType.MONTHLY;
			
			Date start = DateUtils.getPreviousPeriod(measureType).getTime();
			Date end = DateUtils.getEndDateFromStartDate(measureType, start, 1);
			
			TagMeMethodInvocationHarvester methodInvocationHarvester = new TagMeMethodInvocationHarvester(start, end);
			List<HarvestedData> harvestedData = methodInvocationHarvester.getData();
			
			logger.debug("{}", harvestedData);
			
			
		}catch (Exception e) {
			logger.error("", e);
		}
	}
	
}