chore: use time.sleep in rest tests

Author: ttypic

ably/scripts/unasync.py

@@ -182,15 +182,50 @@ def _replace_import(self, tokens, token_counter, new_tokens: list):
 
         full_lib_name = ''
         lib_name_counter = token_counter + 2
-        if len(_IMPORTS_REPLACE.keys()) == 0:
-            return lib_name_counter
-
-        while True:
-            if tokens[lib_name_counter].src == " ":
-                break
+        
+        # Handle special case for "from asyncio import sleep" -> "from time import sleep"
+        # First check if this is an asyncio import with just sleep
+        if (lib_name_counter < len(tokens) and 
+            tokens[lib_name_counter].src == "asyncio"):
+            
+            # Look ahead to see if it's "import sleep" specifically
+            check_counter = lib_name_counter + 1
+            # Skip whitespace  
+            while check_counter < len(tokens) and tokens[check_counter].src == " ":
+                check_counter += 1
+            # Check if next is "import"
+            if check_counter < len(tokens) and tokens[check_counter].src == "import":
+                check_counter += 1
+                # Skip whitespace after "import"
+                while check_counter < len(tokens) and tokens[check_counter].src == " ":
+                    check_counter += 1
+                # Check if next is exactly "sleep"
+                if (check_counter < len(tokens) and tokens[check_counter].src == "sleep"):
+                    # Check what comes after "sleep" to determine if this is a standalone sleep import
+                    next_check = check_counter + 1
+                    is_sleep_only = True
+                    
+                    # Skip any whitespace
+                    while next_check < len(tokens) and tokens[next_check].src == " ":
+                        next_check += 1
+                    
+                    if next_check < len(tokens):
+                        next_token = tokens[next_check]
+                        # If there's a comma, "as", or other import items, it's not a standalone sleep import
+                        if next_token.src in [",", "as"] or next_token.name == "NAME":
+                            is_sleep_only = False
+                    
+                    if is_sleep_only:
+                        # Replace "asyncio" with "time" only for standalone "sleep" import
+                        new_tokens.append(tokenize_rt.Token("NAME", "time"))
+                        return lib_name_counter + 1  # Skip past "asyncio"
+        
+        # Parse the full module name for other cases
+        while lib_name_counter < len(tokens) and tokens[lib_name_counter].src != " ":
             full_lib_name = full_lib_name + tokens[lib_name_counter].src
             lib_name_counter = lib_name_counter + 1
-
+        
+        # Handle existing _IMPORTS_REPLACE logic
         for key, value in _IMPORTS_REPLACE.items():
             if key in full_lib_name:
                 updated_lib_name = full_lib_name.replace(key, value)
@@ -201,7 +236,14 @@ def _replace_import(self, tokens, token_counter, new_tokens: list):
                 new_tokens.pop()
                 return lib_name_counter
 
-        lib_name_counter = token_counter + 2
+        # For all other cases, add the original module name tokens
+        for lib_name_part in full_lib_name.split("."):
+            lib_name_part = self._class_rename(lib_name_part)
+            new_tokens.append(tokenize_rt.Token("NAME", lib_name_part))
+            new_tokens.append(tokenize_rt.Token("OP", "."))
+        if full_lib_name:  # Only remove the last dot if we added tokens
+            new_tokens.pop()
+        
         return lib_name_counter
 
     def _class_rename(self, name):

test/ably/rest/restchannelpublish_test.py

@@ -9,7 +9,6 @@
 import mock
 import msgpack
 import pytest
-import asyncio
 
 from ably import api_version
 from ably import AblyException, IncompatibleClientIdException
@@ -18,6 +17,7 @@
 from ably.types.tokendetails import TokenDetails
 from ably.util import case
 from test.ably import utils
+from asyncio import sleep
 
 from test.ably.testapp import TestApp
 from test.ably.utils import VaryByProtocolTestsMetaclass, dont_vary_protocol, BaseAsyncTestCase
@@ -404,7 +404,7 @@ async def test_interoperability(self):
                 # 1)
                 await channel.publish(data=expected_value)
                 # temporary added delay, we need to investigate why messages don't appear immediately
-                await asyncio.sleep(1)
+                await sleep(1)
                 async with httpx.AsyncClient(http2=True) as client:
                     r = await client.get(url, auth=auth)
                 item = r.json()[0]