import json from mock import patch from django.core.urlresolvers import reverse from seahub.test_utils import BaseTestCase from seahub.api2.views import Search class SearchTest(BaseTestCase): def setUp(self): self.file_path = self.file self.repo_id = self.repo.id self.url = reverse('api_search') self.mock_has_more = False self.mock_total = 15 self.mock_results = [ { "repo_owner_name": "name", "repo_id": self.repo_id, "name": "lian.png", "repo_owner_contact_email": "freeplant@163.com", "repo_owner_email": "freeplant@163.com", "last_modified": 1469415777, "content_highlight": "", "fullpath": self.file_path, "repo_name": "seafile-design", "is_dir": False, "size": 142758 }, ] @patch('seahub.api2.views.HAS_FILE_SEARCH', True) @patch.object(Search, '_search_keyword') def test_can_search_file(self, mock_search_keyword): mock_search_keyword.return_value = self.mock_results, \ self.mock_total, self.mock_has_more self.login_as(self.user) resp = self.client.get(self.url + '?q=lian') json_resp = json.loads(resp.content) self.assertEqual(200, resp.status_code) @patch('seahub.api2.views.HAS_FILE_SEARCH', True) @patch.object(Search, '_search_keyword') def test_can_not_search_with_invalid_repo_permission(self, mock_search_keyword): mock_search_keyword.return_value = self.mock_results, \ self.mock_total, self.mock_has_more self.login_as(self.admin) resp = self.client.get(self.url + '?q=lian&search_repo=%s' % self.repo_id) self.assertEqual(403, resp.status_code) @patch('seahub.api2.views.HAS_FILE_SEARCH', True) @patch.object(Search, '_search_keyword') def test_can_not_search_without_q_parameter(self, mock_search_keyword): mock_search_keyword.return_value = self.mock_results, \ self.mock_total, self.mock_has_more self.login_as(self.admin) resp = self.client.get(self.url) self.assertEqual(400, resp.status_code)