Merge "Supplement test case for typical_logs"

This commit is contained in:
Zuul 2022-09-05 02:56:31 +00:00 committed by Gerrit Code Review
commit 0a9822d617

View File

@ -562,6 +562,27 @@ class TestSearchAction(unittest.TestCase):
'1660808934')
self.assertEqual(expected, result)
@mock.patch('venus.modules.search.es_template.search_params')
@mock.patch('venus.modules.search.action.SearchCore.get_index_names')
@mock.patch('venus.common.utils.request_es')
def test_typical_logs_no_data_group(
self, mock_req_es, mock_get_index_names, mock_search_params):
mock_get_index_names.return_value = 'flog-2022.08.17,flog-2022.08.18'
mock_req_es.return_value = (200, '{"aggregations": {}}')
action = SearchCore()
expected = {"code": 0, "msg": "no data, no data group"}
result = action.typical_logs('error_stats', '1660722534', '1660808934')
self.assertEqual(expected, result)
result = action.typical_logs('rabbitmq_error_stats', '1660722534',
'1660808934')
self.assertEqual(expected, result)
result = action.typical_logs('mysql_error_stats', '1660722534',
'1660808934')
self.assertEqual(expected, result)
result = action.typical_logs('novalidhost_error_stats', '1660722534',
'1660808934')
self.assertEqual(expected, result)
if __name__ == "__main__":
unittest.main()