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
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
|
# Copyright 2024 The Chromium Authors
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.
import unittest
import tempfile
from pathlib import Path
import os
import sys
import glob
PARENT_ROOT = os.path.abspath(os.path.join(os.path.dirname(__file__),
os.pardir))
sys.path.insert(0, PARENT_ROOT)
import gen_android_bp
class GenerateAndroidBpTest(unittest.TestCase):
def test_rust_flags_normalize_success_case(self):
self.assertDictEqual(
gen_android_bp.normalize_rust_flags(
["--cfg=feature=\"float_roundtrip\""]),
{'--cfg': {'feature="float_roundtrip"'}})
def test_rust_flags_normalize_value_without_key(self):
self.assertDictEqual(
gen_android_bp.normalize_rust_flags(['-Aunused-imports']),
{'-Aunused-imports': None})
def test_rust_flags_complicated_case_success(self):
self.assertDictEqual(
gen_android_bp.normalize_rust_flags([
"-Aunused-imports", "-Cforce-unwind-tables=no",
"--target=aarch64-linux-android", "--cfg", "feature=X",
"--cfg=feature2=Y"
]), {
'-Aunused-imports': None,
'-Cforce-unwind-tables': {'no'},
'--target': {'aarch64-linux-android'},
'--cfg': {'feature=X', 'feature2=Y'}
})
def test_rust_flags_throw_invalid_value_no_key(self):
with self.assertRaisesRegex(
ValueError, "Field feature=float_roundtrip does not relate to any key"):
gen_android_bp.normalize_rust_flags(['feature=float_roundtrip'])
def test_rust_flags_throw_invalid_unexpected_equal_signs(self):
with self.assertRaisesRegex(
ValueError,
"Could not normalize flag --cfg=feature=A=B as it has multiple equal signs."
):
gen_android_bp.normalize_rust_flags(['--cfg=feature=A=B'])
def test_get_bindgen_flags(self):
self.assertEqual(
gen_android_bp.get_bindgen_flags(["--bindgen-flags", "flag1", "flag2"]),
["--flag1", "--flag2"])
def test_get_bindgen_flags_empty_should_not_throw(self):
try:
gen_android_bp.get_bindgen_flags(
["--bindgen-flags", "--some_non_bindgen_flag"])
except Exception:
self.fail("Empty bindgen flags should not raise")
def test_get_bindgen_source_stem(self):
self.assertEqual(
gen_android_bp.get_bindgen_source_stem(["a.rs", "foo.cc", "bar.ff"]),
"a")
def test_get_bindgen_source_stem_complex(self):
self.assertEqual(
gen_android_bp.get_bindgen_source_stem(
["foo/bar/file.rs", "path/foo.cc", "bar.ff"]), "file")
def test_get_bindgen_source_stem_multiple_rs_file_throws(self):
with self.assertRaisesRegex(
ValueError,
"Expected a single rust file in the target output but found more than one!"
):
gen_android_bp.get_bindgen_source_stem(
["foo/bar/file.rs", "path/foo.rs", "bar.ff"])
def test_get_bindgen_source_stem_no_rs_file_throws(self):
with self.assertRaisesRegex(
ValueError,
"Expected a single rust file in the target output but found none!"):
gen_android_bp.get_bindgen_source_stem(
["foo/bar/file.cc", "path/foo.h", "bar.ff"])
def test_rust_flag_at_symbol(self):
# @filepath is allowed in rustflags to tell the compiler to write the output
# to that specific file. We don't support it in AOSP but we need to make sure
# that it doesn't cause any issue and is safely ignored.
self.assertDictEqual(gen_android_bp.normalize_rust_flags(['@somefilepath']),
{'@somefilepath': None})
if __name__ == '__main__':
unittest.main()
|
