fix: test fixes

momentohq · Jul 13, 2023 · 30198ef · 30198ef
1 parent ba81a67
commit 30198ef
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/tests/momento/topic_client/test_topics.py b/tests/momento/topic_client/test_topics.py
@@ -58,7 +58,7 @@ def subscribe_happy_path(client: CacheClient, topic_client: TopicClient, cache_n
 
         _ = topic_client.publish(cache_name, topic_name=topic, value=value)
 
-        item_response = subscribe_response.__next__()
+        item_response = next(subscribe_response)
         assert isinstance(item_response, TopicSubscriptionItem.Success)
         assert item_response.value_string == value
 

diff --git a/tests/momento/topic_client/test_topics_async.py b/tests/momento/topic_client/test_topics_async.py
@@ -60,7 +60,7 @@ async def subscribe_happy_path(
         subscribe_response = await topic_client_async.subscribe(cache_name, topic_name=topic)
         assert isinstance(subscribe_response, TopicSubscribe.SubscriptionAsync)
 
-        item_task = subscribe_response.__anext__()
+        item_task = next(subscribe_response)
         publish_response = await topic_client_async.publish(cache_name, topic_name=topic, value=value)
 
         print(publish_response)