Added helper localMembers count assertions.

ably · Feb 22, 2024 · f6e7449 · f6e7449
1 parent 861920f
commit f6e7449
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/Test/Tests/RealtimeClientPresenceTests.swift b/Test/Tests/RealtimeClientPresenceTests.swift
@@ -2945,6 +2945,8 @@ class RealtimeClientPresenceTests: XCTestCase {
 
         // RTP17i (already attached case)
 
+        XCTAssertEqual(channel.internal.presenceMap.localMembers.count, 2, "p1")
+
         let attachedMessage = ARTProtocolMessage()
         attachedMessage.action = .attached
         attachedMessage.channel = channel.name
@@ -2954,11 +2956,15 @@ class RealtimeClientPresenceTests: XCTestCase {
         sentPresenceMessages = newTransport.protocolMessagesSent.filter({ $0.action == .presence }).compactMap { $0.presence?.first }
         expect(sentPresenceMessages).to(haveCount(2)) // still 2, no presense messages sent
 
+        XCTAssertEqual(channel.internal.presenceMap.localMembers.count, 2, "p2")
+
         attachedMessage.flags = 1 // has presense flag, no resume flag
 
         newTransport.receive(attachedMessage)
         sentPresenceMessages = newTransport.protocolMessagesSent.filter({ $0.action == .presence }).compactMap { $0.presence?.first }
         expect(sentPresenceMessages).to(haveCount(4)) // 2 more presense messages sent
+
+        XCTAssertEqual(channel.internal.presenceMap.localMembers.count, 2, "p3")
     }
 
     func skipped__test__083__Presence__private_and_internal_PresenceMap_containing_only_members_that_match_the_current_connectionId__events_applied_to_presence_map__should_be_applied_to_any_LEAVE_event_with_a_connectionId_that_matches_the_current_client_s_connectionId_and_is_not_a_synthesized() throws {