simplejson.tool tests and bugfix for Python 3.x (#60)

CHANGES.txt

 Version 3.1.0 released XXXX-XX-XX
 
+* simplejson.tool tests and bugfix for Python 3.x
+  http://bugs.python.org/issue16549
 * Improve error messages for certain kinds of truncated input
   http://bugs.python.org/issue16009
 * Moved JSONDecodeError to json.scanner (still available for import

simplejson/tests/__init__.py

@@ -59,6 +59,7 @@
         'simplejson.tests.test_decimal',
         'simplejson.tests.test_tuple',
         'simplejson.tests.test_namedtuple',
+        'simplejson.tests.test_tool',
     ])
     suite = additional_tests(suite)
     return OptionalExtensionTestSuite([suite])

simplejson/tests/test_tool.py

+import os
+import sys
+import textwrap
+import unittest
+import subprocess
+import tempfile
+
+class TestTool(unittest.TestCase):
+    data = """
+
+        [["blorpie"],[ "whoops" ] , [
+                                 ],\t"d-shtaeou",\r"d-nthiouh",
+        "i-vhbjkhnth", {"nifty":87}, {"morefield" :\tfalse,"field"
+            :"yes"}  ]
+           """
+
+    expect = textwrap.dedent("""\
+    [
+        [
+            "blorpie"
+        ],
+        [
+            "whoops"
+        ],
+        [],
+        "d-shtaeou",
+        "d-nthiouh",
+        "i-vhbjkhnth",
+        {
+            "nifty": 87
+        },
+        {
+            "field": "yes",
+            "morefield": false
+        }
+    ]
+    """)
+
+    def runTool(self, args=None, data=None):
+        argv = [sys.executable, '-m', 'simplejson.tool']
+        if args:
+            argv.extend(args)
+        proc = subprocess.Popen(argv,
+                                stdin=subprocess.PIPE,
+                                stderr=subprocess.PIPE,
+                                stdout=subprocess.PIPE)
+        out, err = proc.communicate(data)
+        self.assertEqual(len(err), 0)
+        self.assertEqual(proc.returncode, 0)
+        return out
+
+    def test_stdin_stdout(self):
+        self.assertEqual(
+            self.runTool(data=self.data.encode()),
+            self.expect.encode())
+
+    def test_infile_stdout(self):
+        with tempfile.NamedTemporaryFile() as infile:
+            infile.write(self.data.encode())
+            infile.flush()
+            self.assertEqual(
+                self.runTool(args=[infile.name]),
+                self.expect.encode())
+
+    def test_infile_outfile(self):
+        with tempfile.NamedTemporaryFile() as infile:
+            infile.write(self.data.encode())
+            infile.flush()
+            # outfile will get overwritten by tool, so the delete
+            # may not work on some platforms. Do it manually.
+            outfile = tempfile.NamedTemporaryFile(delete=0)
+            try:
+                self.assertEqual(
+                    self.runTool(args=[infile.name, outfile.name]),
+                    ''.encode())
+                with open(outfile.name, 'rb') as f:
+                    self.assertEqual(f.read(), self.expect.encode())
+            finally:
+                outfile.close()
+                os.unlink(outfile.name)

simplejson/tool.py

@@ -18,6 +18,6 @@
         infile = sys.stdin
         outfile = sys.stdout
     elif len(sys.argv) == 2:
-        infile = open(sys.argv[1], 'rb')
+        infile = open(sys.argv[1], 'r')
         outfile = sys.stdout
     elif len(sys.argv) == 3:
@@ -22,6 +22,6 @@
         outfile = sys.stdout
     elif len(sys.argv) == 3:
-        infile = open(sys.argv[1], 'rb')
-        outfile = open(sys.argv[2], 'wb')
+        infile = open(sys.argv[1], 'r')
+        outfile = open(sys.argv[2], 'w')
     else:
         raise SystemExit(sys.argv[0] + " [infile [outfile]]")
@@ -26,8 +26,9 @@
     else:
         raise SystemExit(sys.argv[0] + " [infile [outfile]]")
+    with infile:
         try:
             obj = json.load(infile,
                             object_pairs_hook=json.OrderedDict,
                             use_decimal=True)
         except ValueError:
             raise SystemExit(sys.exc_info()[1])
@@ -28,9 +29,10 @@
         try:
             obj = json.load(infile,
                             object_pairs_hook=json.OrderedDict,
                             use_decimal=True)
         except ValueError:
             raise SystemExit(sys.exc_info()[1])
+    with outfile:
         json.dump(obj, outfile, sort_keys=True, indent='    ', use_decimal=True)
         outfile.write('\n')