File tree Expand file tree Collapse file tree 2 files changed +6
-5
lines changed
Expand file tree Collapse file tree 2 files changed +6
-5
lines changed Original file line number Diff line number Diff line change 11import os
2+ from typing import Generator
23import pytest
34from opengeodeweb_microservice .database .connection import init_database , get_session
45from opengeodeweb_microservice .database .data import Data
89
910
1011@pytest .fixture (scope = "session" , autouse = True )
11- def setup_database ():
12+ def setup_database () -> Generator [ None , None , None ] :
1213 init_database (DB_PATH )
1314 yield
1415 _cleanup_database (DB_PATH )
1516
1617
17- def _cleanup_database (db_path : str ):
18+ def _cleanup_database (db_path : str ) -> None :
1819 try :
1920 session = get_session ()
2021 session .close ()
@@ -29,7 +30,7 @@ def _cleanup_database(db_path: str):
2930
3031
3132@pytest .fixture (autouse = True )
32- def clean_database ():
33+ def clean_database () -> Generator [ None , None , None ] :
3334 with get_session () as session :
3435 session = get_session ()
3536 session .query (Data ).delete ()
Original file line number Diff line number Diff line change 11from opengeodeweb_microservice .database .data import Data
22
33
4- def test_data_crud_operations (clean_database ) :
4+ def test_data_crud_operations (clean_database : None ) -> None :
55 data = Data .create (
66 geode_object = "test_object" ,
77 viewer_object = "test_viewer" ,
@@ -22,7 +22,7 @@ def test_data_crud_operations(clean_database):
2222 assert non_existent is None
2323
2424
25- def test_data_with_additional_files (clean_database ) :
25+ def test_data_with_additional_files (clean_database : None ) -> None :
2626 files = ["file1.txt" , "file2.txt" ]
2727 data = Data .create (
2828 geode_object = "test_files" ,
You can’t perform that action at this time.
0 commit comments