← Back to branch summary

~ubuntu-branches/ubuntu/quantal/python-django/quantal

« back to all changes in this revision

Viewing changes to django/contrib/gis/tests/geoapp/test_sitemaps.py

  • Committer: Bazaar Package Importer
  • Author(s): Scott James Remnant, Eddy Mulyono
  • Date: 2008-09-16 12:18:47 UTC
  • mfrom: (1.1.5 upstream) (4.1.1 lenny)
  • Revision ID: james.westby@ubuntu.com-20080916121847-mg225rg5mnsdqzr0
Tags: 1.0-1ubuntu1
https://launchpad.net/bugs/264191
* Merge from Debian (LP: #264191), remaining changes:
  - Run test suite on build.

[Eddy Mulyono]
* Update patch to workaround network test case failures.

Show diffs side-by-side

added added

removed removed

Lines of Context:
django/contrib/gis/tests/geoapp/test_sitemaps.py
 
1
import unittest, zipfile, cStringIO
 
2
from xml.dom import minidom
 
3
 
 
4
from django.test import Client
 
5
from models import City, Country
 
6
 
 
7
class GeoSitemapTest(unittest.TestCase):
 
8
    client = Client()
 
9
 
 
10
    def assertChildNodes(self, elem, expected):
 
11
        "Taken from regressiontests/syndication/tests.py."
 
12
        actual = set([n.nodeName for n in elem.childNodes])
 
13
        expected = set(expected)
 
14
        self.assertEqual(actual, expected)
 
15
 
 
16
    def test_geositemap_index(self):
 
17
        "Tests geographic sitemap index."
 
18
        # Getting the geo index.
 
19
        doc = minidom.parseString(self.client.get('/geoapp/sitemap.xml').content)
 
20
        index = doc.firstChild
 
21
        self.assertEqual(index.getAttribute(u'xmlns'), u'http://www.sitemaps.org/schemas/sitemap/0.9')
 
22
        self.assertEqual(3, len(index.getElementsByTagName('sitemap')))
 
23
 
 
24
    def test_geositemap_kml(self):
 
25
        "Tests KML/KMZ geographic sitemaps."
 
26
        for kml_type in ('kml', 'kmz'):
 
27
            doc = minidom.parseString(self.client.get('/geoapp/sitemaps/%s.xml' % kml_type).content)
 
28
 
 
29
            # Ensuring the right sitemaps namespaces are present.
 
30
            urlset = doc.firstChild
 
31
            self.assertEqual(urlset.getAttribute(u'xmlns'), u'http://www.sitemaps.org/schemas/sitemap/0.9')
 
32
            self.assertEqual(urlset.getAttribute(u'xmlns:geo'), u'http://www.google.com/geo/schemas/sitemap/1.0')
 
33
        
 
34
            urls = urlset.getElementsByTagName('url')
 
35
            self.assertEqual(2, len(urls)) # Should only be 2 sitemaps.
 
36
            for url in urls:
 
37
                self.assertChildNodes(url, ['loc', 'geo:geo'])
 
38
                # Making sure the 'geo:format' element was properly set.
 
39
                geo_elem = url.getElementsByTagName('geo:geo')[0]
 
40
                geo_format = geo_elem.getElementsByTagName('geo:format')[0]
 
41
                self.assertEqual(kml_type, geo_format.childNodes[0].data)
 
42
 
 
43
                # Getting the relative URL since we don't have a real site.
 
44
                kml_url = url.getElementsByTagName('loc')[0].childNodes[0].data.split('http://example.com')[1]
 
45
                
 
46
                if kml_type == 'kml':
 
47
                    kml_doc = minidom.parseString(self.client.get(kml_url).content)
 
48
                elif kml_type == 'kmz':
 
49
                    # Have to decompress KMZ before parsing.
 
50
                    buf = cStringIO.StringIO(self.client.get(kml_url).content)
 
51
                    zf = zipfile.ZipFile(buf)
 
52
                    self.assertEqual(1, len(zf.filelist))
 
53
                    self.assertEqual('doc.kml', zf.filelist[0].filename)
 
54
                    kml_doc = minidom.parseString(zf.read('doc.kml'))
 
55
                
 
56
                # Ensuring the correct number of placemarks are in the KML doc.
 
57
                if 'city' in kml_url:
 
58
                    model = City
 
59
                elif 'country' in kml_url:
 
60
                    model = Country
 
61
                self.assertEqual(model.objects.count(), len(kml_doc.getElementsByTagName('Placemark')))
 
62
 
 
63
    def test_geositemap_georss(self):
 
64
        "Tests GeoRSS geographic sitemaps."
 
65
        from feeds import feed_dict
 
66
 
 
67
        doc = minidom.parseString(self.client.get('/geoapp/sitemaps/georss.xml').content)
 
68
   
 
69
        # Ensuring the right sitemaps namespaces are present.
 
70
        urlset = doc.firstChild
 
71
        self.assertEqual(urlset.getAttribute(u'xmlns'), u'http://www.sitemaps.org/schemas/sitemap/0.9')
 
72
        self.assertEqual(urlset.getAttribute(u'xmlns:geo'), u'http://www.google.com/geo/schemas/sitemap/1.0')
 
73
 
 
74
        # Making sure the correct number of feed URLs were included.
 
75
        urls = urlset.getElementsByTagName('url')
 
76
        self.assertEqual(len(feed_dict), len(urls))
 
77
 
 
78
        for url in urls:
 
79
            self.assertChildNodes(url, ['loc', 'geo:geo'])
 
80
            # Making sure the 'geo:format' element was properly set to 'georss'.
 
81
            geo_elem = url.getElementsByTagName('geo:geo')[0]
 
82
            geo_format = geo_elem.getElementsByTagName('geo:format')[0]
 
83
            self.assertEqual('georss', geo_format.childNodes[0].data)