More test improvements

TwoWheelDev · Nov 3, 2014 · 094217a · 094217a
1 parent 9dfbbe2
commit 094217a
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/tests/test_topology.py b/tests/test_topology.py
@@ -156,20 +156,35 @@ def test_nodes(self):
 
     def test_links(self):
         self.assertListEqual(self.app.links, [])
+        self.app.links = [{'id': 1}]
+        self.assertListEqual(self.app.links, [{'id': 1}])
 
     def test_notes(self):
         self.assertListEqual(self.app.notes, [])
+        self.app.notes = [{'id': 1}]
+        self.assertListEqual(self.app.notes, [{'id': 1}])
 
     def test_shapes(self):
         self.assertDictEqual(self.app.shapes,
                              {'ellipse': None, 'rectangle': None})
+        self.app.shapes = {'ellipse': {'id': 1},
+                           'rectangle': {'id': 2}}
+        self.assertDictEqual(self.app.shapes, {'ellipse': {'id': 1},
+                                               'rectangle': {'id': 2}})
 
     def test_images(self):
         self.assertListEqual(self.app.images, [])
+        self.app.images = [{'id': 1}]
+        self.assertListEqual(self.app.images, [{'id': 1}])
 
     def test_servers(self):
         exp_res = [{'host': '127.0.0.1', 'id': 1, 'local': True, 'port': 8000}]
         self.assertListEqual(self.app.servers, exp_res)
+        self.app.servers = [{'host': '127.0.0.1', 'id': 2, 'local': True,
+                            'port': 8001}]
+        exp_res = [{'host': '127.0.0.1', 'id': 2, 'local': True,
+                    'port': 8001}]
+        self.assertListEqual(self.app.servers, exp_res)
 
     def test_name(self):
         self.assertIsNone(self.app.name)