Coverage for /var/srv/projects/api.amasfac.comuna18.com/tmp/venv/lib/python3.9/site-packages/factory/mogo.py: 76%

1# Copyright: See the LICENSE file. 

2 

3 

4"""factory_boy extensions for use with the mogo library (pymongo wrapper).""" 

5 

6 

7from . import base 

8 

9 

10class MogoFactory(base.Factory): 

11 """Factory for mogo objects.""" 

12 class Meta: 

13 abstract = True 

14 

15 @classmethod 

16 def _build(cls, model_class, *args, **kwargs): 

17 return model_class(*args, **kwargs) 

18 

19 @classmethod 

20 def _create(cls, model_class, *args, **kwargs): 

21 instance = model_class(*args, **kwargs) 

22 instance.save() 

23 return instance