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
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
|
# Copyright 2020 The Chromium Authors
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.
import os
import shutil
import tempfile
import unittest
import preprocess_if_expr
_HERE_DIR = os.path.dirname(__file__)
class PreprocessIfExprTest(unittest.TestCase):
def setUp(self):
self._out_folder = None
def tearDown(self):
if self._out_folder:
shutil.rmtree(self._out_folder)
def _read_out_file(self, file_name):
assert self._out_folder
with open(os.path.join(self._out_folder, file_name)) as f:
return f.read()
def _run_test(self, additional_options, file_name, expected_file_name):
assert not self._out_folder
self._out_folder = tempfile.mkdtemp(dir=_HERE_DIR)
preprocess_if_expr.main([
'--in-folder',
os.path.join(_HERE_DIR, 'preprocess_tests'),
'--out-folder',
self._out_folder,
'--in-files',
file_name,
] + additional_options)
actual = self._read_out_file(file_name)
with open(os.path.join(_HERE_DIR, 'preprocess_tests', expected_file_name)) as f:
expected = f.read()
self.assertMultiLineEqual(expected, actual)
def testPreprocess(self):
self._run_test(
['-D', 'foo', '-D', 'bar', '-D', 'apple=false', '-D', 'orange=false'],
'test_with_ifexpr.js', 'test_with_ifexpr_expected.js')
def testPreprocessWithComments(self):
self._run_test([
'-D', 'foo', '-D', 'bar', '-D', 'apple=false', '-D', 'orange=false',
'--enable_removal_comments'
], 'test_with_ifexpr.js', 'test_with_ifexpr_expected_comments.js')
def testPreprocessTypescriptWithComments(self):
self._run_test([
'-D', 'foo', '-D', 'bar', '-D', 'orange=false',
'--enable_removal_comments'
], 'test_with_ifexpr.ts', 'test_with_ifexpr_expected.ts')
def testPreprocessHtmlWithComments(self):
self._run_test(
['-D', 'foo', '-D', 'orange=false', '--enable_removal_comments'],
'test_with_ifexpr.html', 'test_with_ifexpr_expected.html')
def testPreprocessJavaScriptHtmlTemplateWithComments(self):
self._run_test(
['-D', 'foo', '-D', 'bar=false', '--enable_removal_comments'],
'test_with_ifexpr.html.js', 'test_with_ifexpr_expected.html.js')
def testPreprocessTypeScriptHtmlTemplateWithComments(self):
self._run_test(
['-D', 'foo', '-D', 'bar=false', '--enable_removal_comments'],
'test_with_ifexpr.html.ts', 'test_with_ifexpr_expected.html.ts')
if __name__ == '__main__':
unittest.main()
|
