Add identifierString tests to MoreTypesTest

google · Aug 31, 2023 · c4d7e8f · c4d7e8f
1 parent 9a043a9
commit c4d7e8f
Show file tree

Hide file tree

Showing 2 changed files with 31 additions and 6 deletions.
diff --git a/datacapture/src/main/java/com/google/android/fhir/datacapture/extensions/MoreTypes.kt b/datacapture/src/main/java/com/google/android/fhir/datacapture/extensions/MoreTypes.kt
@@ -62,17 +62,19 @@ fun Type.displayString(context: Context): String =
 fun Type.getValueAsString(context: Context): String =
   getValueString(this) ?: context.getString(R.string.not_answered)
 
-/*
+/**
  * Returns the unique identifier of a [Type]. Used to differentiate between item answer options that
  * may have similar display strings
  */
 fun Type.identifierString(context: Context): String =
   id
-    ?: (this as? Coding)?.let {
-      arrayOf("${it.system.orEmpty()}${it.version.orEmpty()}", it.code.orEmpty())
-        .joinToString(if (it.hasSystem() && it.hasCode()) "|" else "")
+    ?: when (this) {
+      is Coding ->
+        arrayOf("${this.system.orEmpty()}${this.version.orEmpty()}", this.code.orEmpty())
+          .joinToString(if (this.hasSystem() && this.hasCode()) "|" else "")
+      is Reference -> this.reference ?: displayString(context)
+      else -> displayString(context)
     }
-      ?: (this as? Reference)?.reference ?: displayString(context)
 
 private fun getDisplayString(type: Type, context: Context): String? =
   when (type) {

diff --git a/datacapture/src/test/java/com/google/android/fhir/datacapture/extensions/MoreTypesTest.kt b/datacapture/src/test/java/com/google/android/fhir/datacapture/extensions/MoreTypesTest.kt
@@ -1,5 +1,5 @@
 /*
- * Copyright 2022 Google LLC
+ * Copyright 2022-2023 Google LLC
  *
  * Licensed under the Apache License, Version 2.0 (the "License");
  * you may not use this file except in compliance with the License.
@@ -16,7 +16,9 @@
 
 package com.google.android.fhir.datacapture.extensions
 
+import android.app.Application
 import android.os.Build
+import androidx.test.core.app.ApplicationProvider
 import ca.uhn.fhir.model.api.TemporalPrecisionEnum
 import com.google.common.truth.Truth.assertThat
 import java.time.Instant
@@ -41,6 +43,7 @@ import org.hl7.fhir.r4.model.MarkdownType
 import org.hl7.fhir.r4.model.OidType
 import org.hl7.fhir.r4.model.PositiveIntType
 import org.hl7.fhir.r4.model.Quantity
+import org.hl7.fhir.r4.model.Reference
 import org.hl7.fhir.r4.model.StringType
 import org.hl7.fhir.r4.model.TimeType
 import org.hl7.fhir.r4.model.Type
@@ -57,6 +60,8 @@ import org.robolectric.annotation.Config
 @Config(sdk = [Build.VERSION_CODES.P])
 class MoreTypesTest {
 
+  private val context = ApplicationProvider.getApplicationContext<Application>()
+
   @Test
   fun instant_shouldReturnExpectedStringValue() {
     val value =
@@ -199,6 +204,24 @@ class MoreTypesTest {
     assertThat(code.equalsDeep(CodeType("fakeCode"))).isTrue()
   }
 
+  @Test
+  fun `Coding should return identifierString containing system, version and code`() {
+    val coding = Coding("fakeSystem", "fakeCode", "fakeDisplay").apply { version = "2.0" }
+    assertThat(coding.identifierString(context)).isEqualTo("fakeSystem2.0|fakeCode")
+  }
+
+  @Test
+  fun `Coding without system should return identifierString containing code`() {
+    val coding = Coding().apply { code = "fakeCode" }
+    assertThat(coding.identifierString(context)).isEqualTo("fakeCode")
+  }
+
+  @Test
+  fun `Reference should return reference`() {
+    val reference = Reference().apply { reference = "fakeReference" }
+    assertThat(reference.identifierString(context)).isEqualTo("fakeReference")
+  }
+
   @Test
   fun `should return calculated value for cqf expression`() {
     val today = LocalDate.now().toString()