Merge branch 'master' into kbd-20201125-cql-initial-impl

src/test/java/ca/uhn/fhir/jpa/starter/ElasticsearchLastNR4IT.java

@@ -0,0 +1,144 @@
+package ca.uhn.fhir.jpa.starter;
+
+import ca.uhn.fhir.context.ConfigurationException;
+import ca.uhn.fhir.context.FhirContext;
+import ca.uhn.fhir.jpa.search.lastn.ElasticsearchSvcImpl;
+import ca.uhn.fhir.rest.client.api.IGenericClient;
+import ca.uhn.fhir.rest.client.api.ServerValidationModeEnum;
+import ca.uhn.fhir.rest.client.interceptor.LoggingInterceptor;
+import org.hl7.fhir.instance.model.api.IIdType;
+import org.hl7.fhir.r4.model.Bundle;
+import org.hl7.fhir.r4.model.DateTimeType;
+import org.hl7.fhir.r4.model.IntegerType;
+import org.hl7.fhir.r4.model.Observation;
+import org.hl7.fhir.r4.model.Parameters;
+import org.hl7.fhir.r4.model.Patient;
+import org.hl7.fhir.r4.model.StringType;
+import org.junit.jupiter.api.BeforeAll;
+import org.junit.jupiter.api.BeforeEach;
+import org.junit.jupiter.api.Test;
+import org.junit.jupiter.api.extension.ExtendWith;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.boot.test.context.SpringBootTest;
+import org.springframework.boot.test.util.TestPropertyValues;
+import org.springframework.boot.web.server.LocalServerPort;
+import org.springframework.context.ApplicationContextInitializer;
+import org.springframework.context.ConfigurableApplicationContext;
+import org.springframework.test.context.ContextConfiguration;
+import org.springframework.test.context.junit.jupiter.SpringExtension;
+import pl.allegro.tech.embeddedelasticsearch.EmbeddedElastic;
+import pl.allegro.tech.embeddedelasticsearch.PopularProperties;
+
+import javax.annotation.PreDestroy;
+import java.io.IOException;
+import java.util.Date;
+import java.util.GregorianCalendar;
+import java.util.UUID;
+import java.util.concurrent.TimeUnit;
+
+import static org.junit.jupiter.api.Assertions.assertEquals;
+
+@ExtendWith(SpringExtension.class)
+@SpringBootTest(webEnvironment = SpringBootTest.WebEnvironment.RANDOM_PORT, classes = Application.class, properties =
+  {
+    "spring.batch.job.enabled=false",
+    "spring.datasource.url=jdbc:h2:mem:dbr4",
+    "hapi.fhir.fhir_version=r4",
+    "hapi.fhir.lastn_enabled=true",
+    "elasticsearch.enabled=true",
+    // Because the port is set randomly, we will set the rest_url using the Initializer.
+    // "elasticsearch.rest_url='http://localhost:9200'",
+    "elasticsearch.username=SomeUsername",
+    "elasticsearch.password=SomePassword"
+  })
+@ContextConfiguration(initializers = ElasticsearchLastNR4IT.Initializer.class)
+public class ElasticsearchLastNR4IT {
+
+  private IGenericClient ourClient;
+  private FhirContext ourCtx;
+
+  private static final String ELASTIC_VERSION = "6.5.4";
+  private static EmbeddedElastic embeddedElastic;
+
+  @Autowired
+  private ElasticsearchSvcImpl myElasticsearchSvc;
+
+  @BeforeAll
+  public static void beforeClass() {
+
+    embeddedElastic = null;
+    try {
+      embeddedElastic = EmbeddedElastic.builder()
+        .withElasticVersion(ELASTIC_VERSION)
+        .withSetting(PopularProperties.TRANSPORT_TCP_PORT, 0)
+        .withSetting(PopularProperties.HTTP_PORT, 0)
+        .withSetting(PopularProperties.CLUSTER_NAME, UUID.randomUUID())
+        .withStartTimeout(60, TimeUnit.SECONDS)
+        .build()
+        .start();
+    } catch (IOException | InterruptedException e) {
+      throw new ConfigurationException(e);
+    }
+  }
+
+  @PreDestroy
+  public void stop() {
+    embeddedElastic.stop();
+  }
+
+  @LocalServerPort
+  private int port;
+
+  @Test
+  void testLastN() throws IOException {
+
+    Patient pt = new Patient();
+    pt.addName().setFamily("Lastn").addGiven("Arthur");
+    IIdType id = ourClient.create().resource(pt).execute().getId().toUnqualifiedVersionless();
+
+    Observation obs = new Observation();
+    obs.getSubject().setReferenceElement(id);
+    String observationCode = "testobservationcode";
+    String codeSystem = "http://testobservationcodesystem";
+    obs.getCode().addCoding().setCode(observationCode).setSystem(codeSystem);
+    obs.setValue(new StringType(observationCode));
+    Date effectiveDtm = new GregorianCalendar().getTime();
+    obs.setEffective(new DateTimeType(effectiveDtm));
+    obs.getCategoryFirstRep().addCoding().setCode("testcategorycode").setSystem("http://testcategorycodesystem");
+    IIdType obsId = ourClient.create().resource(obs).execute().getId().toUnqualifiedVersionless();
+
+    myElasticsearchSvc.refreshIndex(ElasticsearchSvcImpl.OBSERVATION_INDEX);
+
+    Parameters output = ourClient.operation().onType(Observation.class).named("lastn")
+      .withParameter(Parameters.class, "max", new IntegerType(1))
+      .andParameter("subject", new StringType("Patient/" + id.getIdPart()))
+      .execute();
+    Bundle b = (Bundle) output.getParameter().get(0).getResource();
+    assertEquals(1, b.getTotal());
+    assertEquals(obsId, b.getEntry().get(0).getResource().getIdElement().toUnqualifiedVersionless());
+  }
+
+  @BeforeEach
+  void beforeEach() {
+
+    ourCtx = FhirContext.forR4();
+    ourCtx.getRestfulClientFactory().setServerValidationMode(ServerValidationModeEnum.NEVER);
+    ourCtx.getRestfulClientFactory().setSocketTimeout(1200 * 1000);
+    String ourServerBase = "http://localhost:" + port + "/fhir/";
+    ourClient = ourCtx.newRestfulGenericClient(ourServerBase);
+    ourClient.registerInterceptor(new LoggingInterceptor(true));
+  }
+
+  static class Initializer
+    implements ApplicationContextInitializer<ConfigurableApplicationContext> {
+
+    @Override
+    public void initialize(
+      ConfigurableApplicationContext configurableApplicationContext) {
+      // Since the port is dynamically generated, replace the URL with one that has the correct port
+      TestPropertyValues.of("elasticsearch.rest_url=http://localhost:" + embeddedElastic.getHttpPort())
+        .applyTo(configurableApplicationContext.getEnvironment());
+    }
+
+  }
+}