mirror of
https://github.com/peeringdb/peeringdb.git
synced 2024-05-11 05:55:09 +00:00
fa85504d71
* Qu20020 - Assorted Fixes - PeeringDB version is missing on the error page footer #432 - Inconsistent page widths #446 - Facility countries show up in different languages #448 - sponsor level in search results #449 * fix tests
143 lines
4.7 KiB
Python
143 lines
4.7 KiB
Python
"""
|
|
Unit-tests for quick search functionality - note that advanced search is not
|
|
tested here as that is using the PDB API entirely.
|
|
"""
|
|
import re
|
|
import datetime
|
|
|
|
import pytest
|
|
|
|
from django.test import TestCase, RequestFactory
|
|
|
|
import peeringdb_server.search as search
|
|
import peeringdb_server.models as models
|
|
import peeringdb_server.views as views
|
|
|
|
|
|
class SearchTests(TestCase):
|
|
"""
|
|
Test quick-search functionality
|
|
"""
|
|
|
|
@classmethod
|
|
def setUpTestData(cls):
|
|
|
|
# in case other tests updated the search index through object
|
|
# creation we need to flush it
|
|
search.SEARCH_CACHE["search_index"] = {}
|
|
|
|
cls.instances = {}
|
|
cls.instances_sponsored = {}
|
|
|
|
# create an instance of each searchable model, so we have something
|
|
# to search for
|
|
cls.org = models.Organization.objects.create(name="Test org")
|
|
for model in search.searchable_models:
|
|
if model.handleref.tag == "net":
|
|
kwargs = {"asn": 1}
|
|
else:
|
|
kwargs = {}
|
|
cls.instances[model.handleref.tag] = model.objects.create(
|
|
status="ok", org=cls.org, name="Test %s" % model.handleref.tag,
|
|
**kwargs)
|
|
|
|
# we also need to test that sponsor ship status comes through
|
|
# accordingly
|
|
cls.org_w_sponsorship = models.Organization.objects.create(name="Sponsor org", status="ok")
|
|
cls.sponsorship = models.Sponsorship.objects.create(
|
|
org=cls.org_w_sponsorship,
|
|
start_date=datetime.datetime.now() - datetime.timedelta(days=1),
|
|
end_date=datetime.datetime.now() + datetime.timedelta(days=1),
|
|
level=1);
|
|
|
|
for model in search.searchable_models:
|
|
if model.handleref.tag == "net":
|
|
kwargs = {"asn": 2}
|
|
else:
|
|
kwargs = {}
|
|
cls.instances_sponsored[model.handleref.tag] = model.objects.create(
|
|
status="ok", org=cls.org_w_sponsorship,
|
|
name="Sponsor %s" % model.handleref.tag,
|
|
**kwargs)
|
|
|
|
|
|
def test_search(self):
|
|
"""
|
|
search for entities containing 'Test' - this should return all
|
|
instances we created during setUp
|
|
"""
|
|
|
|
rv = search.search("Test")
|
|
for k, inst in self.instances.items():
|
|
assert k in rv
|
|
assert len(rv[k]) == 1
|
|
assert rv[k][0]["name"] == inst.search_result_name
|
|
assert rv[k][0]["org_id"] == inst.org_id
|
|
|
|
rv = search.search("as1")
|
|
assert len(rv["net"]) == 1
|
|
assert rv["net"][0]["name"] == self.instances["net"].search_result_name
|
|
assert rv["net"][0]["org_id"] == self.instances["net"].org_id
|
|
|
|
rv = search.search("asn1")
|
|
assert len(rv["net"]) == 1
|
|
assert rv["net"][0]["name"] == self.instances["net"].search_result_name
|
|
assert rv["net"][0]["org_id"] == self.instances["net"].org_id
|
|
|
|
def test_sponsor_badges(self):
|
|
"""
|
|
Test that the sponsor badges show up in search result
|
|
"""
|
|
|
|
factory = RequestFactory()
|
|
request = factory.get("/search",{"q":"Sponsor"})
|
|
response = views.request_search(request)
|
|
m = re.findall(re.escape(
|
|
'<a href="/sponsors" class="sponsor silver">'),
|
|
response.content)
|
|
|
|
assert len(m) == 3
|
|
|
|
|
|
def test_search_case(self):
|
|
"""
|
|
search for entities containing 'test' - this should return all
|
|
instances we created during setUp since matching is case-insensitive
|
|
"""
|
|
rv = search.search("test")
|
|
for k, inst in self.instances.items():
|
|
assert k in rv
|
|
assert len(rv[k]) == 1
|
|
assert rv[k][0]["name"] == inst.search_result_name
|
|
|
|
|
|
def test_index_updates(self):
|
|
"""
|
|
test that indexes get updated correctly when objects are created
|
|
or deleted or updated from pending to ok
|
|
"""
|
|
|
|
# this object will be status pending and should not be returned in the search
|
|
# results
|
|
new_ix_p = models.InternetExchange.objects.create(
|
|
status="pending", org=self.org, name="Test IU ix")
|
|
self.test_search()
|
|
|
|
# this object will be status ok, and should show up in the index
|
|
new_ix_o = models.InternetExchange.objects.create(
|
|
status="ok", org=self.org, name="Test IU P ix")
|
|
rv = search.search("test")
|
|
assert len(rv["ix"]) == 2
|
|
|
|
# now we switch the first object to ok as well and it as well should show up in the
|
|
# index
|
|
new_ix_p.status = "ok"
|
|
new_ix_p.save()
|
|
rv = search.search("test")
|
|
assert len(rv["ix"]) == 3
|
|
|
|
#finally we delete both and they should disappear again
|
|
new_ix_p.delete()
|
|
new_ix_o.delete()
|
|
self.test_search()
|