GitHub Viewer
"""Generated native contracts match a clean Python interpreter."""
import os
import re
import runpy
import shutil
import subprocess
import sys
from pathlib import Path
from typing import Literal
import pytest
from pythonnative.sdk.native_types import NativeTypes, quoted
def test_checked_native_contracts_match_python_definitions(tmp_path: Path) -> None:
subprocess.run(
[
sys.executable,
"-c",
"import pythonnative; from pythonnative.sdk.codegen import generate; " "import sys; generate(sys.argv[1])",
str(tmp_path),
],
check=True,
)
root = Path(__file__).parents[1] / "src/pythonnative/native"
for extension, folder in (
("swift", root / "ios/Sources/PythonNativeKit/Generated"),
("kt", root / "android/src/main/java/com/pythonnative/generated"),
):
for name in ("PNContracts", "NativeProps", "NativeModules", "NativeValues"):
assert (tmp_path / f"{name}.{extension}").read_bytes() == (folder / f"{name}.{extension}").read_bytes()
for name in ("components.py", "modules.py"):
compile((tmp_path / name).read_text(), str(tmp_path / name), "exec")
def test_extension_manifests_keep_builtin_type_names(tmp_path: Path) -> None:
# A plugin build loads extension manifests from JSON, which turns tuples
# into lists. An extension component that redeclares View's props must
# not rename the built-in types that hand-written native code spells.
script = (
"import json, sys; import pythonnative; from pythonnative.sdk import schema; "
"from pythonnative.sdk.codegen import generate; "
"badge = json.loads(json.dumps(schema.manifest()['components']['View'], default=str)); "
"badge['name'] = 'AaaBadge'; "
"schema.load_manifest({'protocol': 4, 'yoga': '3.2.1', 'components': {'AaaBadge': badge}}); "
"generate(sys.argv[1])"
)
subprocess.run([sys.executable, "-c", script, str(tmp_path)], check=True)
root = Path(__file__).parents[1] / "src/pythonnative/native"
declaration = re.compile(r"^(?:sealed class|data class|enum class|public enum|public struct) (PN\w+)", re.M)
for extension, folder in (
("swift", root / "ios/Sources/PythonNativeKit/Generated"),
("kt", root / "android/src/main/java/com/pythonnative/generated"),
):
checked = set(declaration.findall((folder / f"NativeValues.{extension}").read_text()))
assert checked str:
return text.split("open class PNViewProps", 1)[1].split("\n}\n", 1)[0]
assert base((tmp_path / f"NativeProps.{extension}").read_text()) == base(
(folder / f"NativeProps.{extension}").read_text()
)
def test_example_extension_contracts_match_the_builtins(tmp_path: Path) -> None:
# The inbox extension's manifest copies View's props; regenerate it in a
# scratch copy so a built-in change can't leave it stale unnoticed.
source = Path(__file__).parents[1] / "examples/inbox-extension"
copy = tmp_path / "inbox-extension"
shutil.copytree(source / "src", copy / "src")
shutil.copy2(source / "generate_contracts.py", copy / "generate_contracts.py")
subprocess.run(
[sys.executable, str(copy / "generate_contracts.py")],
check=True,
env={**os.environ, "PYTHONPATH": str(copy / "src")},
)
for relative in ("inbox_extension/native/schema.json", "inbox_extension/api.py"):
assert (copy / "src" / relative).read_text() == (source / "src" / relative).read_text(), relative
def test_generated_literals_preserve_interpolation_and_control_characters() -> None:
value = '$total "quoted" \\path\n\x00\b'
assert quoted(value) == r'"$total \"quoted\" \\path\n\u{0}\u{8}"'
assert quoted(value, kotlin=True) == r'"\$total \"quoted\" \\path\n\u0000\u0008"'
types = NativeTypes()
types.types({"enum": ["$total", "\x00"]}, "EscapedLiteral")
assert r'"\$total"' in "\n".join(types.kotlin)
assert r'"\u{0}"' in "\n".join(types.swift)
def test_checked_native_record_fixtures_match_python_annotations() -> None:
fixtures = runpy.run_path("scripts/generate-native-contracts.py")["record_fixtures"]()
for path, expected in fixtures.items():
assert path.read_text() == expected
def test_codegen_escapes_kotlin_literals_once(tmp_path: Path, monkeypatch: pytest.MonkeyPatch) -> None:
from pythonnative.sdk.codegen import generate
from pythonnative.sdk.schema import COMPONENTS, ComponentSchema
from pythonnative.sdk.types import type_schema
monkeypatch.setitem(
COMPONENTS, "LiteralFixture", ComponentSchema("LiteralFixture", {"choice": type_schema(Literal["$total"])})
)
generate(tmp_path)
kotlin = (tmp_path / "NativeValues.kt").read_text()
assert r'"\$total"' in kotlin
assert r'"\\$total"' not in kotlin
assert (
(tmp_path / "modules.py")
.read_text()
.startswith('"""Generated Python interfaces. Regenerate from the declarations."""\n\n')
)