1 changed files with 24 additions and 0 deletions
@ -0,0 +1,24 @@ |
|||
import asyncio |
|||
from contextlib import asynccontextmanager |
|||
from unittest.mock import Mock, patch |
|||
|
|||
from docs_src.dependencies.tutorial007_py39 import get_db |
|||
|
|||
|
|||
def test_get_db(): # Just for coverage |
|||
async def test_async_gen(): |
|||
cm = asynccontextmanager(get_db) |
|||
async with cm() as db_session: |
|||
return db_session |
|||
|
|||
dbsession_moock = Mock() |
|||
|
|||
with patch( |
|||
"docs_src.dependencies.tutorial007_py39.DBSession", |
|||
return_value=dbsession_moock, |
|||
create=True, |
|||
): |
|||
value = asyncio.run(test_async_gen()) |
|||
|
|||
assert value is dbsession_moock |
|||
dbsession_moock.close.assert_called_once() |
|||
Loading…
Reference in new issue