thuban/test/test_save.py

# Copyright (c) 2002 by Intevation GmbH
# Authors:
# Bernhard Herzog <[email protected]>
#
# This program is free software under the GPL (>=v2)
# Read the file COPYING coming with Thuban for details.

"""
Test saving a thuban session as XML
"""

__version__ = "$Revision$"
# $Source$
# $Id$

import os
import unittest
from StringIO import StringIO

import xml.sax
import xml.sax.handler
from xml.sax import make_parser, ErrorHandler

import support
support.initthuban()

from Thuban.Model.save import save_session
from Thuban.Model.session import Session
from Thuban.Model.map import Map
from Thuban.Model.layer import Layer
from Thuban.Model.proj import Projection


class SaxEventLister(xml.sax.handler.ContentHandler):

    def __init__(self):
        self.eventlist = []

    def startElementNS(self, name, qname, attrs):
        items = attrs.items()
        items.sort()
        self.eventlist.append(("start", name, qname, items))

    def endElementNS(self, name, qname):
        self.eventlist.append(("end", name, qname))


def sax_eventlist(data):
    """Return a list of SAX event generated for the XML data.
    """
    handler = SaxEventLister()
    parser = make_parser()
    parser.setContentHandler(handler)
    parser.setErrorHandler(ErrorHandler())
    parser.setFeature(xml.sax.handler.feature_namespaces, 1)

    inpsrc = xml.sax.InputSource()
    inpsrc.setByteStream(StringIO(data))
    parser.parse(inpsrc)

    return handler.eventlist

class SaveSessionTest(unittest.TestCase, support.FileTestMixin):

    def compare_xml(self, xml1, xml2):
        self.assertEquals(sax_eventlist(xml1), sax_eventlist(xml2))

    def testEmptySession(self):
        """Save an empty session"""
        session = Session("empty session")
        filename = self.temp_file_name("save_emptysession.thuban")
        save_session(session, filename)
        session.Destroy()

        file = open(filename)
        written_contents = file.read()
        file.close()
        self.compare_xml(written_contents,
                         '<?xml version="1.0" encoding="UTF-8"?>\n'
                         '<!DOCTYPE session SYSTEM "thuban.dtd">\n'
                         '<session title="empty session">\n</session>\n')

    def testSingleLayer(self):
        """Save a session with a single map with a single layer"""
        # deliberately put an apersand in the title :)
        session = Session("single map&layer")
        proj = Projection(["zone=26", "proj=utm", "ellps=clrk66"])
        map = Map("Test Map", projection = proj)
        session.AddMap(map)
        # use shapefile from the example data
        shpfile = os.path.join(os.path.dirname(__file__),
                               os.pardir, "Data", "iceland", "political.shp")
        layer = Layer("My Layer", shpfile)
        map.AddLayer(layer)

        filename = self.temp_file_name("save_singlemap.thuban")
        save_session(session, filename)
        session.Destroy()

        file = open(filename)
        written_contents = file.read()
        file.close()
        expected_contents = '''<?xml version="1.0" encoding="UTF-8"?>
        <!DOCTYPE session SYSTEM "thuban.dtd">
        <session title="single map&amp;layer">
            <map title="Test Map">
                <projection>
                    <parameter value="zone=26"/>
                    <parameter value="proj=utm"/>
                    <parameter value="ellps=clrk66"/>
                </projection>
                <layer title="My Layer" filename="%s"
                fill="None" stroke="#000000" stroke_width="1"/>
            </map>
        </session>''' % os.path.join("..", "..", "Data", "iceland",
                                     "political.shp")
        #print written_contents
        #print "********************************************"
        #print expected_contents
        self.compare_xml(written_contents, expected_contents)


if __name__ == "__main__":
    # Fake the __file__ global because it's needed by a test
    import sys
    __file__ = sys.argv[0]
    unittest.main()
1	bh	292	# Copyright (c) 2002 by Intevation GmbH
2			# Authors:
3			# Bernhard Herzog <[email protected]>
4			#
5			# This program is free software under the GPL (>=v2)
6			# Read the file COPYING coming with Thuban for details.
7
8			"""
9			Test saving a thuban session as XML
10			"""
11
12			__version__ = "$Revision$"
13			# $Source$
14			# $Id$
15
16			import os
17			import unittest
18			from StringIO import StringIO
19
20			import xml.sax
21			import xml.sax.handler
22			from xml.sax import make_parser, ErrorHandler
23
24			import support
25			support.initthuban()
26
27			from Thuban.Model.save import save_session
28			from Thuban.Model.session import Session
29			from Thuban.Model.map import Map
30			from Thuban.Model.layer import Layer
31			from Thuban.Model.proj import Projection
32
33
34			class SaxEventLister(xml.sax.handler.ContentHandler):
35
36			def __init__(self):
37			self.eventlist = []
38
39			def startElementNS(self, name, qname, attrs):
40			items = attrs.items()
41			items.sort()
42			self.eventlist.append(("start", name, qname, items))
43
44			def endElementNS(self, name, qname):
45			self.eventlist.append(("end", name, qname))
46
47
48			def sax_eventlist(data):
49			"""Return a list of SAX event generated for the XML data.
50			"""
51			handler = SaxEventLister()
52			parser = make_parser()
53			parser.setContentHandler(handler)
54			parser.setErrorHandler(ErrorHandler())
55			parser.setFeature(xml.sax.handler.feature_namespaces, 1)
56
57			inpsrc = xml.sax.InputSource()
58			inpsrc.setByteStream(StringIO(data))
59			parser.parse(inpsrc)
60
61			return handler.eventlist
62
63			class SaveSessionTest(unittest.TestCase, support.FileTestMixin):
64
65			def compare_xml(self, xml1, xml2):
66			self.assertEquals(sax_eventlist(xml1), sax_eventlist(xml2))
67
68			def testEmptySession(self):
69			"""Save an empty session"""
70			session = Session("empty session")
71			filename = self.temp_file_name("save_emptysession.thuban")
72			save_session(session, filename)
73			session.Destroy()
74
75			file = open(filename)
76			written_contents = file.read()
77			file.close()
78			self.compare_xml(written_contents,
79			'<?xml version="1.0" encoding="UTF-8"?>\n'
80			'<!DOCTYPE session SYSTEM "thuban.dtd">\n'
81			'<session title="empty session">\n</session>\n')
82
83			def testSingleLayer(self):
84			"""Save a session with a single map with a single layer"""
85			# deliberately put an apersand in the title :)
86			session = Session("single map&layer")
87			proj = Projection(["zone=26", "proj=utm", "ellps=clrk66"])
88			map = Map("Test Map", projection = proj)
89			session.AddMap(map)
90			# use shapefile from the example data
91			shpfile = os.path.join(os.path.dirname(__file__),
92			os.pardir, "Data", "iceland", "political.shp")
93			layer = Layer("My Layer", shpfile)
94			map.AddLayer(layer)
95
96			filename = self.temp_file_name("save_singlemap.thuban")
97			save_session(session, filename)
98			session.Destroy()
99
100			file = open(filename)
101			written_contents = file.read()
102			file.close()
103			expected_contents = '''<?xml version="1.0" encoding="UTF-8"?>
104			<!DOCTYPE session SYSTEM "thuban.dtd">
105			<session title="single map&layer">
106			<map title="Test Map">
107			<projection>
108			<parameter value="zone=26"/>
109			<parameter value="proj=utm"/>
110			<parameter value="ellps=clrk66"/>
111			</projection>
112			<layer title="My Layer" filename="%s"
113			fill="None" stroke="#000000" stroke_width="1"/>
114			</map>
115			</session>''' % os.path.join("..", "..", "Data", "iceland",
116			"political.shp")
117	jonathan	494	#print written_contents
118			#print "********************************************"
119			#print expected_contents
120	bh	292	self.compare_xml(written_contents, expected_contents)
121
122
123
124			if __name__ == "__main__":
125			# Fake the __file__ global because it's needed by a test
126			import sys
127			__file__ = sys.argv[0]
128			unittest.main()
Name	Value
svn:eol-style	native
svn:keywords	Author Date Id Revision