8f937f0417
* half of the way there * add dataclass support * Forbid Computed var shadowing (#3843) * get it right pyright * fix unit tests * rip out more pydantic * fix weird issues with merge_imports * add missing docstring * make special props a list instead of a set * fix moment pyi * actually ignore the runtime error * it's ruff out there --------- Co-authored-by: benedikt-bartscher <31854409+benedikt-bartscher@users.noreply.github.com>
118 lines
3.1 KiB
Python
118 lines
3.1 KiB
Python
import pytest
|
|
|
|
from reflex.utils.imports import (
|
|
ImportDict,
|
|
ImportVar,
|
|
ParsedImportDict,
|
|
merge_imports,
|
|
parse_imports,
|
|
)
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"import_var, expected_name",
|
|
[
|
|
(
|
|
ImportVar(tag="BaseTag"),
|
|
"BaseTag",
|
|
),
|
|
(
|
|
ImportVar(tag="BaseTag", alias="AliasTag"),
|
|
"BaseTag as AliasTag",
|
|
),
|
|
(
|
|
ImportVar(tag="BaseTag", is_default=True),
|
|
"BaseTag",
|
|
),
|
|
(
|
|
ImportVar(tag="BaseTag", is_default=True, alias="AliasTag"),
|
|
"AliasTag",
|
|
),
|
|
(
|
|
ImportVar(tag="BaseTag", is_default=False),
|
|
"BaseTag",
|
|
),
|
|
(
|
|
ImportVar(tag="BaseTag", is_default=False, alias="AliasTag"),
|
|
"BaseTag as AliasTag",
|
|
),
|
|
],
|
|
)
|
|
def test_import_var(import_var, expected_name):
|
|
"""Test that the import var name is computed correctly.
|
|
|
|
Args:
|
|
import_var: The import var.
|
|
expected_name: The expected name.
|
|
"""
|
|
assert import_var.name == expected_name
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"input_1, input_2, output",
|
|
[
|
|
(
|
|
{"react": {"Component"}},
|
|
{"react": {"Component"}, "react-dom": {"render"}},
|
|
{"react": {ImportVar("Component")}, "react-dom": {ImportVar("render")}},
|
|
),
|
|
(
|
|
{"react": {"Component"}, "next/image": {"Image"}},
|
|
{"react": {"Component"}, "react-dom": {"render"}},
|
|
{
|
|
"react": {ImportVar("Component")},
|
|
"react-dom": {ImportVar("render")},
|
|
"next/image": {ImportVar("Image")},
|
|
},
|
|
),
|
|
(
|
|
{"react": {"Component"}},
|
|
{"": {"some/custom.css"}},
|
|
{"react": {ImportVar("Component")}, "": {ImportVar("some/custom.css")}},
|
|
),
|
|
],
|
|
)
|
|
def test_merge_imports(input_1, input_2, output):
|
|
"""Test that imports are merged correctly.
|
|
|
|
Args:
|
|
input_1: The first dict to merge.
|
|
input_2: The second dict to merge.
|
|
output: The expected output dict after merging.
|
|
|
|
"""
|
|
res = merge_imports(input_1, input_2)
|
|
assert res.keys() == output.keys()
|
|
|
|
for key in output:
|
|
assert set(res[key]) == set(output[key])
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"input, output",
|
|
[
|
|
({}, {}),
|
|
(
|
|
{"react": "Component"},
|
|
{"react": [ImportVar(tag="Component")]},
|
|
),
|
|
(
|
|
{"react": ["Component"]},
|
|
{"react": [ImportVar(tag="Component")]},
|
|
),
|
|
(
|
|
{"react": ["Component", ImportVar(tag="useState")]},
|
|
{"react": [ImportVar(tag="Component"), ImportVar(tag="useState")]},
|
|
),
|
|
(
|
|
{"react": ["Component"], "foo": "anotherFunction"},
|
|
{
|
|
"react": [ImportVar(tag="Component")],
|
|
"foo": [ImportVar(tag="anotherFunction")],
|
|
},
|
|
),
|
|
],
|
|
)
|
|
def test_parse_imports(input: ImportDict, output: ParsedImportDict):
|
|
assert parse_imports(input) == output
|