Coverage for api/handlers/grabbers/organizations.py: 100%

1from mongodb import orgs_col, subsidiaries_col 

2from sotrans_models.models.organizations import ( 

3 OrganizationDBModel, 

4 SubsidiaryDBModel, 

5) 

6from utils.data_grabber import MongoDataGrabber 

7 

8orgs_grabber = MongoDataGrabber( 

9 model_out=OrganizationDBModel, collection=orgs_col 

10) 

11subsidiary_grabber = MongoDataGrabber( 

12 model_out=SubsidiaryDBModel, collection=subsidiaries_col 

13)