67 lines
1.8 KiB
Python
67 lines
1.8 KiB
Python
|
import uuid
|
||
|
|
import requests
|
||
|
|
import random
|
||
|
|
from multiprocessing import Process
|
||
|
|
from time import time
|
||
|
|
from pprint import pprint
|
||
|
|
|
||
|
|
def size_test():
|
||
|
|
session = requests.session()
|
||
|
|
length = 4 * 1024 * 1024 # ~4MB
|
||
|
|
data = {"id": "a" * length}
|
||
|
|
res = session.post("http://localhost:8080/echo", json=data).json()
|
||
|
|
if {"id": "echo", "obj": data} != res:
|
||
|
|
print("size test: failed")
|
||
|
|
else:
|
||
|
|
print("size test: ok")
|
||
|
|
print()
|
||
|
|
|
||
|
|
P = 200 # number of concurrent processes
|
||
|
|
N = 20 # for each process, the number of posts
|
||
|
|
|
||
|
|
def test():
|
||
|
|
session = requests.session()
|
||
|
|
if {"hello": "world"} \
|
||
|
|
!= session.get("http://localhost:8080/hello").json():
|
||
|
|
print('test failed!')
|
||
|
|
if {"id": "get", "val": 0} \
|
||
|
|
!= session.get("http://localhost:8080/get").json():
|
||
|
|
print('test failed!')
|
||
|
|
cnt = 0
|
||
|
|
for _ in range(N):
|
||
|
|
session_id = str(uuid.uuid4())
|
||
|
|
if random.randint(0, 1) == 0:
|
||
|
|
if {"id": "echo2", "str": session_id} \
|
||
|
|
!= session.get(f"http://localhost:8080/echo/{session_id}").json():
|
||
|
|
print('test failed!')
|
||
|
|
else:
|
||
|
|
data = {"id": session_id}
|
||
|
|
if {"id": "echo", "obj": data} \
|
||
|
|
!= session.post("http://localhost:8080/echo", json=data).json():
|
||
|
|
print('test failed!')
|
||
|
|
cnt += 1
|
||
|
|
if {"id": "get", "val": cnt} \
|
||
|
|
!= session.get("http://localhost:8080/get").json():
|
||
|
|
print('test failed!')
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
size_test()
|
||
|
|
|
||
|
|
processes = [Process(target=test) for _ in range(P)]
|
||
|
|
|
||
|
|
print('starting')
|
||
|
|
|
||
|
|
start = time()
|
||
|
|
|
||
|
|
for p in processes:
|
||
|
|
p.start()
|
||
|
|
|
||
|
|
for p in processes:
|
||
|
|
p.join()
|
||
|
|
|
||
|
|
end = time()
|
||
|
|
|
||
|
|
print('test ended')
|
||
|
|
print('elapsed: ', end - start)
|