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
|
"""
This Molotov script has:
- a global setup fixture that sets variables
- an init worker fixture that sets the session headers
- an init session that attachs an object to the current session
- 1 scenario
- 2 tear downs fixtures
"""
import molotov
class SomeObject:
"""Does something smart in real life with the async loop."""
def __init__(self, loop):
self.loop = loop
def cleanup(self):
pass
@molotov.global_setup()
def init_test(args):
molotov.set_var("SomeHeader", "1")
molotov.set_var("endpoint", "http://localhost:8080")
@molotov.setup()
async def init_worker(worker_num, args):
headers = {"AnotherHeader": "1", "SomeHeader": molotov.get_var("SomeHeader")}
return {"headers": headers}
@molotov.setup_session()
async def init_session(worker_num, session):
molotov.get_context(session).attach("ob", SomeObject(loop=session.loop))
@molotov.scenario(100)
async def scenario_one(session):
endpoint = molotov.get_var("endpoint")
async with session.get(endpoint) as resp:
res = await resp.json()
assert res["result"] == "OK"
assert resp.status == 200
@molotov.teardown_session()
async def end_session(worker_num, session):
molotov.get_context(session).ob.cleanup()
@molotov.teardown()
def end_worker(worker_num):
print("This is the end for %d" % worker_num)
@molotov.global_teardown()
def end_test():
print("This is the end of the test.")
|
