add mock.patch to test_generate_must_upper

add mock.patch to test_generate_must_upper method

Change-Id: I943f83bc60d5a6cf4e61e311375a139cc55e891a
This commit is contained in:
shanyunfan33 2022-09-11 14:42:09 +08:00
parent bd1b4b7a2f
commit 70a8fcd246

View File

@ -245,7 +245,12 @@ class TestSearchAction(unittest.TestCase):
expected = {'code': 1, 'msg': 'OK', "values": ['val1', 'val2']}
self.assertEqual(expected, result)
def test_generate_must_upper(self):
@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_generate_must_upper(
self, mock_req_es, mock_get_index_names, mock_search_params):
mock_get_index_names.return_value = 'flog-2021.01.03,flog-2021.01.04'
action = SearchCore()
result = action.generate_must({'log_level.keyword': 'test'})
expected = [{'terms': {'log_level.keyword': ['test', 'TEST']}}]