Removed async_generator test dependency. (#321)

Author: carltongibson

setup.py

@@ -12,7 +12,6 @@
 test_requires = crypto_requires + [
     "pytest",
     "pytest-asyncio==0.14.0",
-    "async_generator",
     "async-timeout",
 ]
 

tests/test_core.py

@@ -3,7 +3,6 @@
 
 import async_timeout
 import pytest
-from async_generator import async_generator, yield_
 
 from asgiref.sync import async_to_sync
 from channels_redis.core import ChannelFull, RedisChannelLayer
@@ -55,26 +54,24 @@ async def group_send_three_messages_with_delay(group_name, channel_layer, delay)
 
 
 @pytest.fixture()
-@async_generator
 async def channel_layer():
     """
     Channel layer fixture that flushes automatically.
     """
     channel_layer = RedisChannelLayer(
         hosts=TEST_HOSTS, capacity=3, channel_capacity={"tiny": 1}
     )
-    await yield_(channel_layer)
+    yield channel_layer
     await channel_layer.flush()
 
 
 @pytest.fixture()
-@async_generator
 async def channel_layer_multiple_hosts():
     """
     Channel layer fixture that flushes automatically.
     """
     channel_layer = RedisChannelLayer(hosts=MULTIPLE_TEST_HOSTS, capacity=3)
-    await yield_(channel_layer)
+    yield channel_layer
     await channel_layer.flush()
 
 

tests/test_pubsub.py

@@ -5,7 +5,6 @@
 
 import async_timeout
 import pytest
-from async_generator import async_generator, yield_
 
 from asgiref.sync import async_to_sync
 from channels_redis.pubsub import RedisPubSubChannelLayer
@@ -14,24 +13,22 @@
 
 
 @pytest.fixture()
-@async_generator
 async def channel_layer():
     """
     Channel layer fixture that flushes automatically.
     """
     channel_layer = RedisPubSubChannelLayer(hosts=TEST_HOSTS)
-    await yield_(channel_layer)
+    yield channel_layer
     await channel_layer.flush()
 
 
 @pytest.fixture()
-@async_generator
 async def other_channel_layer():
     """
     Channel layer fixture that flushes automatically.
     """
     channel_layer = RedisPubSubChannelLayer(hosts=TEST_HOSTS)
-    await yield_(channel_layer)
+    yield channel_layer
     await channel_layer.flush()
 
 
@@ -47,7 +44,6 @@ async def test_send_receive(channel_layer):
     assert message["text"] == "Ahoy-hoy!"
 
 
-@pytest.mark.asyncio
 def test_send_receive_sync(channel_layer, event_loop):
     _await = event_loop.run_until_complete
     channel = _await(channel_layer.new_channel())
@@ -73,7 +69,6 @@ async def test_multi_send_receive(channel_layer):
     assert (await channel_layer.receive(channel))["type"] == "message.3"
 
 
-@pytest.mark.asyncio
 def test_multi_send_receive_sync(channel_layer, event_loop):
     _await = event_loop.run_until_complete
     channel = _await(channel_layer.new_channel())

tests/test_pubsub_sentinel.py

@@ -3,7 +3,6 @@
 
 import async_timeout
 import pytest
-from async_generator import async_generator, yield_
 
 from asgiref.sync import async_to_sync
 from channels_redis.pubsub import RedisPubSubChannelLayer
@@ -13,13 +12,12 @@
 
 
 @pytest.fixture()
-@async_generator
 async def channel_layer():
     """
     Channel layer fixture that flushes automatically.
     """
     channel_layer = RedisPubSubChannelLayer(hosts=TEST_HOSTS)
-    await yield_(channel_layer)
+    yield channel_layer
     await channel_layer.flush()
 
 

tests/test_sentinel.py

@@ -3,7 +3,6 @@
 
 import async_timeout
 import pytest
-from async_generator import async_generator, yield_
 
 from asgiref.sync import async_to_sync
 from channels_redis.core import ChannelFull, RedisChannelLayer
@@ -56,26 +55,24 @@ async def group_send_three_messages_with_delay(group_name, channel_layer, delay)
 
 
 @pytest.fixture()
-@async_generator
 async def channel_layer():
     """
     Channel layer fixture that flushes automatically.
     """
     channel_layer = RedisChannelLayer(
         hosts=TEST_HOSTS, capacity=3, channel_capacity={"tiny": 1}
     )
-    await yield_(channel_layer)
+    yield channel_layer
     await channel_layer.flush()
 
 
 @pytest.fixture()
-@async_generator
 async def channel_layer_multiple_hosts():
     """
     Channel layer fixture that flushes automatically.
     """
     channel_layer = RedisChannelLayer(hosts=MULTIPLE_TEST_HOSTS, capacity=3)
-    await yield_(channel_layer)
+    yield channel_layer
     await channel_layer.flush()