minor refactoring

pact-foundation · Jul 14, 2024 · de83916 · de83916
1 parent 8271134
commit de83916
Showing 1 changed file with 10 additions and 6 deletions.
diff --git a/consumer/junit5/src/test/kotlin/au/com/dius/pact/consumer/junit5/TestTemplateTest.kt b/consumer/junit5/src/test/kotlin/au/com/dius/pact/consumer/junit5/TestTemplateTest.kt
@@ -70,12 +70,16 @@ class TestTemplateTest {
   @TestTemplate
   fun testPactForReservationBooking(message: V4Interaction.AsynchronousMessage) {
     assertThat(message, `is`(notNullValue()))
-    if (message.description == "a purchase started message to book a reservation") {
-      reservationRan = true
-    } else if(message.description == "a cancellation message to cancel a reservation") {
-      cancellationRan = true
-    } else {
-      throw IllegalArgumentException("Unknown message description")
+    when (message.description) {
+        "a purchase started message to book a reservation" -> {
+          reservationRan = true
+        }
+        "a cancellation message to cancel a reservation" -> {
+          cancellationRan = true
+        }
+        else -> {
+          throw IllegalArgumentException("Unknown message description")
+        }
     }
   }
 }