DrugTest.java

/**
 * This Source Code Form is subject to the terms of the Mozilla Public License,
 * v. 2.0. If a copy of the MPL was not distributed with this file, You can
 * obtain one at http://mozilla.org/MPL/2.0/. OpenMRS is also distributed under
 * the terms of the Healthcare Disclaimer located at http://openmrs.org/license.
 *
 * Copyright (C) OpenMRS Inc. OpenMRS is a registered trademark and the OpenMRS
 * graphic logo is a trademark of OpenMRS Inc.
 */
package org.openmrs;

import org.junit.Assert;
import org.junit.Test;
import org.openmrs.test.BaseContextSensitiveTest;

/**
 * Contains test methods for {@link org.openmrs.Drug}.
 */
public class DrugTest extends BaseContextSensitiveTest {
	
	private final static String UUID_1 = "333cd82c-7d3d-11e3-8633-13f177b345d8";
	
	private final static String UUID_2 = "4eef1530-7d3d-11e3-ac6d-e388e198a21e";
	
	/**
	 * @see Drug#addDrugReferenceMap(DrugReferenceMap)
	 */
	@Test
	public void addDrugReferenceMap_shouldSetDrugAsTheDrugToWhichAMappingIsBeingAdded() {
		Drug drug1 = new Drug();
		drug1.setUuid(UUID_1);
		Drug drug2 = new Drug();
		drug2.setUuid(UUID_2);
		
		DrugReferenceMap map = new DrugReferenceMap();
		map.setDrug(drug2);
		drug1.addDrugReferenceMap(map);
		Assert.assertEquals(drug1, drug1.getDrugReferenceMaps().iterator().next().getDrug());
	}
	
	/**
	 * @see Drug#addDrugReferenceMap(DrugReferenceMap)
	 */
	@Test
	public void addDrugReferenceMap_shouldShouldNotAddDuplicateDrugReferenceMaps() {
		Drug drug = new Drug();
		
		DrugReferenceMap map1 = new DrugReferenceMap();
		map1.setUuid(UUID_1);
		DrugReferenceMap map2 = new DrugReferenceMap();
		map2.setUuid(UUID_2);
		DrugReferenceMap map2Duplicate = new DrugReferenceMap();
		map2Duplicate.setUuid(UUID_2);
		
		drug.addDrugReferenceMap(map1);
		drug.addDrugReferenceMap(map2);
		drug.addDrugReferenceMap(map2Duplicate);
		
		Assert.assertEquals(2, drug.getDrugReferenceMaps().size());
	}
}