@@ -321,7 +321,7 @@ async def geocoding_view(request):
321321 return json_response (loc , status = status )
322322
323323
324- @pytest .yield_fixture (name = 'redis' )
324+ @pytest .fixture (name = 'redis' )
325325async def _fix_redis (settings ):
326326 addr = settings .redis_settings .host , settings .redis_settings .port
327327
@@ -334,7 +334,7 @@ async def _fix_redis(settings):
334334 await redis .wait_closed ()
335335
336336
337- @pytest .yield_fixture (name = 'worker_ctx' )
337+ @pytest .fixture (name = 'worker_ctx' )
338338async def _fix_worker_ctx (redis , settings , db_conn ):
339339 session = ClientSession (timeout = ClientTimeout (total = 10 ))
340340 ctx = dict (settings = settings , pg_engine = MockEngine (db_conn ), session = session , redis = redis )
@@ -344,7 +344,7 @@ async def _fix_worker_ctx(redis, settings, db_conn):
344344 await session .close ()
345345
346346
347- @pytest .yield_fixture (name = 'worker' )
347+ @pytest .fixture (name = 'worker' )
348348async def _fix_worker (redis , worker_ctx ):
349349 worker = Worker (functions = WorkerSettings .functions , redis_pool = redis , burst = True , poll_delay = 0.01 , ctx = worker_ctx )
350350
@@ -383,7 +383,7 @@ def image_download_url(other_server):
383383@pytest .fixture
384384def settings (other_server ):
385385 return Settings (
386- pg_dsn = os .getenv ('DATABASE_URL' , DB_DSN ),
386+ database_url = os .getenv ('DATABASE_URL' , DB_DSN ),
387387 redis_database = 7 ,
388388 master_key = MASTER_KEY ,
389389 grecaptcha_secret = 'X' * 30 ,
@@ -393,9 +393,9 @@ def settings(other_server):
393393 )
394394
395395
396- @pytest .yield_fixture (scope = 'session' )
396+ @pytest .fixture (scope = 'session' )
397397def db ():
398- settings_ = Settings (pg_dsn = os .getenv ('DATABASE_URL' , DB_DSN ))
398+ settings_ = Settings (database_url = os .getenv ('DATABASE_URL' , DB_DSN ))
399399 prepare_database (True , settings_ )
400400
401401 engine = sa_create_engine (settings_ .pg_dsn )
@@ -404,9 +404,9 @@ def db():
404404 engine .dispose ()
405405
406406
407- @pytest .yield_fixture
407+ @pytest .fixture
408408def db_conn (loop , settings , db ):
409- engine = loop .run_until_complete (aio_create_engine (settings .pg_dsn , loop = loop ))
409+ engine = loop .run_until_complete (aio_create_engine (settings .database_url , loop = loop ))
410410 conn = loop .run_until_complete (engine .acquire ())
411411 transaction = loop .run_until_complete (conn .begin ())
412412
0 commit comments