Tests

Author: bogdandm

test/test_cli/data/unicode.json

@@ -0,0 +1,22 @@
+{
+    "_id": "5b45bd7bfdc0754510853734",
+    "名称": "MaVol_screen",
+    "路径": "https://gitlab.com/jinniu-stra/strawman/tree/master/src/screen/MaVol_screen",
+    "标识符": "02180706mav001",
+    "备注": "",
+    "雷达图": {
+        "盈利": "4.83",
+        "抗风险": "4.28",
+        "稳定": "4.83",
+        "择股": "5.0"
+    },
+    "原理说明": "对均线和成交量进行评估选股。",
+    "中文名称": "一阳穿多均",
+    "套装配置": "",
+    "入选理由": "该股近期在60日均线之上，当前同时上穿5、10、20和30日均线，成交量是前一日的2倍，短期强势",
+    "策略话术": "",
+    "策略优势": "",
+    "状态": "稳定在线",
+    "装备详情": "http://bbs.jinniuai.com/t/topic/69",
+    "截止时间": "2019-06-11 00:00:00"
+}

test/test_cli/test_script.py

@@ -75,11 +75,23 @@ def test_help():
 
     pytest.param(f"""{executable} -l User - "{test_data_path / 'users.json'}" --strings-converters""",
                  id="users_strings_converters"),
+    pytest.param(f"""{executable} -m SomeUnicode "{test_data_path / 'unicode.json'}" """,
+                 id="convert_unicode"),
+    pytest.param(f"""{executable} -m SomeUnicode "{test_data_path / 'unicode.json'}" --no-unidecode""",
+                 id="dont_convert_unicode"),
+    pytest.param(f"""{executable} -m SomeUnicode "{test_data_path / 'unicode.json'}" --disable-unicode-conversion""",
+                 id="dont_convert_unicode_2"),
 ]
 
 
-def _validate_result(proc: subprocess.Popen) -> Tuple[str, str]:
+def _validate_result(proc: subprocess.Popen, output=None, output_file: Path = None) -> Tuple[str, str]:
     stdout, stderr = map(bytes.decode, proc.communicate())
+    if output_file:
+        assert output is None
+        with output_file.open(encoding='utf-8') as f:
+            output = f.read()
+    if output:
+        stdout = output
     assert not stderr, stderr
     assert stdout, stdout
     assert proc.returncode == 0
@@ -153,3 +165,12 @@ def test_wrong_arguments(command):
     print("Command:", command)
     proc = subprocess.Popen(command, shell=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
     _validate_result(proc)
+
+
+@pytest.mark.parametrize("command", test_commands)
+def test_script_output_file(command):
+    file = tmp_path / 'out.py'
+    command += f" -o {file}"
+    proc = subprocess.Popen(command, shell=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
+    stdout, stderr = _validate_result(proc, output_file=file)
+    print(stdout)

test/test_code_generation/test_models_code_generator.py

@@ -206,3 +206,89 @@ def test_absolute_model_ref():
     assert wrapper.to_typing_code()[1] == "List[List['TestModel']]"
     with AbsoluteModelRef.inject({test_model: test_model}):
         assert wrapper.to_typing_code()[1] == "List[List['TestModel.TestModel']]"
+
+
+test_unicode_data = [
+    pytest.param(
+        model_factory("Test", {
+            "foo": int,
+            "bar": int,
+            "baz": float
+        }),
+        {'convert_unicode': True},
+        trim("""
+        class Test:
+            foo: int
+            bar: int
+            baz: float
+        """),
+        id="test_pytest_setup"
+    ),
+    pytest.param(
+        model_factory("Test", {
+            "поле1": int,
+            "bar": int,
+            "baz": float
+        }),
+        {'convert_unicode': True},
+        trim("""
+        class Test:
+            pole1: int
+            bar: int
+            baz: float
+        """),
+        id="test_field_on"
+    ),
+    pytest.param(
+        model_factory("Test", {
+            "поле1": int,
+            "bar": int,
+            "baz": float
+        }),
+        {'convert_unicode': False},
+        trim("""
+        class Test:
+            поле1: int
+            bar: int
+            baz: float
+        """),
+        id="test_field_off"
+    ),
+    pytest.param(
+        model_factory("Тест", {
+            "поле1": int,
+            "bar": int,
+            "baz": float
+        }),
+        {'convert_unicode': True},
+        trim("""
+        class Test:
+            pole1: int
+            bar: int
+            baz: float
+        """),
+        id="test_field_on"
+    ),
+    pytest.param(
+        model_factory("Тест", {
+            "поле1": int,
+            "bar": int,
+            "baz": float
+        }),
+        {'convert_unicode': False},
+        trim("""
+        class Тест:
+            поле1: int
+            bar: int
+            baz: float
+        """),
+        id="test_classname_off"
+    ),
+]
+
+
+@pytest.mark.parametrize("value,kwargs,expected", test_unicode_data)
+def test_generated(value: ModelMeta, kwargs: dict, expected: str):
+    generated = generate_code(([{"model": value, "nested": []}], {}),
+                              GenericModelCodeGenerator, class_generator_kwargs=kwargs)
+    assert generated.rstrip() == expected, generated

test/test_code_generation/test_typing.py

@@ -56,10 +56,6 @@ def model(data: dict, name: str):
     return ModelPtr(meta)
 
 
-class TestModel:
-    pass
-
-
 # MetaData | Tuple[import_stmnt, type]
 test_data = [
     pytest.param(