def test_simple_negated_search(self):
self.upload_document(label='second_doc')
self.grant_access(
obj=self.test_document, permission=permission_document_view
)
queryset = document_search.search(
{'q': '-non_valid second'}, user=self._test_case_user
)
self.assertEqual(queryset.count(), 1)
queryset = document_search.search(
{'label': '-second'}, user=self._test_case_user
)
self.assertEqual(queryset.count(), 0)
queryset = document_search.search(
{'label': '-second -Mayan'}, user=self._test_case_user
)
self.assertEqual(queryset.count(), 0)
queryset = document_search.search(
{'label': '-second OR -Mayan'}, user=self._test_case_user
)
self.assertEqual(queryset.count(), 1)
queryset = document_search.search(
{'label': '-non_valid -second'}, user=self._test_case_user
)
self.assertEqual(queryset.count(), 0)
def test_search_with_dashed_content(self):
self.upload_document(label='second-document')
self.grant_access(obj=self.test_document,
permission=permission_document_view)
queryset = document_search.search({'label': '-second-document'},
user=self._test_case_user)
self.assertEqual(queryset.count(), 0)
queryset = document_search.search({'label': '-"second-document"'},
user=self._test_case_user)
self.assertEqual(queryset.count(), 0)
def test_meta_only(self):
self.upload_document(label='first_doc')
self.grant_access(obj=self.test_document,
permission=permission_document_view)
queryset = document_search.search({'q': 'OR first'},
user=self._test_case_user)
self.assertEqual(queryset.count(), 1)
def test_advanced_search_after_related_name_change(self):
# Test versions__filename
self.upload_document()
self.grant_access(obj=self.test_document,
permission=permission_document_view)
queryset = document_search.search({'label': self.test_document.label},
user=self._test_case_user)
self.assertEqual(queryset.count(), 1)
self.assertTrue(self.test_document in queryset)
# Test versions__mimetype
queryset = document_search.search(
{'versions__mimetype': self.test_document.file_mimetype},
user=self._test_case_user)
self.assertEqual(queryset.count(), 1)
self.assertTrue(self.test_document in queryset)
def test_search_field_transformation_functions(self):
self.upload_document()
self.grant_access(obj=self.test_document,
permission=permission_document_view)
queryset = document_search.search(
{'uuid': force_text(self.test_document.uuid)},
user=self._test_case_user)
self.assertEqual(queryset.count(), 1)
def test_simple_or_search(self):
self.upload_document(label='first_doc')
self.grant_access(obj=self.test_documents[0],
permission=permission_document_view)
self.upload_document(label='second_doc')
self.grant_access(obj=self.test_documents[1],
permission=permission_document_view)
queryset = document_search.search({'q': 'first OR second'},
user=self._test_case_user)
self.assertEqual(queryset.count(), 2)
self.assertTrue(self.test_documents[0] in queryset)
self.assertTrue(self.test_documents[1] in queryset)
def test_simple_search_after_related_name_change(self):
"""
Test that simple search works after related_name changes to
document versions and document version pages
"""
self.upload_document(label='first_doc')
self.grant_access(obj=self.test_document,
permission=permission_document_view)
queryset = document_search.search({'q': 'first'},
user=self._test_case_user)
self.assertEqual(queryset.count(), 1)
self.assertTrue(self.test_document in queryset)
def test_advanced_search_past_first_page(self):
test_document_label = self.test_documents[0].label
for document in self.test_documents:
self.grant_access(obj=document,
permission=permission_document_view)
# Make sure all documents are returned by the search
queryset = document_search.search({'label': test_document_label},
user=self._test_case_user)
self.assertEqual(queryset.count(), self.test_document_count)
with self.settings(COMMON_PAGINATE_BY=2):
# Functional test for the first page of advanced results
response = self.get(
viewname='search:results',
kwargs={'search_model': document_search.get_full_name()},
data={'label': test_document_label})
# Total (1 - 2 out of 4) (Page 1 of 2)
# 4 results total, 2 pages, current page is 1,
# object in this page: 2
self.assertEqual(response.status_code, 200)
self.assertEqual(response.context['paginator'].object_list.count(),
4)
self.assertEqual(response.context['paginator'].num_pages, 2)
self.assertEqual(response.context['page_obj'].number, 1)
self.assertEqual(response.context['page_obj'].object_list.count(),
2)
# Functional test for the second page of advanced results
response = self.get(
viewname='search:results',
kwargs={'search_model': document_search.get_full_name()},
data={
'label': test_document_label,
'page': 2
})
# Total (3 - 4 out of 4) (Page 2 of 2)
# 4 results total, 2 pages, current page is 1,
# object in this page: 2
self.assertEqual(response.status_code, 200)
self.assertEqual(response.context['paginator'].object_list.count(),
4)
self.assertEqual(response.context['paginator'].num_pages, 2)
self.assertEqual(response.context['page_obj'].number, 2)
self.assertEqual(response.context['page_obj'].object_list.count(),
2)
def _perform_document_search(self):
return document_search.search(
query_string={'q': self.test_document.label}, user=self._test_case_user
)