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
66
67
68
69
70
|
"""
This Molotov script has:
- a global setup fixture that sets a global headers dict
- an init worker fixture that sets the session headers
- 3 scenario
- 2 tear downs fixtures
"""
import json
from molotov import (
get_context,
global_setup,
global_teardown,
scenario,
setup,
teardown,
)
_API = "http://localhost:8080"
_HEADERS = {}
# notice that the global setup, global teardown and teardown
# are not a coroutine.
@global_setup()
def init_test(args):
_HEADERS["SomeHeader"] = "1"
@global_teardown()
def end_test():
print("This is the end")
@setup()
async def init_worker(worker_num, args):
headers = {"AnotherHeader": "1"}
headers.update(_HEADERS)
return {"headers": headers}
@teardown()
def end_worker(worker_num):
print("This is the end for %d" % worker_num)
# @scenario(weight=40)
async def scenario_one(session):
async with session.get(_API) as resp:
if get_context(session).statsd:
get_context(session).statsd.incr("BLEH")
res = await resp.json()
assert res["result"] == "OK"
assert resp.status == 200
@scenario(weight=30)
async def scenario_two(session):
async with session.get(_API) as resp:
assert resp.status == 200
# @scenario(weight=30)
async def scenario_three(session):
somedata = json.dumps({"OK": 1})
async with session.post(_API, data=somedata) as resp:
assert resp.status == 200
|
