def es_get_response(index=0):
""" Creates an example body for a get request """
return {
"_id": f"{get_uuid(index)}",
"_type": "image",
"_source": {
"tags": ["class_diagram", ""],
"created_at": get_date(index),
"updated_at": get_date(index),
"file_url": "test.png",
"file_date": get_date(index),
},
}
def es_get_untagged_image_response():
""" Creates an example body for a get request """
return {
"_id": f"{get_uuid(0)}",
"_type": "image",
"_source": {
"tags": [],
"created_at": get_date(0),
"updated_at": get_date(0),
"file_url": "test.png",
"file_date": get_date(0),
},
}
def es_tags_equals_none_response():
""" An example body where tags are None for regression testing """
return {
"_index": "artifact",
"_type": "image",
"_id": f"{get_uuid(0)}",
"_score": 0.5,
"_source": {
"created_at": get_date(0),
"updated_at": get_date(0),
"file_url": "use_case_diagram.png",
"tags": None,
"file_date": get_date(0),
},
}
def es_find_all_response():
""" Creates an example body for a find_all request """
return {
"docs": [
{
"_index": "artifact",
"_type": "image",
"_id": f"{get_uuid(0)}",
"_source": {
"created_at": get_date(0),
"updated_at": get_date(0),
"file_url": "class_diagram.png",
"tags": ["uml", "class diagram"],
"file_date": get_date(0),
},
},
{
"_index": "artifact",
"_type": "image",
"_id": f"{get_uuid(1)}",
"_source": {
"created_at": get_date(1),
"updated_at": get_date(1),
"file_url": "use_case_diagram.png",
"tags": ["uml", "use case diagram"],
"file_date": get_date(1),
},
},
]
}
def es_search_response_synonyms():
""" Creates an example body for a search request """
return {
"hits": {
"total":
12,
"max_score":
1.0,
"hits": [
{
"_index": "artifact",
"_type": "image",
"_id": f"{get_uuid(0)}",
"_score": 1.0,
"_source": {
"created_at": get_date(0),
"updated_at": get_date(0),
"file_url": "class_diagram.png",
"tags": ["team", "group"],
"file_date": get_date(0),
},
},
{
"_index": "artifact",
"_type": "image",
"_id": f"{get_uuid(1)}",
"_score": 0.5,
"_source": {
"created_at": get_date(1),
"updated_at": get_date(1),
"file_url": "use_case_diagram.png",
"tags": ["group"],
"file_date": get_date(1),
},
},
],
}
}
def es_search_all_response():
""" Creates an example body for a search request """
return {
"hits": {
"total":
12,
"max_score":
1.0,
"hits": [
{
"_index": "artifact",
"_type": "image",
"_id": f"{get_uuid(0)}",
"_score": 1.0,
"_source": {
"created_at": get_date(0),
"updated_at": get_date(0),
"file_url": "class_diagram.png",
"tags": ["uml", "class diagram"],
"file_date": get_date(0),
},
},
{
"_index": "artifact",
"_type": "image",
"_id": f"{get_uuid(1)}",
"_score": 0.5,
"_source": {
"created_at": get_date(1),
"updated_at": get_date(1),
"file_url": "use_case_diagram.png",
"tags": ["uml", "use case diagram", "tomato", "tree"],
"file_date": get_date(1),
},
},
{
"_index": "artifact",
"_type": "image",
"_id": f"{get_uuid(2)}",
"_score": 0.5,
"_source": {
"created_at": get_date(1),
"updated_at": get_date(1),
"file_url": "use_case_diagram.png",
"tags": ["uml", "use case diagram", "tree", "tomato"],
"file_date": get_date(1),
},
},
{
"_index": "artifact",
"_type": "image",
"_id": f"{get_uuid(3)}",
"_score": 0.5,
"_source": {
"created_at": get_date(1),
"updated_at": get_date(1),
"file_url": "use_case_diagram.png",
"tags": ["uml", "use case diagram", "tomato", "tree"],
"file_date": get_date(1),
},
},
{
"_index": "artifact",
"_type": "image",
"_id": f"{get_uuid(4)}",
"_score": 0.5,
"_source": {
"created_at": get_date(1),
"updated_at": get_date(1),
"file_url": "use_case_diagram.png",
"tags": ["uml", "use case diagram", "tomato"],
"file_date": get_date(1),
},
},
{
"_index": "artifact",
"_type": "image",
"_id": f"{get_uuid(5)}",
"_score": 0.5,
"_source": {
"created_at": get_date(1),
"updated_at": get_date(1),
"file_url": "use_case_diagram.png",
"tags": ["uml", "use case diagram"],
"file_date": get_date(1),
},
},
{
"_index": "artifact",
"_type": "image",
"_id": f"{get_uuid(6)}",
"_score": 0.5,
"_source": {
"created_at": get_date(1),
"updated_at": get_date(1),
"file_url": "use_case_diagram.png",
"tags": ["class diagram", "tomato", "apple"],
"file_date": get_date(1),
},
},
{
"_index": "artifact",
"_type": "image",
"_id": f"{get_uuid(7)}",
"_score": 0.5,
"_source": {
"created_at": get_date(1),
"updated_at": get_date(1),
"file_url": "use_case_diagram.png",
"tags": ["apple", "class diagram"],
"file_date": get_date(1),
},
},
{
"_index": "artifact",
"_type": "image",
"_id": f"{get_uuid(8)}",
"_score": 0.5,
"_source": {
"created_at": get_date(1),
"updated_at": get_date(1),
"file_url": "use_case_diagram.png",
"tags": ["tomato", "class diagram", "apple"],
"file_date": get_date(1),
},
},
{
"_index": "artifact",
"_type": "image",
"_id": f"{get_uuid(9)}",
"_score": 0.5,
"_source": {
"created_at": get_date(1),
"updated_at": get_date(1),
"file_url": "use_case_diagram.png",
"tags": ["tomato", "class diagram"],
"file_date": get_date(1),
},
},
{
"_index": "artifact",
"_type": "image",
"_id": f"{get_uuid(10)}",
"_score": 0.5,
"_source": {
"created_at": get_date(1),
"updated_at": get_date(1),
"file_url": "use_case_diagram.png",
"tags": ["apple", "class diagram"],
"file_date": get_date(1),
},
},
{
"_index": "artifact",
"_type": "image",
"_id": f"{get_uuid(11)}",
"_score": 0.5,
"_source": {
"created_at": get_date(1),
"updated_at": get_date(1),
"file_url": "use_case_diagram.png",
"tags": ["uml"],
"file_date": get_date(1),
},
},
],
}
}