openstack/venus
Code Issues Proposed changes

Add unit test for get_all_index api

Browse Source 
Change-Id: I476dc8977d87ec921672fbb4228d586a8effe97c
This commit is contained in:
songwenping 2021-12-28 07:37:32 +00:00 committed by Wenping Song
parent 3e6ff99a1a
commit 1fa5c5ff1e
1 changed files with 19 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
venus/tests/unit/api/test_search_action.py
View File

@ -13,11 +13,15 @@
# under the License.
import unittest
from unittest import mock
from venus.modules.search.action import SearchCore
class TestSearchAction(unittest.TestCase):
def setUp(self):
super(TestSearchAction, self).setUp()
def test_get_interval(self):
action = SearchCore()
@ -121,6 +125,21 @@ class TestSearchAction(unittest.TestCase):
self.assertEqual(want2, res2)
self.assertEqual(want3, res3)
@mock.patch('venus.common.utils.request_es')
def test_get_all_index_empty(self, mock_req_es):
action = SearchCore()
mock_req_es.return_value = (400, '')
index_names = action.get_all_index('test_index')
self.assertEqual("", index_names)
@mock.patch('venus.common.utils.request_es')
def test_get_all_index(self, mock_req_es):
action = SearchCore()
mock_req_es.return_value = (
200, '[{"index":"index1"},{"index":"index2"}]')
index_names = action.get_all_index('test_index')
self.assertEqual(["index1", "index2"], index_names)
if __name__ == "__main__":
unittest.main()