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
|
import datetime
import elasticsearch
from pytest_benchmark.plugin import pytest_benchmark_generate_machine_info
from pytest_benchmark.utils import get_commit_info, get_tag, get_machine_id
from pytest_benchmark.storage.elasticsearch import BenchmarkJSONSerializer
def post(seconds_blocked, request,):
body, doc_id = get_doc(seconds_blocked, request)
url = request.config.getoption("elasticsearch_url")
if url:
es = elasticsearch.Elasticsearch(
hosts=[url],
serializer=BenchmarkJSONSerializer())
es.index(
index='responsiveness',
doc_type='responsiveness',
id=doc_id,
body=body)
else:
print body
def get_doc(seconds_blocked, request):
fullname = request.node._nodeid
name = request.node.name
group = None
marker = request.node.get_marker("responsivness")
if marker:
group = marker.kwargs.get("group")
doc = {
"datetime": datetime.datetime.utcnow().isoformat(),
"commit_info": get_commit_info(),
"fullname": fullname,
"name": name,
"group": group,
"machine_info": pytest_benchmark_generate_machine_info(),
"seconds_blocked": seconds_blocked,
}
# generate a doc id like the one used by pytest-benchmark
machine_id = get_machine_id()
tag = get_tag()
doc_id = machine_id + "_" + tag + "_" + fullname
return doc, doc_id
|
