From bace45558df5d75ee5036a72bb1bd2943102e1a8 Mon Sep 17 00:00:00 2001
From: Marat Al <m.alekperov@gmail.com>
Date: Sun, 15 Dec 2024 16:55:23 +0100
Subject: [PATCH] Added tests for occupancy. Ignored CHA-04d since it's
 impossible to emit invalid JSON on a subscription currently.

---
 .../DefaultRoomOccupancyTests.swift           | 101 ++++++++++++++++++
 1 file changed, 101 insertions(+)
 create mode 100644 Tests/AblyChatTests/DefaultRoomOccupancyTests.swift

diff --git a/Tests/AblyChatTests/DefaultRoomOccupancyTests.swift b/Tests/AblyChatTests/DefaultRoomOccupancyTests.swift
new file mode 100644
index 0000000..198a7b1
--- /dev/null
+++ b/Tests/AblyChatTests/DefaultRoomOccupancyTests.swift
@@ -0,0 +1,101 @@
+import Ably
+@testable import AblyChat
+import Testing
+
+struct DefaultRoomOccupancyTests {
+    // @spec CHA-O1
+    @Test
+    func init_channelNameIsSetAsChatMessagesChannelName() async throws {
+        // Given
+        let realtime = MockRealtime.create()
+        let chatAPI = ChatAPI(realtime: realtime)
+        let channel = MockRealtimeChannel(name: "basketball::$chat::$chatMessages")
+        let featureChannel = MockFeatureChannel(channel: channel)
+
+        // When
+        let defaultOccupancy = DefaultOccupancy(featureChannel: featureChannel, chatAPI: chatAPI, roomID: "basketball", logger: TestLogger())
+
+        // Then
+        #expect(defaultOccupancy.channel.name == "basketball::$chat::$chatMessages")
+    }
+
+    // @spec CHA-O2
+    // @spec CHA-O3
+    @Test
+    func requestOccupancyCheck() async throws {
+        // Given
+        let realtime = MockRealtime.create {
+            (MockHTTPPaginatedResponse(
+                items: [
+                    [
+                        "connections": 5,
+                        "presenceMembers": 2,
+                    ],
+                ]
+            ), nil)
+        }
+        let chatAPI = ChatAPI(realtime: realtime)
+        let channel = MockRealtimeChannel(name: "basketball::$chat::$chatMessages")
+        let featureChannel = MockFeatureChannel(channel: channel)
+        let defaultOccupancy = DefaultOccupancy(featureChannel: featureChannel, chatAPI: chatAPI, roomID: "basketball", logger: TestLogger())
+
+        // When
+        let occupancyInfo = try await defaultOccupancy.get()
+
+        // Then
+        #expect(occupancyInfo.connections == 5)
+        #expect(occupancyInfo.presenceMembers == 2)
+    }
+
+    // @spec CHA-O4
+    @Test
+    func usersCanSubscribeToRealtimeOccupancyUpdates() async throws {
+        // Given
+        let realtime = MockRealtime.create()
+        let chatAPI = ChatAPI(realtime: realtime)
+        let channel = MockRealtimeChannel(name: "basketball::$chat::$chatMessages")
+        let featureChannel = MockFeatureChannel(channel: channel)
+        let defaultOccupancy = DefaultOccupancy(featureChannel: featureChannel, chatAPI: chatAPI, roomID: "basketball", logger: TestLogger())
+
+        // CHA-O4a, CHA-O4c
+
+        // When
+        let subscription = await defaultOccupancy.subscribe()
+        subscription.emit(OccupancyEvent(connections: 5, presenceMembers: 2))
+
+        // Then
+        let occupancyInfo = try #require(await subscription.first { _ in true })
+        #expect(occupancyInfo.connections == 5)
+        #expect(occupancyInfo.presenceMembers == 2)
+
+        // CHA-O4b
+
+        // When
+        subscription.unsubscribe()
+        subscription.emit(OccupancyEvent(connections: 5, presenceMembers: 2))
+
+        // Then
+        let nilOccupancyInfo = await subscription.first { _ in true }
+        #expect(nilOccupancyInfo == nil)
+    }
+
+    // @spec CHA-O5
+    @Test
+    func onDiscontinuity() async throws {
+        // Given
+        let realtime = MockRealtime.create()
+        let chatAPI = ChatAPI(realtime: realtime)
+        let channel = MockRealtimeChannel(name: "basketball::$chat::$chatMessages")
+        let featureChannel = MockFeatureChannel(channel: channel)
+        let defaultOccupancy = DefaultOccupancy(featureChannel: featureChannel, chatAPI: chatAPI, roomID: "basketball", logger: TestLogger())
+
+        // When: The feature channel emits a discontinuity through `onDiscontinuity`
+        let featureChannelDiscontinuity = DiscontinuityEvent(error: ARTErrorInfo.createUnknownError()) // arbitrary error
+        let discontinuitySubscription = await defaultOccupancy.onDiscontinuity()
+        await featureChannel.emitDiscontinuity(featureChannelDiscontinuity)
+
+        // Then: The DefaultOccupancy instance emits this discontinuity through `onDiscontinuity`
+        let discontinuity = try #require(await discontinuitySubscription.first { _ in true })
+        #expect(discontinuity == featureChannelDiscontinuity)
+    }
+}