Test showing include on canonical version does not work

trifork · Apr 9, 2024 · cc441be · cc441be
1 parent 107de2c
commit cc441be
Showing 1 changed file with 36 additions and 0 deletions.
diff --git a/...test-r4/src/test/java/ca/uhn/fhir/jpa/dao/r4/FhirResourceDaoR4VersionedReferenceTest.java b/...test-r4/src/test/java/ca/uhn/fhir/jpa/dao/r4/FhirResourceDaoR4VersionedReferenceTest.java
@@ -4,11 +4,15 @@
 import ca.uhn.fhir.jpa.api.model.DaoMethodOutcome;
 import ca.uhn.fhir.jpa.searchparam.SearchParameterMap;
 import ca.uhn.fhir.jpa.test.BaseJpaR4Test;
+import ca.uhn.fhir.model.api.Include;
 import ca.uhn.fhir.model.primitive.IdDt;
 import ca.uhn.fhir.rest.api.server.IBundleProvider;
 import ca.uhn.fhir.rest.api.server.SystemRequestDetails;
+import ca.uhn.fhir.rest.param.StringParam;
 import ca.uhn.fhir.rest.param.TokenParam;
+import ca.uhn.fhir.rest.param.UriParam;
 import ca.uhn.fhir.rest.server.exceptions.InvalidRequestException;
+
 import ca.uhn.fhir.rest.server.servlet.ServletRequestDetails;
 import ca.uhn.fhir.util.BundleBuilder;
 import org.hl7.fhir.instance.model.api.IBaseResource;
@@ -24,6 +28,8 @@
 import org.hl7.fhir.r4.model.Observation;
 import org.hl7.fhir.r4.model.Organization;
 import org.hl7.fhir.r4.model.Patient;
+import org.hl7.fhir.r4.model.Questionnaire;
+import org.hl7.fhir.r4.model.QuestionnaireResponse;
 import org.hl7.fhir.r4.model.Reference;
 import org.hl7.fhir.r4.model.StringType;
 import org.hl7.fhir.r4.model.Task;
@@ -50,7 +56,9 @@
 import static ca.uhn.fhir.util.HapiExtensions.EXTENSION_AUTO_VERSION_REFERENCES_AT_PATH;
 import static org.hamcrest.CoreMatchers.is;
 import static org.hamcrest.MatcherAssert.assertThat;
+import static org.hamcrest.Matchers.containsInAnyOrder;
 import static org.hamcrest.Matchers.equalTo;
+import static org.hamcrest.Matchers.hasSize;
 import static org.hamcrest.Matchers.matchesPattern;
 import static org.junit.jupiter.api.Assertions.assertEquals;
 import static org.junit.jupiter.api.Assertions.assertNotNull;
@@ -1056,4 +1064,32 @@ public void bundleTransaction_withRequestURLWithPrecedingSlash_processesAsExpect
 		Observation obs = myObservationDao.read(idType);
 		Assertions.assertNotNull(obs);
 	}
+
+	@Test
+	public void test_canonical_lookup(){
+		String canonicalUrl = "http://foo";
+		Questionnaire questionnaire = (Questionnaire) myQuestionnaireDao.create(new Questionnaire().setUrl(canonicalUrl)).getResource();
+		QuestionnaireResponse questionnaireResponse = (QuestionnaireResponse) myQuestionnaireResponseDao.create(new QuestionnaireResponse().setQuestionnaire(canonicalUrl)).getResource();
+		List<IBaseResource> result = myQuestionnaireResponseDao.search(
+			SearchParameterMap.newSynchronous("_id", new StringParam(questionnaireResponse.getIdElement().getIdPart()))
+				.addInclude(new Include("QuestionnaireResponse:questionnaire"))).getAllResources();
+		assertThat(result, hasSize(2));
+		assertThat(questionnaireResponse.getId(), equalTo(result.get(0).getIdElement().getValue()));
+		assertThat(questionnaire.getId(), equalTo(result.get(1).getIdElement().getValue()));
+		// If one could only do ... assertThat(result, containsInAnyOrder(questionnaireResponse, questionnaire));
+	}
+
+	@Test
+	public void test_canonical_versioned_lookup(){
+		String canonicalUrl = "http://foo";
+		String version = "2.4";
+		Questionnaire questionnaire = (Questionnaire) myQuestionnaireDao.create(new Questionnaire().setUrl(canonicalUrl).setVersion(version)).getResource();
+		QuestionnaireResponse questionnaireResponse = (QuestionnaireResponse) myQuestionnaireResponseDao.create(new QuestionnaireResponse().setQuestionnaire(canonicalUrl + "|" + version)).getResource();
+		List<IBaseResource> result = myQuestionnaireResponseDao.search(
+			SearchParameterMap.newSynchronous("_id", new StringParam(questionnaireResponse.getIdElement().getIdPart()))
+				.addInclude(new Include("QuestionnaireResponse:questionnaire"))).getAllResources();
+		assertThat(result, hasSize(2));
+		assertThat(questionnaireResponse.getId(), equalTo(result.get(0).getIdElement().getValue()));
+		assertThat(questionnaire.getId(), equalTo(result.get(1).getIdElement().getValue()));
+	}
 }