Update channel tests - takeComplete()

cashapp · Aug 1, 2024 · e1079b9 · e1079b9
1 parent 4375456
commit e1079b9
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/src/commonTest/kotlin/app/cash/turbine/ChannelTest.kt b/src/commonTest/kotlin/app/cash/turbine/ChannelTest.kt
@@ -261,6 +261,19 @@ class ChannelTest {
     assertEquals("Expected item but found Error(CustomThrowable)", actual.message)
     assertSame(error, actual.cause)
   }
+
+  @Test fun takeComplete() = withTestScope {
+    val channel = channelOf("item!")
+    channel.takeItem()
+    channel.takeComplete()
+  }
+
+  @Test fun takeCompleteFailsForItem() = withTestScope {
+    val channel = channelOf("item!")
+    assertFailsWith<AssertionError> {
+      channel.takeComplete()
+    }
+  }
 
   @Test
   fun expectMostRecentItemButNoItemWasFoundThrowsWithName() = runTest {