Update NiaAppStateTest.kt

app/src/androidTest/kotlin/com/google/samples/apps/nowinandroid/ui/NiaAppStateTest.kt

@@ -242,10 +242,10 @@ class NiaAppStateTest {
         }
 
         backgroundScope.launch { state.errorMessage.collect() }
-        state.addErrorMessage("Test Error Message")
+        val id = state.addErrorMessage("Test Error Message")
         assertEquals(
-            "Test Error Message",
+            id,
-            state.errorMessage.value,
+            state.errorMessage.value?.id,
         )
     }
 
@@ -263,11 +263,11 @@ class NiaAppStateTest {
         }
 
         backgroundScope.launch { state.errorMessage.collect() }
-        state.addErrorMessage("Test Error Message 1")
+        val id1 = state.addErrorMessage("Test Error Message 1")
-        state.addErrorMessage("Test Error Message 2")
+        val id2 = state.addErrorMessage("Test Error Message 2")
         assertEquals(
-            "Test Error Message 1",
+            id1,
-            state.errorMessage.value,
+            state.errorMessage.value?.id,
         )
     }
 
@@ -285,13 +285,15 @@ class NiaAppStateTest {
         }
 
         backgroundScope.launch { state.errorMessage.collect() }
-        state.addErrorMessage("Test Error Message")
+        val id = state.addErrorMessage("Test Error Message 1")
-        state.clearErrorMessage()
+        if (id != null) {
+            state.clearErrorMessage(id)
             assertEquals(
                 null,
                 state.errorMessage.value,
             )
         }
+    }
 
     @Test
     fun niaAppState_whenErrorsAreCleared_NextErrorMessageIsPresent() = runTest(UnconfinedTestDispatcher()) {
@@ -307,15 +309,16 @@ class NiaAppStateTest {
         }
 
         backgroundScope.launch { state.errorMessage.collect() }
-        state.addErrorMessage("Test Error Message 1")
+        val id1 = state.addErrorMessage("Test Error Message 1")
-        state.addErrorMessage("Test Error Message 2")
+        val id2 = state.addErrorMessage("Test Error Message 2")
-        state.clearErrorMessage()
+        if (id1 != null) {
-
+            state.clearErrorMessage(id1)
             assertEquals(
-            "Test Error Message 2",
+                id2,
-            state.errorMessage.value,
+                state.errorMessage.value?.id,
             )
         }
+    }
 
     private fun getCompactWindowClass() = WindowSizeClass.calculateFromSize(DpSize(500.dp, 300.dp))
 }