ObsValueTextAsCodedConverterTest.java

package org.openmrs.module.reporting.data.converter;

import static org.hamcrest.CoreMatchers.is;
import static org.junit.Assert.assertNull;
import static org.junit.Assert.assertThat;

import org.junit.Test;
import org.openmrs.Location;
import org.openmrs.Obs;
import org.openmrs.Patient;
import org.openmrs.api.LocationService;
import org.openmrs.test.BaseModuleContextSensitiveTest;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Qualifier;

public class ObsValueTextAsCodedConverterTest extends BaseModuleContextSensitiveTest {

    @Autowired
    @Qualifier("locationService")
    private LocationService locationService;

    @Test
    public void shouldReturnLocationForObs() {

        Location expectedLocation = locationService.getLocation(2);
        Obs obs = new Obs();
        obs.setValueText("2");
        
        DataConverter converter = new ObsValueTextAsCodedConverter<Location>(Location.class);
        
        assertThat((Location) converter.convert(obs), is(expectedLocation));
    }

    @Test
    public void shouldReturnNullIfValueTextEmpty() {

        Obs obs = new Obs();
        obs.setValueText("");

        DataConverter converter = new ObsValueTextAsCodedConverter<Location>(Location.class);

        assertNull(converter.convert(obs));
    }

    // TODO we can remove test below once we support other OpenmrsObjects
    
    @Test(expected = IllegalArgumentException.class)
    public void shouldFailIfTypeOtherThanLocation() {
        ObsValueTextAsCodedConverter converter = new ObsValueTextAsCodedConverter<Patient>(Patient.class);
        converter.convert(new Obs());
    }

}