Skip to content

Commit af8877c

Browse files
authored
fix for issue #667 (#668)
* fix for issue 667 * Update RELEASE_NOTES.md * Fix for issue #667 * Resolved TODO in test_timescale_types.py * Update RELEASE_NOTES.md
1 parent 1c3ce68 commit af8877c

3 files changed

Lines changed: 19 additions & 20 deletions

File tree

RELEASE_NOTES.md

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -23,6 +23,7 @@
2323
- Added logs in src/wq/core/task.py (#662)
2424
- Replaced entity with getter (#652)
2525
- Resolved TODO in Dockerfile (#680)
26+
- Resolved TODO at src/reporter/tests/test_timescale_types.py (#667)
2627

2728
### Bug fixes
2829

src/reporter/tests/test_timescale_types.py

Lines changed: 1 addition & 20 deletions
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,7 @@
1212
from utils.kvt import merge_dicts
1313

1414
from .geo_queries_fixture import query_1t1ena
15-
from .utils import notify_url
15+
from .utils import notify_url, insert_entities
1616
from src.utils.tests.tenant import gen_tenant_id
1717

1818

@@ -109,25 +109,6 @@ def gen_entity(entity_id: int,
109109
# Similar to gen_entity in test_timescale_insert module in translators.tests.
110110

111111

112-
def insert_entities(entities: Union[List[dict], dict],
113-
service: str = None, service_path: str = None,
114-
expected_status_code=200) -> Response:
115-
headers = {
116-
'Content-Type': 'application/json',
117-
'fiware-Service': service,
118-
'fiware-ServicePath': service_path
119-
}
120-
body = json.dumps({
121-
'data': entities if isinstance(entities, List) else [entities]
122-
})
123-
response = requests.post(notify_url(), data=body, headers=headers)
124-
assert response.status_code == expected_status_code
125-
return response
126-
# TODO: factor out?
127-
# This function could sit in a separate module and be shared across all
128-
# reporter tests.
129-
130-
131112
def entity_name_value_pairs(entity: dict) -> dict:
132113
"""
133114
Transform an NGSI entity ``e`` into the format::

src/reporter/tests/utils.py

Lines changed: 17 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -6,6 +6,7 @@
66
from typing import Callable, Iterable, List, Optional, Tuple, Union
77
import re
88
import random
9+
from requests import Response
910

1011
from translators.factory import translator_for
1112
from translators.sql_translator import ENTITY_ID_COL, TENANT_PREFIX, \
@@ -96,6 +97,22 @@ def get_notification_different_types(
9697
}
9798

9899

100+
def insert_entities(entities: Union[List[dict], dict],
101+
service: str = None, service_path: str = None,
102+
expected_status_code=200) -> Response:
103+
headers = {
104+
'Content-Type': 'application/json',
105+
'fiware-Service': service,
106+
'fiware-ServicePath': service_path
107+
}
108+
body = json.dumps({
109+
'data': entities if isinstance(entities, List) else [entities]
110+
})
111+
response = requests.post(notify_url(), data=body, headers=headers)
112+
assert response.status_code == expected_status_code
113+
return response
114+
115+
99116
def send_notifications(service, notifications, service_path='/'):
100117
assert isinstance(notifications, list)
101118
h = {'Content-Type': 'application/json'}

0 commit comments

Comments
 (0)