File tree Expand file tree Collapse file tree 1 file changed +3
-3
lines changed
Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Original file line number Diff line number Diff line change 66from collections .abc import AsyncGenerator , Callable , Generator
77from queue import Queue
88from typing import Any
9- from unittest .mock import AsyncMock , Mock , patch
9+ from unittest .mock import AsyncMock , MagicMock , Mock , patch
1010
1111import pytest
1212from aioresponses import aioresponses
@@ -360,12 +360,12 @@ def __init__(self):
360360 self .publish = AsyncMock (side_effect = self ._publish )
361361 self .subscribe = AsyncMock (side_effect = self ._subscribe )
362362 self .connect = AsyncMock (side_effect = self ._connect )
363- self .close = AsyncMock (side_effect = self ._close )
363+ self .close = MagicMock (side_effect = self ._close )
364364
365365 async def _connect (self ) -> None :
366366 self ._is_connected = True
367367
368- async def _close (self ) -> None :
368+ def _close (self ) -> None :
369369 self ._is_connected = False
370370
371371 @property
You can’t perform that action at this time.
0 commit comments