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	# 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	#print written_contents
118	#print "********************************************"
119	#print expected_contents
120	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