Logs all state changes so you can find out what causes a self-invalidating loop in your app in case you keep running into timeouts in your tests

ObserveStateChangesCreateComposeRule.kt

// Use this template if you use ComposeTestRule (createComposeRule())
// Use the other template if you use ComposeUiTest (runComposeUiTest())

private val TAG = "TEST_TAG"
private var I = 0 // add a sequence number to every log message for disambiguation
private var unregisterApplyObserver: ObserverHandle? = null

@Test
fun test() {
    // your test here
}

@Before
fun setup() {
    unregisterApplyObserver = Snapshot.registerApplyObserver { mutatedObjects, snapshot ->
        Log.i(TAG, "[${I++}] applying snapshot")
        for (obj in mutatedObjects) {
            Log.d(TAG, "[${I++}] object $obj modified")
        }
    }
}

@After
fun tearDown() {
    unregisterApplyObserver?.dispose()
}

ObserveStateChangesRunComposeUiTest.kt

// Use this template if you use ComposeUiTest (runComposeUiTest())
// Use the other template if you use ComposeTestRule (createComposeRule())

private val TAG = "TEST_TAG"
private var I = 0 // add a sequence number to every log message for disambiguation

@Test
fun test() = runComposeUiTest {
    withApplyObserver {
        // your test here
    }
}

private inline fun withApplyObserver(block: () -> Any) {
    var unregisterApplyObserver: ObserverHandle? = null
    try {
        unregisterApplyObserver = Snapshot.registerApplyObserver { mutatedObjects, snapshot ->
            Log.i(TAG, "[${I++}] applying snapshot")
            for (obj in mutatedObjects) {
                Log.d(TAG, "[${I++}] object $obj modified")
            }
        }
        block()
    } finally {
        unregisterApplyObserver?.dispose()
    }
}