Adding upstream version 0.91.0.upstream/0.91.0upstream

Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>

1 files changed, 49 insertions, 0 deletions

diff --git a/_test/lib/test_reader.py b/_test/lib/test_reader.py
new file mode 100644
index 0000000..56ad671
--- /dev/null
+++ b/_test/lib/test_reader.py
@@ -0,0 +1,49 @@
+import codecs  # NOQA
+import io
+
+# Skipped because we have no idea where the "error_filename"
+# fixture is supposed to come from
+import pytest
+
+import ruyaml.reader
+
+pytestmark = pytest.mark.skip
+
+
+def _run_reader(data, verbose):
+    try:
+        stream = ruyaml.py.reader.Reader(data)
+        while stream.peek() != '\0':
+            stream.forward()
+    except ruyaml.py.reader.ReaderError as exc:
+        if verbose:
+            print(exc)
+    else:
+        raise AssertionError('expected an exception')
+
+
+def test_stream_error(error_filename, verbose=False):
+    with open(error_filename, 'rb') as fp0:
+        _run_reader(fp0, verbose)
+    with open(error_filename, 'rb') as fp0:
+        _run_reader(fp0.read(), verbose)
+    for encoding in ['utf-8', 'utf-16-le', 'utf-16-be']:
+        try:
+            with open(error_filename, 'rb') as fp0:
+                data = fp0.read().decode(encoding)
+            break
+        except UnicodeDecodeError:
+            pass
+    else:
+        return
+    _run_reader(data, verbose)
+    with io.open(error_filename, encoding=encoding) as fp:
+        _run_reader(fp, verbose)
+
+
+test_stream_error.unittest = ['.stream-error']
+
+if __name__ == '__main__':
+    import test_appliance
+
+    test_appliance.run(globals())