105 lines
3.4 KiB
Python
105 lines
3.4 KiB
Python
from __future__ import annotations
|
|
|
|
import ast
|
|
from pathlib import Path
|
|
|
|
import pytest
|
|
|
|
from foreignthon.transpiler import transpile, _detect_lang, _check_shebang
|
|
|
|
# ---------------------------------------------------------------------------
|
|
# All tests use the real foreignthon-es pack — no mocks
|
|
# ---------------------------------------------------------------------------
|
|
|
|
def es(source: str) -> str:
|
|
return transpile(source, "es")
|
|
|
|
|
|
# ---------------------------------------------------------------------------
|
|
# Keywords
|
|
# ---------------------------------------------------------------------------
|
|
|
|
def test_if_else():
|
|
out = es("si x > 0:\n imprimir(x)\nsino:\n pasar")
|
|
assert "if" in out and "else" in out and "pass" in out
|
|
assert "si" not in out and "sino" not in out
|
|
|
|
def test_for_loop():
|
|
out = es("para i en rango(10):\n imprimir(i)")
|
|
assert "for" in out and "in" in out and "range" in out
|
|
|
|
def test_function_def():
|
|
out = es("definir saludar(nombre):\n retornar nombre")
|
|
assert "def" in out and "return" in out
|
|
|
|
def test_class_def():
|
|
out = es("clase Animal:\n pasar")
|
|
assert "class" in out and "pass" in out
|
|
|
|
def test_booleans_and_none():
|
|
out = es("x = Verdadero\ny = Falso\nz = Nada")
|
|
assert "True" in out and "False" in out and "None" in out
|
|
|
|
def test_try_except():
|
|
out = es(
|
|
"intentar:\n"
|
|
" imprimir(x)\n"
|
|
"excepto ErrorDeValor:\n"
|
|
" pasar\n"
|
|
"finalmente:\n"
|
|
" pasar"
|
|
)
|
|
assert "try" in out and "except" in out and "finally" in out
|
|
assert "ValueError" in out
|
|
|
|
# ---------------------------------------------------------------------------
|
|
# Safety — strings and comments must never be touched
|
|
# ---------------------------------------------------------------------------
|
|
|
|
def test_strings_not_transpiled():
|
|
out = es('x = "si esto es para mientras definir"')
|
|
assert '"si esto es para mientras definir"' in out
|
|
|
|
def test_comments_not_transpiled():
|
|
out = es("# si para mientras\nx = 1")
|
|
assert "# si para mientras" in out
|
|
|
|
def test_fstring_not_touched():
|
|
out = es('imprimir(f"si {x} para")')
|
|
assert "si" in out # inside the string, untouched
|
|
|
|
# ---------------------------------------------------------------------------
|
|
# Output is always valid Python
|
|
# ---------------------------------------------------------------------------
|
|
|
|
def test_output_is_valid_python():
|
|
out = es(
|
|
"definir sumar(a, b):\n"
|
|
" retornar a + b\n\n"
|
|
"para i en rango(5):\n"
|
|
" imprimir(sumar(i, 1))\n"
|
|
)
|
|
ast.parse(out) # raises if invalid
|
|
|
|
# ---------------------------------------------------------------------------
|
|
# Language detection
|
|
# ---------------------------------------------------------------------------
|
|
|
|
def test_detect_lang_from_extension():
|
|
assert _detect_lang(Path("script.es.py")) == "es"
|
|
assert _detect_lang(Path("script.ta.py")) == "ta"
|
|
|
|
def test_detect_lang_bad_extension():
|
|
with pytest.raises(ValueError):
|
|
_detect_lang(Path("script.py"))
|
|
|
|
# ---------------------------------------------------------------------------
|
|
# Shebang override
|
|
# ---------------------------------------------------------------------------
|
|
|
|
def test_shebang_override():
|
|
assert _check_shebang("# foreignthon: fr\nsi x:\n pasar", "es") == "fr"
|
|
|
|
def test_shebang_default_when_absent():
|
|
assert _check_shebang("si x:\n pasar", "es") == "es"
|