from app.sandbox.client import prefilter


def test_prefilter_blocks_hostile_code() -> None:
    for code in ["import subprocess", "import socket", "os.system('ls')", "__import__('os')",
                 "import requests", "eval('1')", "open('/etc/passwd')"]:
        assert prefilter(code) is not None, code


def test_prefilter_allows_normal_code() -> None:
    assert prefilter("import pandas as pd\nprint(pd.DataFrame({'a':[1]}).sum())") is None
    assert prefilter("import numpy_financial as npf\nprint(npf.pmt(0.05/12, 300, -350000))") is None