1 files changed, 134 insertions, 0 deletions
diff --git a/tests/test_cases/config/runner_test.py b/tests/test_cases/config/runner_test.py
new file mode 100644
index 0000000..2f0920e
--- /dev/null
+++ b/tests/test_cases/config/runner_test.py
@@ -0,0 +1,134 @@
+# Copyright 2024 Huawei Cloud Computing Technology Co., Ltd.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+from runner import (
+    compute_value,
+    get_tokens,
+    handle01,
+    replace_value,
+    undefine,
+)
+
+from hypothesis import given
+from hypothesis.strategies import text
+import re
+
+
+def test_get_tokens():
+    magic_string = "cmakedefine"
+    match, is_01 = get_tokens(
+        f"# \t \t {magic_string} \t  TOKEN_KEY  \t\t  TOKEN_VALUE \t\n", magic_string
+    )
+    assert match
+    assert not is_01
+    groups = match.groups()
+    assert len(groups) == 6
+    assert groups[1] == magic_string
+    assert groups[3] == "TOKEN_KEY"
+    assert groups[5] == "TOKEN_VALUE \t"
+
+    match, _ = get_tokens("#cmakedefine TOKEN_KEY TOKEN_VALUE", magic_string)
+    assert match
+    groups = match.groups()
+    assert groups[1] == magic_string
+    assert groups[3] == "TOKEN_KEY"
+    assert groups[5] == "TOKEN_VALUE"
+
+    match, _ = get_tokens("#cmakedefine TOKEN_KEY @TOKEN_VALUE@", magic_string)
+    assert match
+    groups = match.groups()
+    assert len(groups) == 6
+    assert groups[1] == magic_string
+    assert groups[3] == "TOKEN_KEY"
+    assert groups[5] == "@TOKEN_VALUE@"
+
+    match, _ = get_tokens("#cmakedefine TOKEN_KEY ${TOKEN_VALUE}", magic_string)
+    assert match
+    groups = match.groups()
+    assert len(groups) == 6
+    assert groups[1] == magic_string
+    assert groups[3] == "TOKEN_KEY"
+    assert groups[5] == "${TOKEN_VALUE}"
+
+    match, _ = get_tokens("#cmakedefine TOKEN_KEY", magic_string)
+    assert match
+    groups = match.groups()
+    assert len(groups) == 6
+    assert groups[1] == magic_string
+    assert groups[3] == "TOKEN_KEY"
+    assert groups[5] == ""
+
+    match, _ = get_tokens("#differnt_magic_string TOKEN_KEY", magic_string)
+    assert not match
+
+    match, is_01 = get_tokens(f"#{magic_string}01 TOKEN_KEY", magic_string)
+    assert match
+    assert is_01
+
+    match, _ = get_tokens(f'#cmakedefine FOO "@FOO@"', "cmakedefine")
+    groups = match.groups()
+    assert groups[5] == '"@FOO@"'
+
+
+def test_handle_01():
+    line = "#cmakedefine01 FOO"
+    tokens, _ = get_tokens(line, "cmakedefine")
+    assert handle01(line, tokens, True) == "#define FOO 1"
+    assert handle01(line, tokens, False) == "#define FOO 0"
+
+
+def test_undefine():
+    lines = [
+        "#cmakedefine FOO",
+        "#cmakedefine FOO foo",
+        "#cmakedefine FOO ${FOO}",
+        "#cmakedefine FOO @FOO@",
+    ]
+    for line in lines:
+        tokens, _ = get_tokens(line, "cmakedefine")
+        assert undefine(tokens) == "/* #undef FOO */"
+
+
+def test_replace_value():
+    line = "#cmakedefine FOO whatever@VVV@${FFF}"
+    tokens, _ = get_tokens(line, "cmakedefine")
+    groups = tokens.groups()
+    assert len(groups) == 6
+    assert groups[1] == "cmakedefine"
+    assert groups[3] == "FOO"
+    assert groups[5] == "whatever@VVV@${FFF}"
+    assert replace_value(tokens, "FOO", "VALUE") == "#define FOO VALUE"
+
+
+def test_compute_value():
+    token = "@KEY1@foo${KEY2}@KEY3@@KEY4@"
+    key_value_dict = {
+        "KEY1": "value1",
+        "KEY2": "value2",
+        "KEY3": "value3",
+        "KEY4": "value4",
+        "KEY5": "0",
+    }
+
+    assert compute_value(token, True, key_value_dict) == "value1foo${KEY2}value3value4"
+    assert compute_value(token, False, key_value_dict) == "value1foovalue2value3value4"
+    assert compute_value("@KEY5@", True, key_value_dict) == "0"
+
+
+@given(text())
+def test_substitution_with_given_value(s):
+    kv = {"FOO": s}
+    assert compute_value("@FOO@", True, kv) == s
+    assert compute_value("${FOO}", False, kv) == s
+    assert compute_value("${FOO}", True, kv) == "${FOO}"