Skip to content

Commit

Permalink
Browse files Browse the repository at this point in the history
Add a test for the geo-enabled inspectdb command
  • Loading branch information
claudep committed Sep 12, 2013
1 parent 37faf12 commit c82f6c2
Show file tree
Hide file tree
Showing 2 changed files with 19 additions and 0 deletions.
1 change: 1 addition & 0 deletions django/contrib/gis/tests/inspectapp/models.py
Expand Up @@ -9,5 +9,6 @@ class AllOGRFields(models.Model):
f_datetime = models.DateTimeField() f_datetime = models.DateTimeField()
f_time = models.TimeField() f_time = models.TimeField()
geom = models.PolygonField() geom = models.PolygonField()
point = models.PointField()


objects = models.GeoManager() objects = models.GeoManager()
18 changes: 18 additions & 0 deletions django/contrib/gis/tests/inspectapp/tests.py
Expand Up @@ -3,11 +3,13 @@
import os import os
from unittest import skipUnless from unittest import skipUnless


from django.core.management import call_command
from django.db import connections from django.db import connections
from django.test import TestCase from django.test import TestCase
from django.contrib.gis.gdal import HAS_GDAL from django.contrib.gis.gdal import HAS_GDAL
from django.contrib.gis.geometry.test_data import TEST_DATA from django.contrib.gis.geometry.test_data import TEST_DATA
from django.contrib.gis.tests.utils import HAS_SPATIAL_DB from django.contrib.gis.tests.utils import HAS_SPATIAL_DB
from django.utils.six import StringIO


if HAS_GDAL: if HAS_GDAL:
from django.contrib.gis.gdal import Driver from django.contrib.gis.gdal import Driver
Expand All @@ -16,6 +18,22 @@
from .models import AllOGRFields from .models import AllOGRFields




@skipUnless(HAS_GDAL and HAS_SPATIAL_DB, "GDAL and spatial db are required.")
class InspectDbTests(TestCase):
def test_geom_columns(self):
"""
Test the geo-enabled inspectdb command.
"""
out = StringIO()
call_command('inspectdb',
table_name_filter=lambda tn:tn.startswith('inspectapp_'),
stdout=out)
output = out.getvalue()
self.assertIn('geom = models.PolygonField()', output)
self.assertIn('point = models.PointField()', output)
self.assertIn('objects = models.GeoManager()', output)


@skipUnless(HAS_GDAL and HAS_SPATIAL_DB, "GDAL and spatial db are required.") @skipUnless(HAS_GDAL and HAS_SPATIAL_DB, "GDAL and spatial db are required.")
class OGRInspectTest(TestCase): class OGRInspectTest(TestCase):
maxDiff = 1024 maxDiff = 1024
Expand Down

0 comments on commit c82f6c2

Please sign in to comment.