/**
 * The contents of this file are subject to the OpenMRS Public License
 * Version 1.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://license.openmrs.org
 *
 * Software distributed under the License is distributed on an "AS IS"
 * basis, WITHOUT WARRANTY OF ANY KIND, either express or implied. See the
 * License for the specific language governing rights and limitations
 * under the License.
 *
 * Copyright (C) OpenMRS, LLC.  All Rights Reserved.
 */
package org.openmrs.module.reporting.calculation;

import java.util.Arrays;
import java.util.Collections;
import java.util.HashMap;
import java.util.Map;
import java.util.Set;

import org.apache.commons.collections.CollectionUtils;
import org.junit.Assert;
import org.junit.Before;
import org.junit.Test;
import org.openmrs.PatientIdentifier;
import org.openmrs.PatientIdentifierType;
import org.openmrs.api.PatientService;
import org.openmrs.api.context.Context;
import org.openmrs.calculation.result.CalculationResultMap;
import org.openmrs.calculation.result.ListResult;
import org.openmrs.calculation.result.ResultUtil;
import org.openmrs.module.reporting.common.TestUtil;
import org.openmrs.test.BaseModuleContextSensitiveTest;

public class PatientDataCalculationBehaviorTest extends BaseModuleContextSensitiveTest {
	
	protected static final String XML_DATASET_PATH = "org/openmrs/module/reporting/include/";
	
	protected static final String XML_REPORT_TEST_DATASET = "ReportTestDataset";
	
	private PatientService ps;
	
	@Before
	public void setup() throws Exception {
		executeDataSet(XML_DATASET_PATH + new TestUtil().getTestDatasetFilename(XML_REPORT_TEST_DATASET));
		ps = Context.getPatientService();
	}
	
	@Test
	public void evaluate_shouldEvaluateAPatientCalculation() throws Exception {
		Integer patientId1 = 2;
		Integer patientId2 = 7;
		Set<PatientIdentifier> identifiers1 = ps.getPatient(patientId1).getIdentifiers();
		Set<PatientIdentifier> identifiers2 = ps.getPatient(patientId2).getIdentifiers();
		PatientDataCalculation calculation = new PatientDataCalculationProvider().getCalculation(
		    "org.openmrs.module.reporting.data.patient.definition.PatientIdentifierDataDefinition", null);
		Map<String, Object> parameters = new HashMap<String, Object>();
		parameters.put("types", ps.getAllPatientIdentifierTypes(false));
		
		CalculationResultMap results = calculation.evaluate(Arrays.asList(patientId1, patientId2), parameters, null);
		
		Assert.assertEquals(identifiers2.iterator().next(), ResultUtil.getFirst(results.get(patientId2)).getValue());
		
		ListResult lr = (ListResult) results.get(patientId1);
		Assert.assertEquals(3, lr.size());
		
		Assert.assertTrue(CollectionUtils.isEqualCollection(identifiers1, lr.getValues()));
	}
	
	@Test
	public void evaluate_shouldEvaluateAPatientCalculationWithTheSpecifiedParameterValues() throws Exception {
		Integer patientId1 = 2;
		Integer patientId2 = 7;
		PatientIdentifierType type = Context.getPatientService().getPatientIdentifierType(1);
		PatientIdentifier id1 = ps.getPatient(patientId1).getPatientIdentifier(type);
		PatientIdentifier id2 = ps.getPatient(patientId2).getPatientIdentifier(type);
		PatientDataCalculation calculation = new PatientDataCalculationProvider().getCalculation(
		    "org.openmrs.module.reporting.data.patient.definition.PatientIdentifierDataDefinition", null);
		Map<String, Object> parameters = new HashMap<String, Object>();
		parameters.put("types", Collections.singletonList(type));
		parameters.put("includeFirstNonNullOnly", true);
		
		CalculationResultMap results = calculation.evaluate(Arrays.asList(patientId1, patientId2), parameters, null);
		
		Assert.assertEquals(id1, ResultUtil.getFirst(results.get(patientId1)).getValue());
		Assert.assertEquals(id2, ResultUtil.getFirst(results.get(patientId2)).getValue());
	}
	
	@Test
	public void evaluate_shouldEvaluateTheSpecifiedPersonCalculation() throws Exception {
		Integer patientId1 = 2;
		Integer patientId2 = 7;
		String gender1 = ps.getPatient(patientId1).getGender();
		String gender2 = ps.getPatient(patientId2).getGender();
		PatientDataCalculation calculation = new PatientDataCalculationProvider().getCalculation(
		    "org.openmrs.module.reporting.data.person.definition.GenderDataDefinition", null);
		
		CalculationResultMap results = calculation.evaluate(Arrays.asList(patientId1, patientId2), null, null);
		
		Assert.assertEquals(gender1, ResultUtil.getFirst(results.get(patientId1)).getValue());
		Assert.assertEquals(gender2, ResultUtil.getFirst(results.get(patientId2)).getValue());
	}
}