test_tool.py

import os
import re
import sys
import textwrap
import unittest
import subprocess
import tempfile

def strip_python_stderr(stderr):
    """Strip debug-build refcount output from stderr."""
    return re.sub(b"\\[\\d+ refs\\]\\r?\\n?$", b"", stderr).strip()

def open_temp_file():
    file = tempfile.NamedTemporaryFile(delete=False)
    filename = file.name
    return file, filename

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):
        if sys.platform == 'emscripten':
            self.skipTest("subprocess not available on Emscripten")
        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)
        return out.decode('utf8').splitlines()

def test_stdin_stdout(self):
        self.assertEqual(
            self.runTool(data=self.data.encode()),
            self.expect.splitlines())

def test_infile_stdout(self):
        infile, infile_name = open_temp_file()
        try:
            infile.write(self.data.encode())
            infile.close()
            self.assertEqual(
                self.runTool(args=[infile_name]),
                self.expect.splitlines())
        finally:
            os.unlink(infile_name)

def test_infile_outfile(self):
        infile, infile_name = open_temp_file()
        try:
            infile.write(self.data.encode())
            infile.close()
            # outfile will get overwritten by tool, so the delete
            # may not work on some platforms. Do it manually.
            outfile, outfile_name = open_temp_file()
            try:
                outfile.close()
                self.assertEqual(
                    self.runTool(args=[infile_name, outfile_name]),
                    [])
                with open(outfile_name, 'rb') as f:
                    self.assertEqual(
                        f.read().decode('utf8').splitlines(),
                        self.expect.splitlines()
                    )
            finally:
                os.unlink(outfile_name)
        finally:
            os.unlink(infile_name)

docroot: /home/induvedam/public_html

curdir: /opt/imunify360/venv/lib/python3.11/site-packages/simplejson/tests

	.	4096
	..	4096
X	__init__.py	2917
	__pycache__	4096
X	_cibw_runner.py	173
X	_helpers.py	405
X	test_bigint_as_string.py	2238
X	test_bitsize_int_as_string.py	4535
X	test_check_circular.py	1579
X	test_decimal.py	2544
X	test_decode.py	7470
X	test_default.py	221
X	test_dump.py	17600
X	test_encode_basestring_ascii.py	2337
X	test_encode_for_html.py	1515
X	test_errors.py	3503
X	test_fail.py	6454
X	test_float.py	1983
X	test_for_json.py	4126
X	test_free_threading.py	3169
X	test_indent.py	2568
X	test_item_sort_key.py	1376
X	test_iterable.py	2580
X	test_namedtuple.py	7391
X	test_pass1.py	1746
X	test_pass2.py	386
X	test_pass3.py	482
X	test_raw_json.py	1062
X	test_recursion.py	1679
X	test_scanstring.py	12098
X	test_separators.py	942
X	test_speedups.py	12590
X	test_str_subclass.py	740
X	test_subclass.py	1124
X	test_subinterpreters.py	4639
X	test_tool.py	2996
X	test_tuple.py	1831
X	test_unicode.py	7056