1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61
|
import uuid
from django.db import models
from django.conf import settings
from django.core.exceptions import ImproperlyConfigured
try:
from django.contrib.gis.geos import *
from django.contrib.gis.db import models as geomodels
except ImproperlyConfigured:
pass # environment without geo libs
from django.test import TestCase
from django_dynamic_fixture.fixture_algorithms.default_fixture import BaseDataFixture
class BaseDataFixtureTestCase(TestCase):
def setUp(self):
self.fixture = BaseDataFixture()
def test_uuid(self):
assert isinstance(self.fixture.generate_data(models.UUIDField()), uuid.UUID)
if (hasattr(settings, 'DDF_TEST_GEODJANGO') and settings.DDF_TEST_GEODJANGO):
from django_dynamic_fixture.fixture_algorithms.default_fixture import GeoDjangoFixtureMixin
# Mixing for tests
class GeoDjangoFixtureMixin(BaseDataFixture, GeoDjangoFixtureMixin):
pass
class GeoDjangoDataFixtureTestCase(TestCase):
def setUp(self):
self.fixture = GeoDjangoFixtureMixin()
def test_geometryfield_config(self):
assert isinstance(self.fixture.generate_data(geomodels.GeometryField()), GEOSGeometry)
def test_pointfield_config(self):
assert isinstance(self.fixture.generate_data(geomodels.PointField()), Point)
def test_linestringfield_config(self):
assert isinstance(self.fixture.generate_data(geomodels.LineStringField()), LineString)
def test_polygonfield_config(self):
assert isinstance(self.fixture.generate_data(geomodels.PolygonField()), Polygon)
def test_multipointfield_config(self):
assert isinstance(self.fixture.generate_data(geomodels.MultiPointField()), MultiPoint)
def test_multilinesstringfield_config(self):
assert isinstance(self.fixture.generate_data(geomodels.MultiLineStringField()), MultiLineString)
def test_multipolygonfield_config(self):
assert isinstance(self.fixture.generate_data(geomodels.MultiPolygonField()), MultiPolygon)
def test_geometrycollectionfield_config(self):
assert isinstance(self.fixture.generate_data(geomodels.GeometryCollectionField()), GeometryCollection)
|