|
@@ -222,7 +222,7 @@ class SearchBackendTestCase(TestCase):
|
|
|
# Unnamed devices render as "<manufacturer> <model> (<pk>)"; prefixes display their scope (a site)
|
|
# Unnamed devices render as "<manufacturer> <model> (<pk>)"; prefixes display their scope (a site)
|
|
|
for site in sites:
|
|
for site in sites:
|
|
|
Device.objects.create(site=site, device_type=device_type, role=role, description='Batched')
|
|
Device.objects.create(site=site, device_type=device_type, role=role, description='Batched')
|
|
|
- Prefix.objects.create(prefix=f'10.0.{site.pk}.0/24', scope=site, description='Batched')
|
|
|
|
|
|
|
+ Prefix.objects.create(prefix=f'10.0.{all_sites.index(site)}.0/24', scope=site, description='Batched')
|
|
|
search_backend.cache(Device.objects.all())
|
|
search_backend.cache(Device.objects.all())
|
|
|
search_backend.cache(Prefix.objects.all())
|
|
search_backend.cache(Prefix.objects.all())
|
|
|
|
|
|
|
@@ -232,15 +232,15 @@ class SearchBackendTestCase(TestCase):
|
|
|
rendered = [(result.name, list(result.display_attrs.values())) for result in results]
|
|
rendered = [(result.name, list(result.display_attrs.values())) for result in results]
|
|
|
return rendered, len(queries)
|
|
return rendered, len(queries)
|
|
|
|
|
|
|
|
- sites = list(Site.objects.order_by('name'))
|
|
|
|
|
- rendered, query_count = render_results(sites[:1])
|
|
|
|
|
|
|
+ all_sites = list(Site.objects.order_by('name'))
|
|
|
|
|
+ rendered, query_count = render_results(all_sites[:1])
|
|
|
self.assertEqual(len(rendered), 2)
|
|
self.assertEqual(len(rendered), 2)
|
|
|
- rendered, more_query_count = render_results(sites[1:])
|
|
|
|
|
|
|
+ rendered, more_query_count = render_results(all_sites[1:])
|
|
|
self.assertEqual(len(rendered), 6)
|
|
self.assertEqual(len(rendered), 6)
|
|
|
self.assertEqual(more_query_count, query_count)
|
|
self.assertEqual(more_query_count, query_count)
|
|
|
|
|
|
|
|
self.assertEqual(sum(name.startswith('Manufacturer 1 Device Type 1') for name, _ in rendered), 3)
|
|
self.assertEqual(sum(name.startswith('Manufacturer 1 Device Type 1') for name, _ in rendered), 3)
|
|
|
- self.assertCountEqual([attrs[0] for name, attrs in rendered if name.startswith('10.0.')], sites)
|
|
|
|
|
|
|
+ self.assertCountEqual([attrs[0] for name, attrs in rendered if name.startswith('10.0.')], all_sites)
|
|
|
|
|
|
|
|
|
|
|
|
|
class DeferredCachingTestCase(RQQueueTestMixin, TestCase):
|
|
class DeferredCachingTestCase(RQQueueTestMixin, TestCase):
|