package ca.uhn.fhir.rest.server; import static org.hamcrest.Matchers.containsString; import static org.hamcrest.Matchers.not; import static org.hamcrest.Matchers.startsWith; import static org.hamcrest.Matchers.stringContainsInOrder; import static org.junit.Assert.assertEquals; import static org.junit.Assert.assertNull; import static org.junit.Assert.assertThat; import java.util.ArrayList; import java.util.Collections; import java.util.List; import java.util.concurrent.TimeUnit; import org.apache.commons.io.IOUtils; import org.apache.http.HttpResponse; import org.apache.http.NameValuePair; import org.apache.http.client.entity.UrlEncodedFormEntity; import org.apache.http.client.methods.CloseableHttpResponse; import org.apache.http.client.methods.HttpGet; import org.apache.http.client.methods.HttpPost; import org.apache.http.client.methods.HttpPut; import org.apache.http.entity.ContentType; import org.apache.http.entity.StringEntity; import org.apache.http.impl.client.CloseableHttpClient; import org.apache.http.impl.client.HttpClientBuilder; import org.apache.http.impl.conn.PoolingHttpClientConnectionManager; import org.apache.http.message.BasicNameValuePair; import org.eclipse.jetty.server.Server; import org.eclipse.jetty.servlet.ServletHandler; import org.eclipse.jetty.servlet.ServletHolder; import org.hl7.fhir.instance.model.api.IBaseResource; import org.junit.AfterClass; import org.junit.Before; import org.junit.BeforeClass; import org.junit.Test; import ca.uhn.fhir.context.FhirContext; import ca.uhn.fhir.model.api.IResource; import ca.uhn.fhir.model.dstu2.composite.IdentifierDt; import ca.uhn.fhir.model.dstu2.resource.Bundle; import ca.uhn.fhir.model.dstu2.resource.Bundle.Link; import ca.uhn.fhir.model.dstu2.resource.Patient; import ca.uhn.fhir.model.primitive.InstantDt; import ca.uhn.fhir.rest.annotation.Create; import ca.uhn.fhir.rest.annotation.RequiredParam; import ca.uhn.fhir.rest.annotation.ResourceParam; import ca.uhn.fhir.rest.annotation.Search; import ca.uhn.fhir.rest.api.MethodOutcome; import ca.uhn.fhir.rest.param.DateAndListParam; import ca.uhn.fhir.rest.param.ParamPrefixEnum; import ca.uhn.fhir.rest.param.QuantityParam; import ca.uhn.fhir.rest.param.ReferenceParam; import ca.uhn.fhir.rest.param.StringParam; import ca.uhn.fhir.util.PatternMatcher; import ca.uhn.fhir.util.PortUtil; import ca.uhn.fhir.util.TestUtil; public class SearchDstu2Test { private static CloseableHttpClient ourClient; private static FhirContext ourCtx = FhirContext.forDstu2(); private static DateAndListParam ourLastDateAndList; private static String ourLastMethod; private static QuantityParam ourLastQuantity; private static ReferenceParam ourLastRef; private static final org.slf4j.Logger ourLog = org.slf4j.LoggerFactory.getLogger(SearchDstu2Test.class); private static int ourPort; private static InstantDt ourReturnPublished; private static Server ourServer; private static RestfulServer ourServlet; @Before public void before() { ourLastMethod = null; ourLastDateAndList = null; ourLastRef = null; ourLastQuantity = null; ourServlet.setIgnoreServerParsedRequestParameters(true); } @Test public void testSearchWithInvalidPostUrl() throws Exception { // should end with _search HttpPost filePost = new HttpPost("http://localhost:" + ourPort + "/Patient?name=Central"); // add parameters to the post method List<NameValuePair> parameters = new ArrayList<NameValuePair>(); parameters.add(new BasicNameValuePair("_id", "aaa")); UrlEncodedFormEntity sendentity = new UrlEncodedFormEntity(parameters, "UTF-8"); filePost.setEntity(sendentity); HttpResponse status = ourClient.execute(filePost); String responseContent = IOUtils.toString(status.getEntity().getContent()); IOUtils.closeQuietly(status.getEntity().getContent()); ourLog.info(responseContent); assertEquals(400, status.getStatusLine().getStatusCode()); assertThat(responseContent, containsString("<diagnostics value=\"Incorrect Content-Type header value of "application/x-www-form-urlencoded; charset=UTF-8" was provided in the request. A FHIR Content-Type is required for "CREATE" operation\"/>")); } @Test public void testEncodeConvertsReferencesToRelative() throws Exception { HttpGet httpGet = new HttpGet("http://localhost:" + ourPort + "/Patient?_query=searchWithRef"); HttpResponse status = ourClient.execute(httpGet); String responseContent = IOUtils.toString(status.getEntity().getContent()); IOUtils.closeQuietly(status.getEntity().getContent()); ourLog.info(responseContent); assertThat(responseContent, not(containsString("text"))); assertEquals(200, status.getStatusLine().getStatusCode()); Patient patient = (Patient) ourCtx.newXmlParser().parseResource(Bundle.class, responseContent).getEntry().get(0).getResource(); String ref = patient.getManagingOrganization().getReference().getValue(); assertEquals("Organization/555", ref); assertNull(status.getFirstHeader(Constants.HEADER_CONTENT_LOCATION)); } @Test public void testEncodeConvertsReferencesToRelativeJson() throws Exception { HttpGet httpGet = new HttpGet("http://localhost:" + ourPort + "/Patient?_query=searchWithRef&_format=json"); HttpResponse status = ourClient.execute(httpGet); String responseContent = IOUtils.toString(status.getEntity().getContent()); IOUtils.closeQuietly(status.getEntity().getContent()); ourLog.info(responseContent); assertThat(responseContent, not(containsString("text"))); assertEquals(200, status.getStatusLine().getStatusCode()); Patient patient = (Patient) ourCtx.newJsonParser().parseResource(Bundle.class, responseContent).getEntry().get(0).getResource(); String ref = patient.getManagingOrganization().getReference().getValue(); assertEquals("Organization/555", ref); assertNull(status.getFirstHeader(Constants.HEADER_CONTENT_LOCATION)); } @Test public void testResultBundleHasUpdateTime() throws Exception { ourReturnPublished = new InstantDt("2011-02-03T11:22:33Z"); HttpGet httpGet = new HttpGet("http://localhost:" + ourPort + "/Patient?_query=searchWithBundleProvider&_pretty=true"); HttpResponse status = ourClient.execute(httpGet); String responseContent = IOUtils.toString(status.getEntity().getContent()); IOUtils.closeQuietly(status.getEntity().getContent()); ourLog.info(responseContent); assertThat(responseContent, stringContainsInOrder("<lastUpdated value=\"2011-02-03T11:22:33Z\"/>")); } @Test public void testResultBundleHasUuid() throws Exception { HttpGet httpGet = new HttpGet("http://localhost:" + ourPort + "/Patient?_query=searchWithRef"); HttpResponse status = ourClient.execute(httpGet); String responseContent = IOUtils.toString(status.getEntity().getContent()); IOUtils.closeQuietly(status.getEntity().getContent()); ourLog.info(responseContent); assertEquals(200, status.getStatusLine().getStatusCode()); assertThat(responseContent, PatternMatcher.pattern("id value..[0-9a-f-]+\\\"")); } @Test public void testSearchBlacklist01Failing() throws Exception { HttpGet httpGet = new HttpGet("http://localhost:" + ourPort + "/Patient?_query=searchBlacklist01&ref.black1=value"); HttpResponse status = ourClient.execute(httpGet); String responseContent = IOUtils.toString(status.getEntity().getContent()); IOUtils.closeQuietly(status.getEntity().getContent()); ourLog.info(responseContent); assertEquals(400, status.getStatusLine().getStatusCode()); } @Test public void testSearchBlacklist01Passing() throws Exception { HttpGet httpGet = new HttpGet("http://localhost:" + ourPort + "/Patient?_query=searchBlacklist01&ref.white1=value"); HttpResponse status = ourClient.execute(httpGet); String responseContent = IOUtils.toString(status.getEntity().getContent()); IOUtils.closeQuietly(status.getEntity().getContent()); ourLog.info(responseContent); assertEquals(200, status.getStatusLine().getStatusCode()); assertEquals("searchBlacklist01", ourLastMethod); } @Test public void testSearchByPost() throws Exception { HttpPost httpGet = new HttpPost("http://localhost:" + ourPort + "/Patient/_search"); StringEntity entity = new StringEntity("searchDateAndList=2001,2002&searchDateAndList=2003,2004", ContentType.APPLICATION_FORM_URLENCODED); httpGet.setEntity(entity); HttpResponse status = ourClient.execute(httpGet); String responseContent = IOUtils.toString(status.getEntity().getContent()); IOUtils.closeQuietly(status.getEntity().getContent()); ourLog.info(responseContent); assertEquals("searchDateAndList", ourLastMethod); assertEquals(2, ourLastDateAndList.getValuesAsQueryTokens().size()); assertEquals(2, ourLastDateAndList.getValuesAsQueryTokens().get(0).getValuesAsQueryTokens().size()); assertEquals(2, ourLastDateAndList.getValuesAsQueryTokens().get(1).getValuesAsQueryTokens().size()); assertEquals("2001", ourLastDateAndList.getValuesAsQueryTokens().get(0).getValuesAsQueryTokens().get(0).getValueAsString()); assertEquals("2002", ourLastDateAndList.getValuesAsQueryTokens().get(0).getValuesAsQueryTokens().get(1).getValueAsString()); assertThat(responseContent, containsString("SYSTEM")); } @Test public void testSearchMethodReturnsNull() throws Exception { HttpGet httpGet = new HttpGet("http://localhost:" + ourPort + "/Patient?_query=searchReturnNull"); HttpResponse status = ourClient.execute(httpGet); String responseContent = IOUtils.toString(status.getEntity().getContent()); IOUtils.closeQuietly(status.getEntity().getContent()); ourLog.info(responseContent); assertEquals("searchReturnNull", ourLastMethod); assertThat(responseContent, containsString("<total value=\"0\"/>")); } @Test public void testSearchByPostWithBodyAndUrlParams() throws Exception { HttpPost httpGet = new HttpPost("http://localhost:" + ourPort + "/Patient/_search?_format=json"); StringEntity entity = new StringEntity("searchDateAndList=2001,2002&searchDateAndList=2003,2004", ContentType.APPLICATION_FORM_URLENCODED); httpGet.setEntity(entity); CloseableHttpResponse status = ourClient.execute(httpGet); String responseContent = IOUtils.toString(status.getEntity().getContent()); IOUtils.closeQuietly(status.getEntity().getContent()); ourLog.info(responseContent); assertEquals("searchDateAndList", ourLastMethod); assertEquals(2, ourLastDateAndList.getValuesAsQueryTokens().size()); assertEquals(2, ourLastDateAndList.getValuesAsQueryTokens().get(0).getValuesAsQueryTokens().size()); assertEquals(2, ourLastDateAndList.getValuesAsQueryTokens().get(1).getValuesAsQueryTokens().size()); assertEquals("2001", ourLastDateAndList.getValuesAsQueryTokens().get(0).getValuesAsQueryTokens().get(0).getValueAsString()); assertEquals("2002", ourLastDateAndList.getValuesAsQueryTokens().get(0).getValuesAsQueryTokens().get(1).getValueAsString()); assertThat(responseContent, containsString(":\"SYSTEM\"")); } @Test public void testSearchByPostWithBodyAndUrlParamsNoManual() throws Exception { ourServlet.setIgnoreServerParsedRequestParameters(false); HttpPost httpGet = new HttpPost("http://localhost:" + ourPort + "/Patient/_search?_format=json"); StringEntity entity = new StringEntity("searchDateAndList=2001,2002&searchDateAndList=2003,2004", ContentType.APPLICATION_FORM_URLENCODED); httpGet.setEntity(entity); CloseableHttpResponse status = ourClient.execute(httpGet); String responseContent = IOUtils.toString(status.getEntity().getContent()); IOUtils.closeQuietly(status.getEntity().getContent()); ourLog.info(responseContent); assertEquals("searchDateAndList", ourLastMethod); assertEquals(2, ourLastDateAndList.getValuesAsQueryTokens().size()); assertEquals(2, ourLastDateAndList.getValuesAsQueryTokens().get(0).getValuesAsQueryTokens().size()); assertEquals(2, ourLastDateAndList.getValuesAsQueryTokens().get(1).getValuesAsQueryTokens().size()); assertEquals("2001", ourLastDateAndList.getValuesAsQueryTokens().get(0).getValuesAsQueryTokens().get(0).getValueAsString()); assertEquals("2002", ourLastDateAndList.getValuesAsQueryTokens().get(0).getValuesAsQueryTokens().get(1).getValueAsString()); assertThat(responseContent, containsString(":\"SYSTEM\"")); } @Test public void testSearchByPut() throws Exception { HttpPut httpGet = new HttpPut("http://localhost:" + ourPort + "/Patient/_search"); StringEntity entity = new StringEntity("searchDateAndList=2001,2002&searchDateAndList=2003,2004", ContentType.APPLICATION_FORM_URLENCODED); httpGet.setEntity(entity); HttpResponse status = ourClient.execute(httpGet); String responseContent = IOUtils.toString(status.getEntity().getContent()); IOUtils.closeQuietly(status.getEntity().getContent()); ourLog.info(responseContent); assertEquals(400, status.getStatusLine().getStatusCode()); } @Test public void testSearchDateAndList() throws Exception { HttpGet httpGet = new HttpGet("http://localhost:" + ourPort + "/Patient?searchDateAndList=2001,2002&searchDateAndList=2003,2004"); HttpResponse status = ourClient.execute(httpGet); String responseContent = IOUtils.toString(status.getEntity().getContent()); IOUtils.closeQuietly(status.getEntity().getContent()); ourLog.info(responseContent); assertEquals("searchDateAndList", ourLastMethod); assertEquals(2, ourLastDateAndList.getValuesAsQueryTokens().size()); assertEquals(2, ourLastDateAndList.getValuesAsQueryTokens().get(0).getValuesAsQueryTokens().size()); assertEquals(2, ourLastDateAndList.getValuesAsQueryTokens().get(1).getValuesAsQueryTokens().size()); assertEquals("2001", ourLastDateAndList.getValuesAsQueryTokens().get(0).getValuesAsQueryTokens().get(0).getValueAsString()); assertEquals("2002", ourLastDateAndList.getValuesAsQueryTokens().get(0).getValuesAsQueryTokens().get(1).getValueAsString()); } /** * See #247 */ @Test public void testSearchPagesAllHaveCorrectBundleType() throws Exception { Bundle resp; { HttpGet httpGet = new HttpGet("http://localhost:" + ourPort + "/Patient?searchHugeResults=yes&_count=10&_pretty=true"); HttpResponse status = ourClient.execute(httpGet); String responseContent = IOUtils.toString(status.getEntity().getContent()); IOUtils.closeQuietly(status.getEntity().getContent()); ourLog.info(responseContent); resp = ourCtx.newXmlParser().parseResource(Bundle.class, responseContent); assertEquals("searchset", resp.getType()); assertEquals(100, resp.getTotal().intValue()); } Link nextLink = resp.getLink("next"); assertThat(nextLink.getUrl(), startsWith("http://")); // Now try the next page { HttpGet httpGet = new HttpGet(nextLink.getUrl()); HttpResponse status = ourClient.execute(httpGet); String responseContent = IOUtils.toString(status.getEntity().getContent()); IOUtils.closeQuietly(status.getEntity().getContent()); ourLog.info(responseContent); resp = ourCtx.newXmlParser().parseResource(Bundle.class, responseContent); assertEquals("searchset", resp.getType()); assertEquals(100, resp.getTotal().intValue()); } nextLink = resp.getLink("next"); assertThat(nextLink.getUrl(), startsWith("http://")); // Now try a third page { HttpGet httpGet = new HttpGet(nextLink.getUrl()); HttpResponse status = ourClient.execute(httpGet); String responseContent = IOUtils.toString(status.getEntity().getContent()); IOUtils.closeQuietly(status.getEntity().getContent()); ourLog.info(responseContent); resp = ourCtx.newXmlParser().parseResource(Bundle.class, responseContent); assertEquals("searchset", resp.getType()); assertEquals(100, resp.getTotal().intValue()); } } /** * See #296 */ @Test public void testSearchQuantityMissingTrue() throws Exception { HttpGet httpGet = new HttpGet("http://localhost:" + ourPort + "/Patient?quantity:missing=true"); HttpResponse status = ourClient.execute(httpGet); String responseContent = IOUtils.toString(status.getEntity().getContent()); IOUtils.closeQuietly(status.getEntity().getContent()); ourLog.info(responseContent); assertEquals(200, status.getStatusLine().getStatusCode()); assertEquals(Boolean.TRUE, ourLastQuantity.getMissing()); } /** * See #296 */ @Test public void testSearchQuantityValue() throws Exception { HttpGet httpGet = new HttpGet("http://localhost:" + ourPort + "/Patient?quantity=gt100"); HttpResponse status = ourClient.execute(httpGet); String responseContent = IOUtils.toString(status.getEntity().getContent()); IOUtils.closeQuietly(status.getEntity().getContent()); ourLog.info(responseContent); assertEquals(200, status.getStatusLine().getStatusCode()); assertEquals(ParamPrefixEnum.GREATERTHAN, ourLastQuantity.getPrefix()); assertEquals(100, ourLastQuantity.getValue().intValue()); } @Test public void testSearchReferenceParams01() throws Exception { HttpGet httpGet = new HttpGet("http://localhost:" + ourPort + "/Patient?_query=searchNoList&ref=123"); HttpResponse status = ourClient.execute(httpGet); String responseContent = IOUtils.toString(status.getEntity().getContent()); IOUtils.closeQuietly(status.getEntity().getContent()); ourLog.info(responseContent); assertEquals(200, status.getStatusLine().getStatusCode()); assertEquals("123", ourLastRef.getIdPart()); assertEquals(null, ourLastRef.getResourceType()); } @Test public void testSearchReferenceParams02() throws Exception { HttpGet httpGet = new HttpGet("http://localhost:" + ourPort + "/Patient?_query=searchNoList&ref=Patient/123"); HttpResponse status = ourClient.execute(httpGet); String responseContent = IOUtils.toString(status.getEntity().getContent()); IOUtils.closeQuietly(status.getEntity().getContent()); ourLog.info(responseContent); assertEquals(200, status.getStatusLine().getStatusCode()); assertEquals("123", ourLastRef.getIdPart()); assertEquals("Patient", ourLastRef.getResourceType()); } @Test public void testSearchReferenceParams03() throws Exception { HttpGet httpGet = new HttpGet("http://localhost:" + ourPort + "/Patient?_query=searchNoList&ref:Patient=Patient/123"); HttpResponse status = ourClient.execute(httpGet); String responseContent = IOUtils.toString(status.getEntity().getContent()); IOUtils.closeQuietly(status.getEntity().getContent()); ourLog.info(responseContent); assertEquals(200, status.getStatusLine().getStatusCode()); assertEquals("123", ourLastRef.getIdPart()); assertEquals("Patient", ourLastRef.getResourceType()); } @Test public void testSearchReferenceParams04() throws Exception { HttpGet httpGet = new HttpGet("http://localhost:" + ourPort + "/Patient?_query=searchNoList&ref:Patient=123"); HttpResponse status = ourClient.execute(httpGet); String responseContent = IOUtils.toString(status.getEntity().getContent()); IOUtils.closeQuietly(status.getEntity().getContent()); ourLog.info(responseContent); assertEquals(200, status.getStatusLine().getStatusCode()); assertEquals("123", ourLastRef.getIdPart()); assertEquals("Patient", ourLastRef.getResourceType()); } @Test public void testSearchWhitelist01Failing() throws Exception { HttpGet httpGet = new HttpGet("http://localhost:" + ourPort + "/Patient?_query=searchWhitelist01&ref=value"); HttpResponse status = ourClient.execute(httpGet); String responseContent = IOUtils.toString(status.getEntity().getContent()); IOUtils.closeQuietly(status.getEntity().getContent()); ourLog.info(responseContent); assertEquals(400, status.getStatusLine().getStatusCode()); } @Test public void testSearchWhitelist01Passing() throws Exception { HttpGet httpGet = new HttpGet("http://localhost:" + ourPort + "/Patient?_query=searchWhitelist01&ref.white1=value"); HttpResponse status = ourClient.execute(httpGet); String responseContent = IOUtils.toString(status.getEntity().getContent()); IOUtils.closeQuietly(status.getEntity().getContent()); ourLog.info(responseContent); assertEquals(200, status.getStatusLine().getStatusCode()); assertEquals("searchWhitelist01", ourLastMethod); } @AfterClass public static void afterClassClearContext() throws Exception { ourServer.stop(); TestUtil.clearAllStaticFieldsForUnitTest(); } @BeforeClass public static void beforeClass() throws Exception { ourPort = PortUtil.findFreePort(); ourServer = new Server(ourPort); DummyPatientResourceProvider patientProvider = new DummyPatientResourceProvider(); ServletHandler proxyHandler = new ServletHandler(); ourServlet = new RestfulServer(ourCtx); ourServlet.setPagingProvider(new FifoMemoryPagingProvider(10)); ourServlet.setResourceProviders(patientProvider); ServletHolder servletHolder = new ServletHolder(ourServlet); proxyHandler.addServletWithMapping(servletHolder, "/*"); ourServer.setHandler(proxyHandler); ourServer.start(); PoolingHttpClientConnectionManager connectionManager = new PoolingHttpClientConnectionManager(5000, TimeUnit.MILLISECONDS); HttpClientBuilder builder = HttpClientBuilder.create(); builder.setConnectionManager(connectionManager); ourClient = builder.build(); } public static class DummyPatientResourceProvider implements IResourceProvider { @Override public Class<? extends IResource> getResourceType() { return Patient.class; } //@formatter:off @Search(queryName="searchBlacklist01") public List<Patient> searchBlacklist01( @RequiredParam(chainBlacklist="black1", name = "ref") ReferenceParam theParam) { ourLastMethod = "searchBlacklist01"; return Collections.emptyList(); } //@formatter:on /** * For testSearchWithInvalidPostUrl, ok to add a real body later */ @Create public MethodOutcome create(@ResourceParam Patient thePatient) { throw new UnsupportedOperationException(); } //@formatter:off @Search() public List<Patient> searchDateAndList( @RequiredParam(name = "searchDateAndList") DateAndListParam theParam) { ourLastMethod = "searchDateAndList"; ourLastDateAndList = theParam; ArrayList<Patient> retVal = new ArrayList<Patient>(); Patient patient = new Patient(); patient.setId("1"); retVal.add(patient.addIdentifier(new IdentifierDt("SYSTEM", "CODE"))); return retVal; } //@formatter:on //@formatter:off @Search() public List<Patient> searchHugeResults( @RequiredParam(name = "searchHugeResults") StringParam theParam) { ourLastMethod = "searchHugeResults"; ArrayList<Patient> retVal = new ArrayList<Patient>(); for (int i = 0; i < 100; i++) { Patient patient = new Patient(); patient.setId("" + i); retVal.add(patient.addIdentifier(new IdentifierDt("SYSTEM", "CODE"+ i))); } return retVal; } //@formatter:on //@formatter:off @Search(queryName="searchNoList") public List<Patient> searchNoList( @RequiredParam(name = "ref") ReferenceParam theParam) { ourLastMethod = "searchNoList"; ourLastRef = theParam; return Collections.emptyList(); } //@formatter:on //@formatter:off @Search(queryName="searchReturnNull") public List<Patient> searchNoList() { ourLastMethod = "searchReturnNull"; return null; } //@formatter:on //@formatter:off @Search() public List<Patient> searchQuantity( @RequiredParam(name="quantity") QuantityParam theParam) { ourLastMethod = "searchQuantity"; ourLastQuantity = theParam; return Collections.emptyList(); } //@formatter:on //@formatter:off @Search(queryName="searchWhitelist01") public List<Patient> searchWhitelist01( @RequiredParam(chainWhitelist="white1", name = "ref") ReferenceParam theParam) { ourLastMethod = "searchWhitelist01"; return Collections.emptyList(); } //@formatter:on @Search(queryName = "searchWithBundleProvider") public IBundleProvider searchWithBundleProvider() { return new IBundleProvider() { @Override public InstantDt getPublished() { return ourReturnPublished; } @Override public List<IBaseResource> getResources(int theFromIndex, int theToIndex) { throw new IllegalStateException(); } @Override public Integer preferredPageSize() { return null; } @Override public Integer size() { return 0; } @Override public String getUuid() { return null; } }; } @Search(queryName = "searchWithRef") public Patient searchWithRef() { Patient patient = new Patient(); patient.setId("Patient/1/_history/1"); patient.getManagingOrganization().setReference("http://localhost:" + ourPort + "/Organization/555/_history/666"); return patient; } } }