# Copyright 2018-2019 the Deno authors. All rights reserved. MIT license.
import sys
import unittest

from util import (
    pattern_match,
    parse_exit_code,
    shell_quote_win,
    parse_wrk_output,
    root_path,
)
import os


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():
    suite = unittest.TestLoader().loadTestsFromTestCase(TestUtil)
    result = unittest.TextTestRunner(verbosity=2).run(suite)
    if not result.wasSuccessful():
        sys.exit(1)


if __name__ == '__main__':
    util_test()