openmedialibrary_platform_w.../Lib/site-packages/simplejson/tests/test_tool.py

115 lines
3.2 KiB
Python
Raw Normal View History

2016-01-31 14:44:46 +00:00
from __future__ import with_statement
import os
import sys
import textwrap
import unittest
import subprocess
import tempfile
try:
# Python 3.x
from test.support import strip_python_stderr
except ImportError:
# Python 2.6+
try:
from test.test_support import strip_python_stderr
except ImportError:
# Python 2.5
import re
def strip_python_stderr(stderr):
return re.sub(
r"\[\d+ refs\]\r?\n?$".encode(),
"".encode(),
stderr).strip()
2019-01-20 10:35:31 +00:00
def open_temp_file():
if sys.version_info >= (2, 6):
file = tempfile.NamedTemporaryFile(delete=False)
filename = file.name
else:
fd, filename = tempfile.mkstemp()
file = os.fdopen(fd, 'w+b')
return file, filename
2016-01-31 14:44:46 +00:00
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(strip_python_stderr(err), ''.encode())
self.assertEqual(proc.returncode, 0)
2019-01-20 10:35:31 +00:00
return out.decode('utf8').splitlines()
2016-01-31 14:44:46 +00:00
def test_stdin_stdout(self):
self.assertEqual(
self.runTool(data=self.data.encode()),
2019-01-20 10:35:31 +00:00
self.expect.splitlines())
2016-01-31 14:44:46 +00:00
def test_infile_stdout(self):
2019-01-20 10:35:31 +00:00
infile, infile_name = open_temp_file()
try:
2016-01-31 14:44:46 +00:00
infile.write(self.data.encode())
2019-01-20 10:35:31 +00:00
infile.close()
2016-01-31 14:44:46 +00:00
self.assertEqual(
2019-01-20 10:35:31 +00:00
self.runTool(args=[infile_name]),
self.expect.splitlines())
finally:
os.unlink(infile_name)
2016-01-31 14:44:46 +00:00
def test_infile_outfile(self):
2019-01-20 10:35:31 +00:00
infile, infile_name = open_temp_file()
try:
2016-01-31 14:44:46 +00:00
infile.write(self.data.encode())
2019-01-20 10:35:31 +00:00
infile.close()
2016-01-31 14:44:46 +00:00
# outfile will get overwritten by tool, so the delete
# may not work on some platforms. Do it manually.
2019-01-20 10:35:31 +00:00
outfile, outfile_name = open_temp_file()
2016-01-31 14:44:46 +00:00
try:
2019-01-20 10:35:31 +00:00
outfile.close()
2016-01-31 14:44:46 +00:00
self.assertEqual(
2019-01-20 10:35:31 +00:00
self.runTool(args=[infile_name, outfile_name]),
[])
with open(outfile_name, 'rb') as f:
self.assertEqual(
f.read().decode('utf8').splitlines(),
self.expect.splitlines()
)
2016-01-31 14:44:46 +00:00
finally:
2019-01-20 10:35:31 +00:00
os.unlink(outfile_name)
finally:
os.unlink(infile_name)