34 |
|
|
35 |
"""Very simple test cases for Session""" |
"""Very simple test cases for Session""" |
36 |
|
|
37 |
|
def setUp(self): |
38 |
|
"""Initialize self.session to None""" |
39 |
|
self.session = None |
40 |
|
|
41 |
|
def tearDown(self): |
42 |
|
"""Call self.session.Destroy() and reset self.session to None""" |
43 |
|
self.session.Destroy() |
44 |
|
self.session = None |
45 |
|
|
46 |
def test_initial_state(self): |
def test_initial_state(self): |
47 |
"""Test Session's initial state""" |
"""Test Session's initial state""" |
48 |
session = Session("Test Session") |
session = self.session = Session("Test Session") |
49 |
self.assertEquals(session.Title(), "Test Session") |
self.assertEquals(session.Title(), "Test Session") |
50 |
self.assertEquals(session.Maps(), []) |
self.assertEquals(session.Maps(), []) |
51 |
self.assertEquals(session.Tables(), []) |
self.assertEquals(session.Tables(), []) |
53 |
self.assertEquals(session.filename, None) |
self.assertEquals(session.filename, None) |
54 |
self.failIf(session.HasMaps()) |
self.failIf(session.HasMaps()) |
55 |
self.failIf(session.WasModified()) |
self.failIf(session.WasModified()) |
|
session.Destroy() |
|
56 |
|
|
57 |
def test_add_table(self): |
def test_add_table(self): |
58 |
"""Test Session.AddTable()""" |
"""Test Session.AddTable()""" |
59 |
session = Session("Test Session") |
session = self.session = Session("Test Session") |
60 |
memtable = MemoryTable([("type", FIELDTYPE_STRING), |
memtable = MemoryTable([("type", FIELDTYPE_STRING), |
61 |
("value", FIELDTYPE_DOUBLE), |
("value", FIELDTYPE_DOUBLE), |
62 |
("code", FIELDTYPE_INT)], |
("code", FIELDTYPE_INT)], |
69 |
table = session.AddTable(memtable) |
table = session.AddTable(memtable) |
70 |
self.assertEquals(session.Tables(), [table]) |
self.assertEquals(session.Tables(), [table]) |
71 |
|
|
72 |
|
def test_remove_table(self): |
73 |
|
"""Test Session.RemoveTable()""" |
74 |
|
session = self.session = Session("Test Session") |
75 |
|
memtable = MemoryTable([("type", FIELDTYPE_STRING), |
76 |
|
("value", FIELDTYPE_DOUBLE), |
77 |
|
("code", FIELDTYPE_INT)], |
78 |
|
[("OTHER/UNKNOWN", -1.5, 11), |
79 |
|
("RUINS", 0.0, 1), |
80 |
|
("FARM", 3.141, 2), |
81 |
|
("BUILDING", 2.5, 3), |
82 |
|
("HUT", 1e6, 4), |
83 |
|
("LIGHTHOUSE", -0.01, 5)]) |
84 |
|
table = session.AddTable(memtable) |
85 |
|
self.assertEquals(session.Tables(), [table]) |
86 |
|
session.RemoveTable(table) |
87 |
|
self.assertEquals(session.Tables(), []) |
88 |
|
self.assertRaises(ValueError, session.RemoveTable, table) |
89 |
|
|
90 |
|
def test_open_shapefile(self): |
91 |
|
"""Test Session.OpenShapefile()""" |
92 |
|
session = self.session = Session("Test Session") |
93 |
|
filename = os.path.join("..", "Data", "iceland", |
94 |
|
"roads-line.shp") |
95 |
|
store = session.OpenShapefile(filename) |
96 |
|
self.assertEquals(store.FileName(), os.path.abspath(filename)) |
97 |
|
# The filetype of a shapefile is "shapefile" |
98 |
|
self.assertEquals(store.FileType(), "shapefile") |
99 |
|
# The shapestore itself depends on nothing else |
100 |
|
self.assertEquals(store.Dependencies(), ()) |
101 |
|
# The shapestore's table depends on the shapestore |
102 |
|
self.assertEquals(store.Table().Dependencies(), (store,)) |
103 |
|
|
104 |
|
self.assertEquals(session.Tables(), [store.Table()]) |
105 |
|
|
106 |
|
|
107 |
class TestSessionBase(unittest.TestCase, support.SubscriberMixin): |
class TestSessionBase(unittest.TestCase, support.SubscriberMixin): |
108 |
|
|