From 48829cd3c6e723eadf5eb60fd8e521a1761fe8b3 Mon Sep 17 00:00:00 2001
From: Damien George <damien.p.george@gmail.com>
Date: Wed, 13 Jun 2018 13:13:36 +1000
Subject: [PATCH] tests/extmod: Add test for ujson.dump writing to a user
 IOBase object.

---
 tests/extmod/ujson_dump_iobase.py | 32 +++++++++++++++++++++++++++++++
 1 file changed, 32 insertions(+)
 create mode 100644 tests/extmod/ujson_dump_iobase.py

diff --git a/tests/extmod/ujson_dump_iobase.py b/tests/extmod/ujson_dump_iobase.py
new file mode 100644
index 0000000000..d30d1b561e
--- /dev/null
+++ b/tests/extmod/ujson_dump_iobase.py
@@ -0,0 +1,32 @@
+# test ujson.dump in combination with uio.IOBase
+
+try:
+    import uio as io
+    import ujson as json
+except ImportError:
+    try:
+        import io, json
+    except ImportError:
+        print('SKIP')
+        raise SystemExit
+
+if not hasattr(io, 'IOBase'):
+    print('SKIP')
+    raise SystemExit
+
+
+# a user stream that only has the write method
+class S(io.IOBase):
+    def __init__(self):
+        self.buf = ''
+    def write(self, buf):
+        if type(buf) == bytearray:
+            # uPy passes a bytearray, CPython passes a str
+            buf = str(buf, 'ascii')
+        self.buf += buf
+
+
+# dump to the user stream
+s = S()
+json.dump([123, {}], s) 
+print(s.buf)