1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
|
from django.conf import settings
from django.contrib.auth.models import User
from django.test import TestCase
from django.test.utils import override_settings
from django.urls import reverse
from haystack import connections, reset_search_queries
from haystack.utils.loading import UnifiedIndex
from ..core.models import MockModel
from .test_solr_backend import SolrMockModelSearchIndex, clear_solr_index
@override_settings(DEBUG=True)
class SearchModelAdminTestCase(TestCase):
fixtures = ["base_data.json", "bulk_data.json"]
def setUp(self):
super().setUp()
# With the models setup, you get the proper bits.
# Stow.
self.old_ui = connections["solr"].get_unified_index()
self.ui = UnifiedIndex()
smmsi = SolrMockModelSearchIndex()
self.ui.build(indexes=[smmsi])
connections["solr"]._index = self.ui
# Wipe it clean.
clear_solr_index()
# Force indexing of the content.
smmsi.update(using="solr")
superuser = User.objects.create_superuser(
username="superuser", password="password", email="super@user.com"
)
def tearDown(self):
# Restore.
connections["solr"]._index = self.old_ui
super().tearDown()
def test_usage(self):
reset_search_queries()
self.assertEqual(len(connections["solr"].queries), 0)
self.assertEqual(
self.client.login(username="superuser", password="password"), True
)
# First, non-search behavior.
resp = self.client.get("/admin/core/mockmodel/")
self.assertEqual(resp.status_code, 200)
self.assertEqual(len(connections["solr"].queries), 0)
self.assertEqual(resp.context["cl"].full_result_count, 23)
# Then search behavior.
resp = self.client.get("/admin/core/mockmodel/", data={"q": "Haystack"})
self.assertEqual(resp.status_code, 200)
self.assertEqual(len(connections["solr"].queries), 3)
self.assertEqual(resp.context["cl"].full_result_count, 23)
# Ensure they aren't search results.
self.assertEqual(isinstance(resp.context["cl"].result_list[0], MockModel), True)
result_pks = [i.pk for i in resp.context["cl"].result_list]
self.assertIn(5, result_pks)
# Make sure only changelist is affected.
resp = self.client.get(reverse("admin:core_mockmodel_change", args=(1,)))
self.assertEqual(resp.status_code, 200)
self.assertEqual(resp.context["original"].id, 1)
self.assertTemplateUsed(resp, "admin/change_form.html")
# The Solr query count should be unchanged:
self.assertEqual(len(connections["solr"].queries), 3)
|
