Changeset 141 for pyyaml/trunk/tests/test_errors.py

Timestamp:

04/18/06 12:48:00 (7 years ago)

Author:

xi

Message:

Add more unit tests for error messages.

File:

• pyyaml/trunk/tests/test_errors.py (modified) (3 diffs)

pyyaml/trunk/tests/test_errors.py

@@ -1 +1 @@
 
 import test_appliance
+import test_emitter
+
+import StringIO
 
 from yaml import *
@@ -13 +16 @@
         #self._load_string(invalid_filename)
         self.failUnlessRaises(YAMLError, lambda: self._load_string(invalid_filename))
+
+    def _testEmitterErrors(self, test_name, invalid_filename):
+        events = list(load(file(invalid_filename, 'rb').read(),
+            Loader=test_emitter.EventsLoader))
+        self.failUnlessRaises(YAMLError, lambda: self._emit(events))
+
+    def _testDumperErrors(self, test_name, invalid_filename):
+        code = file(invalid_filename, 'rb').read()
+        self.failUnlessRaises(YAMLError, lambda: self._dump(code))
+
+    def _dump(self, code):
+        try:
+            exec code
+        except YAMLError, exc:
+            #print '.'*70
+            #print "%s:" % exc.__class__.__name__, exc
+            raise
+
+    def _emit(self, events):
+        try:
+            emit(events)
+        except EmitterError, exc:
+            #print '.'*70
+            #print "%s:" % exc.__class__.__name__, exc
+            raise
 
     def _load(self, filename):
@@ -41 +69 @@
 TestErrors.add_tests('testLoaderErrors', '.loader-error')
 TestErrors.add_tests('testLoaderStringErrors', '.loader-error')
+TestErrors.add_tests('testEmitterErrors', '.emitter-error')
+TestErrors.add_tests('testDumperErrors', '.dumper-error')