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
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
|
# from promise import Promise
# from promise.dataloader import DataLoader
# def id_loader(**options):
# load_calls = []
# def fn(keys):
# load_calls.append(keys)
# return Promise.resolve(keys)
# identity_loader = DataLoader(fn, **options)
# return identity_loader, load_calls
# def test_batches_multiple_requests():
# identity_loader, load_calls = id_loader()
# @Promise.safe
# def safe():
# promise1 = identity_loader.load(1)
# promise2 = identity_loader.load(2)
# return promise1, promise2
# promise1, promise2 = safe()
# value1, value2 = Promise.all([promise1, promise2]).get()
# assert value1 == 1
# assert value2 == 2
# assert load_calls == [[1, 2]]
# def test_batches_multiple_requests_two():
# identity_loader, load_calls = id_loader()
# @Promise.safe
# def safe():
# promise1 = identity_loader.load(1)
# promise2 = identity_loader.load(2)
# return Promise.all([promise1, promise2])
# p = safe()
# value1, value2 = p.get()
# assert value1 == 1
# assert value2 == 2
# assert load_calls == [[1, 2]]
# @Promise.safe
# def test_batches_multiple_requests_safe():
# identity_loader, load_calls = id_loader()
# promise1 = identity_loader.load(1)
# promise2 = identity_loader.load(2)
# p = Promise.all([promise1, promise2])
# value1, value2 = p.get()
# assert value1 == 1
# assert value2 == 2
# assert load_calls == [[1, 2]]
|
