1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
|
import gc
import weakref
from typing import AbstractSet
import aio_pika
async def test_leak_unclosed_channel(create_connection):
rabbitmq_connection = await create_connection()
weakset: AbstractSet[aio_pika.abc.AbstractChannel] = weakref.WeakSet()
async def f(rabbitmq_connection: aio_pika.Connection, weakset):
weakset.add(await rabbitmq_connection.channel())
async with rabbitmq_connection:
for i in range(5):
await f(rabbitmq_connection, weakset)
gc.collect()
assert len(tuple(weakset)) == 0
async def test_leak_closed_channel(create_connection):
rabbitmq_connection = await create_connection()
weakset: AbstractSet[aio_pika.abc.AbstractConnection] = weakref.WeakSet()
async def f(rabbitmq_connection: aio_pika.Connection, weakset):
async with rabbitmq_connection.channel() as channel:
weakset.add(channel)
async with rabbitmq_connection:
for i in range(5):
await f(rabbitmq_connection, weakset)
gc.collect()
assert len(tuple(weakset)) == 0
|
