summaryrefslogtreecommitdiff
path: root/venv/lib/python3.11/site-packages/jsbeautifier/tests/testindentation.py
blob: e4193aab0a809470e6f8e9b41a5b5db69c10a802 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
import re
import unittest
import jsbeautifier


class TestJSBeautifierIndentation(unittest.TestCase):
    def test_tabs(self):
        test_fragment = self.decodesto

        self.options.indent_with_tabs = 1
        test_fragment("{tabs()}", "{\n\ttabs()\n}")

    def test_function_indent(self):
        test_fragment = self.decodesto

        self.options.indent_with_tabs = 1
        self.options.keep_function_indentation = 1
        test_fragment(
            "var foo = function(){ bar() }();", "var foo = function() {\n\tbar()\n}();"
        )

        self.options.tabs = 1
        self.options.keep_function_indentation = 0
        test_fragment(
            "var foo = function(){ baz() }();", "var foo = function() {\n\tbaz()\n}();"
        )

    def decodesto(self, input, expectation=None):
        self.assertEqual(
            jsbeautifier.beautify(input, self.options), expectation or input
        )

    @classmethod
    def setUpClass(cls):
        options = jsbeautifier.default_options()
        options.indent_size = 4
        options.indent_char = " "
        options.preserve_newlines = True
        options.jslint_happy = False
        options.keep_array_indentation = False
        options.brace_style = "collapse"
        options.indent_level = 0

        cls.options = options
        cls.wrapregex = re.compile("^(.+)$", re.MULTILINE)


if __name__ == "__main__":
    unittest.main()