Changeset 1064

Timestamp:

11/05/06 20:54:08 (2 years ago)

Author:

arjanmol

Message:

added load/save unit tests

Files:

• gaphor/branches/new-canvas/gaphor/tests/test_storage.py (modified) (2 diffs)

gaphor/branches/new-canvas/gaphor/tests/test_storage.py

@@ -69 +69 @@
-        UML.create(UML.Comment)
-        UML.create(UML.Class)
-
+
-        fd = open(filename, 'w')
-        storage.save(XMLWriter(fd))
@@ -86 +86 @@
-        
-
+    def test_load_uml(self):
+        """Test loading of a freshly saved model.
+        """
+        filename = os.tmpnam()
+
+        UML.create(UML.Package)
+        diagram = UML.create(UML.Diagram)
+        diagram.create(items.CommentItem, subject=UML.create(UML.Comment))
+        diagram.create(items.ClassItem, subject=UML.create(UML.Class))
+ 
+        fd = open(filename, 'w')
+        storage.save(XMLWriter(fd))
+        fd.close()
+
+        UML.flush()
+        assert not list(UML.select())
+
+        storage.load(filename)
+
+        assert len(UML.lselect()) == 4
+        assert len(UML.lselect(lambda e: e.isKindOf(UML.Package))) == 1
+        assert len(UML.lselect(lambda e: e.isKindOf(UML.Diagram))) == 1
+        assert len(UML.lselect(lambda e: e.isKindOf(UML.Comment))) == 1
+        assert len(UML.lselect(lambda e: e.isKindOf(UML.Class))) == 1
+        
+        # TODO: check load/save of other canvas items.
+
+    def load_x_gaphor(self):
+        storage.load('x.gaphor')
+