Improve waitForMapClear to check expected removed keys

Instead of waiting for any update, waitForMapClear now takes an
expectedRemovedKeys parameter and waits for an update where all
specified keys have .removed status. This serves as a better proxy
for the JS operation action check.

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>

Author: lawrence-forooghian

Tests/AblyLiveObjectsTests/JS Integration Tests/ObjectsIntegrationTests.swift

@@ -96,14 +96,17 @@ func waitForCounterUpdate(_ updates: AsyncStream<LiveCounterUpdate>) async {
     _ = await updates.first { _ in true }
 }
 
-/// Waits for a MAP_CLEAR operation to be applied to a LiveMap.
+/// Waits for a MAP_CLEAR operation to be applied to a LiveMap, by waiting for an update where
+/// all of the specified keys have `.removed` status.
 ///
 /// The JS equivalent subscribes and checks `message.operation.action === 'map.clear'`, but Swift's
 /// `LiveMapUpdate` doesn't expose the operation action — it only provides per-key changes via
-/// `update: [String: LiveMapUpdateAction]`. Since callers of this helper only send a MAP_CLEAR
-/// operation, waiting for any update on the stream is sufficient.
-func waitForMapClear(_ updates: AsyncStream<LiveMapUpdate>) async {
-    _ = await updates.first { _ in true }
+/// `update: [String: LiveMapUpdateAction]`. So we use the per-key `.removed` entries as a proxy
+/// instead.
+func waitForMapClear(_ updates: AsyncStream<LiveMapUpdate>, expectedRemovedKeys: Set<String>) async {
+    _ = await updates.first { update in
+        expectedRemovedKeys.allSatisfy { update.update[$0] == .removed }
+    }
 }
 
 // Note that Cursor decided to implement this in a different way to the waitForObjectSync that I'd already implemented; TODO pick one of the two approaches (this one might be cleaner).
@@ -2597,7 +2600,7 @@ private struct ObjectsIntegrationTests {
 
                         // send MAP_CLEAR
                         let clearAppliedPromiseUpdates = try root.updates()
-                        async let clearAppliedPromise: Void = waitForMapClear(clearAppliedPromiseUpdates)
+                        async let clearAppliedPromise: Void = waitForMapClear(clearAppliedPromiseUpdates, expectedRemovedKeys: ["foo", "baz"])
                         try await objectsHelper.sendMapClearOnChannel(objects: objects, objectId: "root")
                         await clearAppliedPromise