import unittest
import importlib
import io
import re
import ast
from contextlib import redirect_stdout

# ──────────────────────────────────────────────────────────────────────────────
# helpers (как в твоём шаблоне)
# ──────────────────────────────────────────────────────────────────────────────

def _dynamic_test(test_case, condition, success_message, failure_message):
    if condition:
        test_case._testMethodName = success_message
        test_case.assertTrue(True, success_message)
    else:
        test_case._testMethodName = failure_message
        test_case.fail(failure_message)

def normalize_text(text):
    text = text.lower()
    text = re.sub(r"\s{2,}", " ", text)
    text = re.sub(r"\s*([,:?])\s*", r"\1 ", text)
    return text.strip()

def change_var(code: str, var_name: str, value: str) -> str:
    tree = ast.parse(code)
    lines = code.splitlines()
    for i, node in enumerate(tree.body):
        if isinstance(node, ast.Assign):
            for target in node.targets:
                if isinstance(target, ast.Name) and target.id == var_name:
                    start_line = node.lineno - 1
                    line = lines[start_line]
                    indent = ' ' * (len(line) - len(line.lstrip()))
                    lines[start_line] = f"{indent}{var_name} = {value}"
                    next_line = len(lines)
                    for next_node in tree.body[i + 1:]:
                        if hasattr(next_node, "lineno"):
                            next_line = next_node.lineno - 1
                            break
                    if next_line > start_line + 1:
                        lines[start_line + 1:next_line] = []
                    return "\\n".join(lines)
    return code

# ──────────────────────────────────────────────────────────────────────────────
# tests
# ──────────────────────────────────────────────────────────────────────────────

class TestTask(unittest.TestCase):
    def setUp(self):
        """Каждый тест начинает с «чистого» импорта user_code и гасит его print'ы при импорте."""
        import sys
        if "user_code" in sys.modules:
            del sys.modules["user_code"]
        buf = io.StringIO()
        with redirect_stdout(buf):
            import user_code  # noqa: F401
        self.import_output = buf.getvalue()
        self.mod = importlib.import_module("user_code")

    def test_sensitive_action_defined(self):
        sensitive_action = getattr(self.mod, "sensitive_action", None)
        _dynamic_test(
            self,
            callable(sensitive_action),
            "sensitive_action is defined",
            "sensitive_action is not defined",
        )

    def test_permission_error_when_not_admin(self):
        # выставляем False и ожидаем PermissionError
        self.mod.IS_ADMIN = False
        sensitive_action = getattr(self.mod, "sensitive_action", None)
        _dynamic_test(
            self,
            callable(sensitive_action),
            "sensitive_action exists for non-admin check",
            "sensitive_action missing for non-admin check",
        )
        try:
            sensitive_action()
            _dynamic_test(
                self,
                False,
                "PermissionError raised when IS_ADMIN is False",
                "PermissionError was not raised when IS_ADMIN is False",
            )
        except PermissionError:
            _dynamic_test(
                self,
                True,
                "PermissionError raised when IS_ADMIN is False",
                "PermissionError was not raised when IS_ADMIN is False",
            )
        except Exception as e:
            _dynamic_test(
                self,
                False,
                "PermissionError raised when IS_ADMIN is False",
                f"Unexpected exception: {type(e)}: {e}",
            )

    def test_allows_execution_when_admin(self):
        # выставляем True и ожидаем корректный возврат
        self.mod.IS_ADMIN = True
        sensitive_action = getattr(self.mod, "sensitive_action", None)
        _dynamic_test(
            self,
            callable(sensitive_action),
            "sensitive_action exists for admin check",
            "sensitive_action missing for admin check",
        )
        try:
            result = sensitive_action()
            _dynamic_test(
                self,
                result == "Sensitive action performed",
                "Function returns correct value when IS_ADMIN is True",
                f"Expected 'Sensitive action performed', got '{result}'",
            )
        except Exception as e:
            _dynamic_test(
                self,
                False,
                "Function executes when IS_ADMIN is True",
                f"Unexpected exception: {type(e)}: {e}",
            )

    def test_import_side_effect_output(self):
        """
        В твоём user_code при импорте:
          - сначала IS_ADMIN = False и вызов sensitive_action() печатает PermissionError
          - затем IS_ADMIN = True и вызов печатает 'Sensitive action performed'
        Проверим, что оба сообщения присутствуют в выводе импорта.
        """
        text = normalize_text(self.import_output)
        has_perm = "permissionerror" in text or "permissionerror:" in text
        has_ok = "sensitive action performed" in text
        _dynamic_test(
            self,
            has_perm and has_ok,
            "Import prints both PermissionError and success line",
            f"Import output missing expected lines. Got: {self.import_output!r}",
        )

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


test_main.py

Master the art of Python decorators, from foundational concepts to advanced applications. This course guides you through the mechanics, use cases, and best practices of decorators, empowering you to write more expressive and reusable Python code.

Explore the foundational concepts of Python decorators, including their syntax, how they work, and why they're useful.

Delve deeper into decorators by exploring arguments, chaining, and built-in decorators.

Explore Python's built-in decorators and practical use cases for custom decorators.

Apply your knowledge of decorators in a series of hands-on coding challenges.

Challenge: Access Control Decorator

Рішення