@@ -17,8 +17,8 @@ async def test_read_batch(self, driver, topic_with_messages, topic_consumer):
1717 await reader .close ()
1818
1919 async def test_link_to_client (self , driver , topic_path , topic_consumer ):
20- reader = driver .topic_client .reader (topic_path , topic_consumer )
21- assert reader ._parent is driver .topic_client
20+ async with driver .topic_client .reader (topic_path , topic_consumer ) as reader :
21+ assert reader ._parent is driver .topic_client
2222
2323 async def test_read_message (self , driver , topic_with_messages , topic_consumer ):
2424 reader = driver .topic_client .reader (topic_with_messages , topic_consumer )
@@ -138,8 +138,8 @@ def test_read_batch(self, driver_sync, topic_with_messages, topic_consumer):
138138 reader .close ()
139139
140140 def test_link_to_client (self , driver_sync , topic_path , topic_consumer ):
141- reader = driver_sync .topic_client .reader (topic_path , topic_consumer )
142- assert reader ._parent is driver_sync .topic_client
141+ with driver_sync .topic_client .reader (topic_path , topic_consumer ) as reader :
142+ assert reader ._parent is driver_sync .topic_client
143143
144144 def test_read_message (self , driver_sync , topic_with_messages , topic_consumer ):
145145 reader = driver_sync .topic_client .reader (topic_with_messages , topic_consumer )
0 commit comments