chore: Port Python tests to unittest (#2408)

1 files changed, 68 insertions, 66 deletions

diff --git a/tools/util_test.py b/tools/util_test.py
index b5d5dcc98..7d189797d 100644..100755
--- a/tools/util_test.py
+++ b/tools/util_test.py
@@ -1,76 +1,78 @@
 # Copyright 2018-2019 the Deno authors. All rights reserved. MIT license.
-from util import pattern_match, parse_exit_code, shell_quote_win
-import util
-import os
 import sys
+import unittest
+
+from util import (
+    pattern_match,
+    parse_exit_code,
+    shell_quote_win,
+    parse_wrk_output,
+    root_path,
+)
+import os
 
 
-def pattern_match_test():
-    print "Testing util.pattern_match()..."
-    # yapf: disable
-    fixtures = [("foobarbaz", "foobarbaz", True),
-                ("[WILDCARD]", "foobarbaz", True),
-                ("foobar", "foobarbaz", False),
-                ("foo[WILDCARD]baz", "foobarbaz", True),
-                ("foo[WILDCARD]baz", "foobazbar", False),
-                ("foo[WILDCARD]baz[WILDCARD]qux", "foobarbazqatqux", True),
-                ("foo[WILDCARD]", "foobar", True),
-                ("foo[WILDCARD]baz[WILDCARD]", "foobarbazqat", True)]
-    # yapf: enable
-
-    # Iterate through the fixture lists, testing each one
-    for (pattern, string, expected) in fixtures:
-        actual = pattern_match(pattern, string)
-        assert expected == actual, \
-            "expected %s for\nExpected:\n%s\nTo equal actual:\n%s" % (
-            expected, pattern, string)
-
-    assert pattern_match("foo[BAR]baz", "foobarbaz",
-                         "[BAR]") == True, "expected wildcard to be set"
-    assert pattern_match("foo[BAR]baz", "foobazbar",
-                         "[BAR]") == False, "expected wildcard to be set"
-
-
-def parse_exit_code_test():
-    print "Testing util.parse_exit_code()..."
-    assert 54 == parse_exit_code('hello_error54_world')
-    assert 1 == parse_exit_code('hello_error_world')
-    assert 0 == parse_exit_code('hello_world')
-
-
-def shell_quote_win_test():
-    print "Testing util.shell_quote_win()..."
-    assert 'simple' == shell_quote_win('simple')
-    assert 'roof/\\isoprojection' == shell_quote_win('roof/\\isoprojection')
-    assert '"with space"' == shell_quote_win('with space')
-    assert '"embedded""quote"' == shell_quote_win('embedded"quote')
-    assert '"a""b""""c\\d\\\\""e\\\\\\\\"' == shell_quote_win(
-        'a"b""c\\d\\"e\\\\')
-
-
-def parse_wrk_output_test():
-    print "Testing util.parse_wrk_output_test()..."
-    f = open(os.path.join(util.root_path, "tools/testdata/wrk1.txt"))
-    stats = util.parse_wrk_output(f.read())
-    assert stats['req_per_sec'] == 1837
-    assert stats['max_latency'] == 34.96
-
-    f2 = open(os.path.join(util.root_path, "tools/testdata/wrk2.txt"))
-    stats2 = util.parse_wrk_output(f2.read())
-    assert stats2['req_per_sec'] == 53435
-    assert stats2['max_latency'] == 0.00125
-
-    f3 = open(os.path.join(util.root_path, "tools/testdata/wrk3.txt"))
-    stats3 = util.parse_wrk_output(f3.read())
-    assert stats3['req_per_sec'] == 96037
-    assert stats3['max_latency'] == 1630.0
+class TestUtil(unittest.TestCase):
+    def test_pattern_match(self):
+        # yapf: disable
+        fixtures = [("foobarbaz", "foobarbaz", True),
+                    ("[WILDCARD]", "foobarbaz", True),
+                    ("foobar", "foobarbaz", False),
+                    ("foo[WILDCARD]baz", "foobarbaz", True),
+                    ("foo[WILDCARD]baz", "foobazbar", False),
+                    ("foo[WILDCARD]baz[WILDCARD]qux", "foobarbazqatqux", True),
+                    ("foo[WILDCARD]", "foobar", True),
+                    ("foo[WILDCARD]baz[WILDCARD]", "foobarbazqat", True)]
+        # yapf: enable
+
+        # Iterate through the fixture lists, testing each one
+        for (pattern, string, expected) in fixtures:
+            actual = pattern_match(pattern, string)
+            assert expected == actual, \
+                "expected %s for\nExpected:\n%s\nTo equal actual:\n%s" % (
+                expected, pattern, string)
+
+        assert pattern_match("foo[BAR]baz", "foobarbaz",
+                             "[BAR]") == True, "expected wildcard to be set"
+        assert pattern_match("foo[BAR]baz", "foobazbar",
+                             "[BAR]") == False, "expected wildcard to be set"
+
+    def test_parse_exit_code(self):
+        assert 54 == parse_exit_code('hello_error54_world')
+        assert 1 == parse_exit_code('hello_error_world')
+        assert 0 == parse_exit_code('hello_world')
+
+    def test_shell_quote_win(self):
+        assert 'simple' == shell_quote_win('simple')
+        assert 'roof/\\isoprojection' == shell_quote_win(
+            'roof/\\isoprojection')
+        assert '"with space"' == shell_quote_win('with space')
+        assert '"embedded""quote"' == shell_quote_win('embedded"quote')
+        assert '"a""b""""c\\d\\\\""e\\\\\\\\"' == shell_quote_win(
+            'a"b""c\\d\\"e\\\\')
+
+    def test_parse_wrk_output(self):
+        f = open(os.path.join(root_path, "tools/testdata/wrk1.txt"))
+        stats = parse_wrk_output(f.read())
+        assert stats['req_per_sec'] == 1837
+        assert stats['max_latency'] == 34.96
+
+        f2 = open(os.path.join(root_path, "tools/testdata/wrk2.txt"))
+        stats2 = parse_wrk_output(f2.read())
+        assert stats2['req_per_sec'] == 53435
+        assert stats2['max_latency'] == 0.00125
+
+        f3 = open(os.path.join(root_path, "tools/testdata/wrk3.txt"))
+        stats3 = parse_wrk_output(f3.read())
+        assert stats3['req_per_sec'] == 96037
+        assert stats3['max_latency'] == 1630.0
 
 
 def util_test():
-    pattern_match_test()
-    parse_exit_code_test()
-    shell_quote_win_test()
-    parse_wrk_output_test()
+    suite = unittest.TestLoader().loadTestsFromTestCase(TestUtil)
+    result = unittest.TextTestRunner(verbosity=2).run(suite)
+    if not result.wasSuccessful():
+        sys.exit(1)
 
 
 if __name__ == '__main__':