Fixed #13670 -- Comparisons with the spatial adapter won't blow up in some corner cases. Thanks, milosu for the bug report and jpaulett for the patch.

git-svn-id: http://code.djangoproject.com/svn/django/trunk@16757 bcc190cf-cafb-0310-a4f2-bffc1f526a37
This commit is contained in:
Justin Bronn 2011-09-10 00:29:34 +00:00
parent e061b036a5
commit 0516ac3d28
3 changed files with 15 additions and 1 deletions

View File

@ -8,6 +8,8 @@ class WKTAdapter(object):
self.srid = geom.srid
def __eq__(self, other):
if not isinstance(other, WKTAdapter):
return False
return self.wkt == other.wkt and self.srid == other.srid
def __str__(self):

View File

@ -21,6 +21,8 @@ class PostGISAdapter(object):
raise Exception('Error implementing psycopg2 protocol. Is psycopg2 installed?')
def __eq__(self, other):
if not isinstance(other, PostGISAdapter):
return False
return (self.ewkb == other.ewkb) and (self.srid == other.srid)
def __str__(self):

View File

@ -2,7 +2,7 @@ from datetime import datetime
from django.contrib.gis.tests.utils import no_mysql, no_spatialite
from django.contrib.gis.shortcuts import render_to_kmz
from django.test import TestCase
from models import City, PennsylvaniaCity
from models import City, PennsylvaniaCity, State
class GeoRegressionTests(TestCase):
@ -43,3 +43,13 @@ class GeoRegressionTests(TestCase):
mansfield = PennsylvaniaCity.objects.create(name='Mansfield', county='Tioga', point='POINT(-77.071445 41.823881)',
founded=founded)
self.assertEqual(founded, PennsylvaniaCity.objects.dates('founded', 'day')[0])
def test05_empty_count(self):
"Testing that PostGISAdapter.__eq__ does check empty strings, see #13670"
# contrived example, but need a geo lookup paired with an id__in lookup
pueblo = City.objects.get(name='Pueblo')
state = State.objects.filter(poly__contains=pueblo.point)
cities_within_state = City.objects.filter(id__in=state)
# .count() should not throw TypeError in __eq__
self.assertEqual(cities_within_state.count(), 1)