summaryrefslogtreecommitdiff
path: root/tests/responsiveness/elastic.py
blob: fed1506b9442004d53b2588e130b5c22dd968b8a (plain)
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
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(),
    }

    # 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