aboutsummaryrefslogtreecommitdiff
path: root/tests/basics/async_with_return.py
blob: 9af88b839f88c682443ac3b1c8ee5b3de6cd26f5 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
# test async with, escaped by a return

class AContext:
    async def __aenter__(self):
        print('enter')
        return 1
    async def __aexit__(self, exc_type, exc, tb):
        print('exit', exc_type, exc)

async def f1():
    async with AContext():
        print('body')
        return

o = f1()
try:
    o.send(None)
except StopIteration:
    print('finished')

async def f2():
    try:
        async with AContext():
            print('body')
            return
    finally:
        print('finally')

o = f2()
try:
    o.send(None)
except StopIteration:
    print('finished')

async def f3():
    try:
        try:
            async with AContext():
                print('body')
                return
        finally:
            print('finally inner')
    finally:
        print('finally outer')

o = f3()
try:
    o.send(None)
except StopIteration:
    print('finished')