from __future__ import annotations

import pytest

a: int
b: unknown

def test_toplevel_annotation():
    assert __annotations__['a'] == "int"
    assert __annotations__['b'] == "unknown"


def test_simple_with_target():
    class C:
        a: int = 1
        assert __annotations__["a"] == "int"
        assert a == 1

def test_class_annotation():
    class C:
        a: list[int]
        b: str
        assert "__annotations__" in locals()
    assert C.__annotations__ == {"a": "list[int]", "b": "str"}

def test_func_annotations():
    def f(a: list) -> a ** 39:
        return 12
    assert f.__annotations__ == {"a": "list", "return": "a ** 39"}

def test_async_return_bug():
    async def foo(x) -> Foobar:
        pass
    assert foo.__annotations__ == {"return": "Foobar"}

def test_future_bug():
    source = """
from __future__ import annotations

class A:
    def __init__(
        self,
        appname: str | None = None,
        appauthor: str | None | Literal[False] = None,
        version: str | None = None,
    ) -> None:
        pass
    """
    code = compile(source, "<dummy>", "exec", dont_inherit=1)
    d = {}
    exec(code, d) # used to crash
    assert d["A"].__init__.__annotations__ == {'appname': 'str | None', 'appauthor': 'str | None | Literal[False]', 'version': 'str | None', 'return': 'None'}