def test_get_query_ts(self):
with patch.object(HistoryUtil, 'model_value') as mock_model_value:
mock_model_value.return_value = 'bar'
ts_filter = {}
ts_filter['$gte'] = dateutil.parser.parse(
"2021-07-13T21:31:04.912000Z")
ts_filter['$lte'] = dateutil.parser.parse(
"2021-07-13T21:31:45.560000Z")
filter_query = {
"key": "foo",
"dateFrom": "2021-07-13T21:31:04.912000Z",
"dateTo": "2021-07-13T21:31:45.560000Z"
}
returned_query = NotificationHistory.get_query(filter_query)
expected_query = {
'query': {
'metaAttrsFilter.key': 'bar',
'ts': ts_filter
},
'limit_val': 0,
'sort': [('ts', -1)],
'filter': {
'_id': False,
'@timestamp': False,
'@version': False
}
}
assert returned_query == expected_query
def test_get_query_limit(self):
with patch.object(HistoryUtil, 'model_value') as mock_model_value:
mock_model_value.return_value = 'bar'
filter_query = {"key": "foo", "limit": 15}
returned_query = NotificationHistory.get_query(filter_query)
expected_query = {
'query': {
'metaAttrsFilter.key': 'bar'
},
'limit_val': 15,
'sort': [('ts', -1)],
'filter': {
'_id': False,
'@timestamp': False,
'@version': False
}
}
assert returned_query == expected_query