Add test for ParcelableAnnotationParam

PiperOrigin-RevId: 344837912

javatests/arcs/android/host/parcelables/ParcelableAnnotationParamTest.kt

@@ -0,0 +1,92 @@
+/*
+ * Copyright 2020 Google LLC.
+ *
+ * This code may only be used under the BSD style license found at
+ * http://polymer.github.io/LICENSE.txt
+ *
+ * Code distributed by Google as part of this project is also subject to an additional IP rights
+ * grant found at
+ * http://polymer.github.io/PATENTS.txt
+ */
+
+package arcs.android.host.parcelables
+
+import android.os.Parcel
+import androidx.test.ext.junit.runners.AndroidJUnit4
+import arcs.core.data.AnnotationParam
+import com.google.common.truth.Truth.assertThat
+import kotlin.test.assertFailsWith
+import org.junit.Test
+import org.junit.runner.RunWith
+
+/** Tests for [ParcelableAnnotationParam]'s classes. */
+@RunWith(AndroidJUnit4::class)
+class ParcelableAnnotationParamTest {
+
+  @Test
+  fun annotationParamString_parcelableRoundTrip_works() {
+    val annotationParam = AnnotationParam.Str("string")
+
+    val marshalled = with(Parcel.obtain()) {
+      writeTypedObject(annotationParam.toParcelable(), 0)
+      marshall()
+    }
+
+    val unmarshalled = with(Parcel.obtain()) {
+      unmarshall(marshalled, 0, marshalled.size)
+      setDataPosition(0)
+      readTypedObject(requireNotNull(ParcelableAnnotationParam.CREATOR))
+    }
+
+    assertThat(unmarshalled?.describeContents()).isEqualTo(0)
+    assertThat(unmarshalled?.actual).isEqualTo(annotationParam)
+  }
+
+  @Test
+  fun readTypedObject_malformedWrongTypeParcel_fails() {
+    val malformedParcel =
+      Parcel.obtain().apply {
+        writeInt(1) // Says value to come is non-empty.
+        writeString("Bogus") // Should be one of "Str", "Bool", "Num"
+        writeString("value")
+      }
+
+    malformedParcel.setDataPosition(0)
+    assertFailsWith<IllegalStateException>("unknown annotation param type") {
+      malformedParcel.readTypedObject(requireNotNull(ParcelableAnnotationParam))
+    }
+  }
+
+  @Test
+  fun readTypedObject_malformedParcel_fails() {
+    val malformedParcel =
+      Parcel.obtain().apply {
+        writeInt(1) // Says value to come is non-empty.
+        writeString(null)
+      }
+
+    malformedParcel.setDataPosition(0)
+    assertFailsWith<IllegalArgumentException>("no name found") {
+      malformedParcel.readTypedObject(requireNotNull(ParcelableAnnotationParam))
+    }
+  }
+
+  @Test
+  fun array_parcelableRoundTrip_works() {
+    val annotationParam = AnnotationParam.Str("string").toParcelable()
+    val arr = arrayOf(annotationParam, annotationParam)
+    val marshalled = with(Parcel.obtain()) {
+      writeTypedArray(arr, 0)
+      marshall()
+    }
+
+    val unmarshalled = with(Parcel.obtain()) {
+      unmarshall(marshalled, 0, marshalled.size)
+      setDataPosition(0)
+      createTypedArray(requireNotNull(ParcelableAnnotationParam.CREATOR))
+    }
+
+    assertThat(unmarshalled?.size).isEqualTo(2)
+    assertThat(unmarshalled?.get(0)?.actual).isEqualTo(annotationParam.actual)
+  }
+}