Clear compose container references in tearDown

Author: svastven

compose/ui/ui/src/uikitInstrumentedTest/kotlin/androidx/compose/ui/test/UIKitInstrumentedTest.kt

@@ -351,6 +351,8 @@ internal class UIKitInstrumentedTest(
     }
 
     fun tearDown() {
+        clearComposeContainerReferencesIfDetached()
+
         // Stop text editing and hide keyboard if any
         viewController.view.endEditing(force = true)
         waitForIdle()
@@ -364,6 +366,15 @@ internal class UIKitInstrumentedTest(
         hostingViewController?.viewControllerDidLeaveWindowHierarchy()
     }
 
+    private fun clearComposeContainerReferencesIfDetached() {
+        if (hostingView != null && hostingView?.window == null) {
+            hostingView = null
+        }
+        if (hostingViewController != null && hostingViewController?.view?.window == null) {
+            hostingViewController = null
+        }
+    }
+
     private val isIdle: Boolean
         get() {
             val hadSnapshotChanges = Snapshot.current.hasPendingChanges()