import ast
import os
import unittest

from _common import SRC

# Ограничение проверки: ast.parse(feature_version=(3, 9)) ловит синтаксис 3.10+ (match и т. п.) и сторонние
# импорты; API и конструкции новее 3.9 (например, `X | Y` — синтаксис 3.10: аннотации функций
# вычисляются при импорте и их ловит прогон на /usr/bin/python3 3.9.6, аннотации локальных переменных не вычисляются), не проверяет.


class Py39Test(unittest.TestCase):
    FILES = ("agent.py", "run.py")

    def test_parses_as_python_39(self):
        for name in self.FILES:
            with self.subTest(name), open(os.path.join(SRC, name), encoding="utf-8") as f:
                ast.parse(f.read(), feature_version=(3, 9))

    def test_no_third_party_imports(self):
        allowed = {"argparse", "base64", "json", "os", "signal", "ssl", "http", "sys", "time", "urllib", "datetime", "typing", "uuid", "fcntl", "hashlib", "re", "subprocess"}
        for name in self.FILES:
            with self.subTest(name), open(os.path.join(SRC, name), encoding="utf-8") as f:
                tree = ast.parse(f.read())
            for node in ast.walk(tree):
                if isinstance(node, ast.Import):
                    for a in node.names:
                        self.assertIn(a.name.split(".")[0], allowed, "%s: %s" % (name, a.name))
                elif isinstance(node, ast.ImportFrom):
                    self.assertIn((node.module or "").split(".")[0], allowed, "%s: %s" % (name, node.module))

if __name__ == "__main__":
    unittest.main()