From ddb3e1833948be20be718c48bc005f26a9e17e23 Mon Sep 17 00:00:00 2001
From: Marat Al <m.alekperov@gmail.com>
Date: Sun, 19 Jan 2025 19:28:27 +0100
Subject: [PATCH] Added occupancy tests.

---
 .../DefaultRoomOccupancyTests.swift           | 75 +++++++++++++++++++
 1 file changed, 75 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..572f154
--- /dev/null
+++ b/Tests/AblyChatTests/DefaultRoomOccupancyTests.swift
@@ -0,0 +1,75 @@
+import Ably
+@testable import AblyChat
+import Testing
+
+struct DefaultRoomOccupancyTests {
+    // @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-O4a
+    // @spec CHA-O4c
+    @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)
+    }
+
+    // @spec CHA-O5
+    @Test
+    func onDiscontinuity() async throws {
+        // Given
+        let realtime = MockRealtime.create()
+        let chatAPI = ChatAPI(realtime: realtime)
+        let channel = MockRealtimeChannel()
+        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)
+    }
+}