From 30ad922ce4c9d9b088f0a64f6d26046a4ca114c1 Mon Sep 17 00:00:00 2001
From: sacOO7 <sachinshinde7676@gmail.com>
Date: Tue, 5 Nov 2024 20:47:22 +0530
Subject: [PATCH] Refactored RoomLifecycleManagerTest, added more assertions
 for CHA-RL1d

---
 .../com/ably/chat/AtomicCoroutineScope.kt     |  5 ++--
 .../com/ably/chat/RoomLifecycleManagerTest.kt | 24 +++++++++++--------
 2 files changed, 17 insertions(+), 12 deletions(-)

diff --git a/chat-android/src/main/java/com/ably/chat/AtomicCoroutineScope.kt b/chat-android/src/main/java/com/ably/chat/AtomicCoroutineScope.kt
index ad215f5..001781f 100644
--- a/chat-android/src/main/java/com/ably/chat/AtomicCoroutineScope.kt
+++ b/chat-android/src/main/java/com/ably/chat/AtomicCoroutineScope.kt
@@ -1,6 +1,5 @@
 package com.ably.chat
 
-import io.ably.annotation.Experimental
 import java.util.concurrent.PriorityBlockingQueue
 import kotlinx.coroutines.CompletableDeferred
 import kotlinx.coroutines.CoroutineScope
@@ -76,7 +75,9 @@ class AtomicCoroutineScope(private val scope: CoroutineScope = CoroutineScope(Di
         }
     }
 
-    @Experimental
     val finishedProcessing: Boolean
         get() = jobs.isEmpty() && !isRunning
+
+    val queuedJobs: Int
+        get() = jobs.count()
 }
diff --git a/chat-android/src/test/java/com/ably/chat/RoomLifecycleManagerTest.kt b/chat-android/src/test/java/com/ably/chat/RoomLifecycleManagerTest.kt
index 7c07454..cc19b16 100644
--- a/chat-android/src/test/java/com/ably/chat/RoomLifecycleManagerTest.kt
+++ b/chat-android/src/test/java/com/ably/chat/RoomLifecycleManagerTest.kt
@@ -66,36 +66,40 @@ class RoomLifecycleManagerTest {
 
     @Test
     fun `(CHA-RL1d) Attach op should wait for existing operation as per (CHA-RL7)`() = runTest {
-        val status = spyk<DefaultStatus>().apply {
-            setStatus(RoomLifecycle.Released)
-        }
+        val status = spyk<DefaultStatus>()
+        Assert.assertEquals(RoomLifecycle.Initializing, status.current)
+
         val roomLifecycle = spyk(RoomLifecycleManager(roomScope, status, emptyList()))
 
-        val channelReleased = Channel<Unit>()
+        val roomReleased = Channel<Boolean>()
         coEvery {
             roomLifecycle.release()
         } coAnswers {
             roomLifecycle.atomicCoroutineScope.async {
                 status.setStatus(RoomLifecycle.Releasing)
-                channelReleased.receive()
+                roomReleased.receive()
                 status.setStatus(RoomLifecycle.Released)
             }
         }
-        launch { roomLifecycle.release() }
 
-        // Release op started
+        // Release op started from separate coroutine
+        launch { roomLifecycle.release() }
         assertWaiter { !roomLifecycle.atomicCoroutineScope.finishedProcessing }
+        Assert.assertEquals(0, roomLifecycle.atomicCoroutineScope.queuedJobs) // no queued jobs, one job running
         assertWaiter { status.current == RoomLifecycle.Releasing }
 
+        // Attach op started from separate coroutine
         val roomAttachOpDeferred = async(SupervisorJob()) { roomLifecycle.attach() }
+        assertWaiter { roomLifecycle.atomicCoroutineScope.queuedJobs == 1 } // attach op queued
         Assert.assertEquals(RoomLifecycle.Releasing, status.current)
-        channelReleased.send(Unit)
 
-        // Release op finished
-        assertWaiter { roomLifecycle.atomicCoroutineScope.finishedProcessing }
+        // Finish release op, so ATTACH op can start
+        roomReleased.send(true)
         assertWaiter { status.current == RoomLifecycle.Released }
 
         val result = kotlin.runCatching { roomAttachOpDeferred.await() }
+        Assert.assertTrue(roomLifecycle.atomicCoroutineScope.finishedProcessing)
+
         Assert.assertTrue(result.isFailure)
         val exception = result.exceptionOrNull() as AblyException