File tree Expand file tree Collapse file tree 2 files changed +9
-3
lines changed Expand file tree Collapse file tree 2 files changed +9
-3
lines changed Original file line number Diff line number Diff line change @@ -5,3 +5,4 @@ pytest==4.3.1
55six == 1.12.0
66trio == 0.11.0
77pytest-trio == 0.5.2
8+ async-generator == 1.10
Original file line number Diff line number Diff line change 1+ from collections import namedtuple
12import os
23import sys
3- from collections import namedtuple
4+
5+ from async_generator import async_generator , yield_
46import pytest
57from rethinkdb import RethinkDB
68from rethinkdb .errors import ReqlRuntimeError
79import trio
810
11+
912INTEGRATION_TEST_DB = 'integration_test'
1013r = RethinkDB ()
1114r .set_loop_type ('trio' )
1215
1316
1417@pytest .fixture
18+ @async_generator
1519async def integration_db (nursery ):
1620 async with r .open (db = 'test' , nursery = nursery ) as conn :
1721 try :
1822 await r .db_create (INTEGRATION_TEST_DB ).run (conn )
1923 except ReqlRuntimeError :
2024 pass
21- yield r .db (INTEGRATION_TEST_DB )
25+ await yield_ ( r .db (INTEGRATION_TEST_DB ) )
2226
2327
2428@pytest .fixture
29+ @async_generator
2530async def marvel_table (integration_db , nursery ):
2631 async with r .open (db = 'test' , nursery = nursery ) as conn :
2732 await r .table_create ('marvel' ).run (conn )
28- yield r .table ('marvel' )
33+ await yield_ ( r .table ('marvel' ) )
2934 await r .table_drop ('marvel' ).run (conn )
3035
3136
You can’t perform that action at this time.
0 commit comments