71 lines
1.5 KiB
Python
71 lines
1.5 KiB
Python
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
|
|
}
|