import datetime import pytest from itim.models.services import Service from itim.serializers.service import ( ServiceBaseSerializer, ServiceModelSerializer, ServiceViewSerializer, ) @pytest.fixture( scope = 'class') def model_service(): yield Service @pytest.fixture( scope = 'class') def kwargs_service(django_db_blocker, kwargs_centurionmodel, kwargs_device, model_device, kwargs_port, model_port, ): random_str = str(datetime.datetime.now(tz=datetime.timezone.utc)) random_str = str(random_str).replace( ' ', '').replace(':', '').replace('+', '').replace('.', '') with django_db_blocker.unblock(): kwargs = kwargs_device.copy() kwargs.update({ 'name': 'svc' + random_str }) device = model_device.objects.create( **kwargs ) port = model_port.objects.create( **kwargs_port ) kwargs = { **kwargs_centurionmodel.copy(), 'name': 'service_' + random_str, 'device': device, 'config_key_variable': 'svc', 'port': [ port ], 'config': { 'config_key_1': 'config_value_1' } } yield kwargs.copy() with django_db_blocker.unblock(): try: device.delete() except: pass port.delete() @pytest.fixture( scope = 'class') def serializer_service(): yield { 'base': ServiceBaseSerializer, 'model': ServiceModelSerializer, 'view': ServiceViewSerializer }