Optimize test_get_provider_graphs()

This commit is contained in:
Jeremy Stretch 2020-06-05 09:36:38 -04:00
parent 1d5f2fbd11
commit 86e5a09b01

View File

@ -50,22 +50,13 @@ class ProviderTest(APIViewTestCases.APIViewTestCase):
Test retrieval of Graphs assigned to Providers. Test retrieval of Graphs assigned to Providers.
""" """
provider = self.model.objects.first() provider = self.model.objects.first()
provider_ct = ContentType.objects.get(app_label='circuits', model='provider') ct = ContentType.objects.get(app_label='circuits', model='provider')
self.graph1 = Graph.objects.create( graphs = (
type=provider_ct, Graph(type=ct, name='Graph 1', source='http://example.com/graphs.py?site={{ obj.slug }}&foo=1'),
name='Test Graph 1', Graph(type=ct, name='Graph 2', source='http://example.com/graphs.py?site={{ obj.slug }}&foo=2'),
source='http://example.com/graphs.py?provider={{ obj.slug }}&foo=1' Graph(type=ct, name='Graph 3', source='http://example.com/graphs.py?site={{ obj.slug }}&foo=3'),
)
self.graph2 = Graph.objects.create(
type=provider_ct,
name='Test Graph 2',
source='http://example.com/graphs.py?provider={{ obj.slug }}&foo=2'
)
self.graph3 = Graph.objects.create(
type=provider_ct,
name='Test Graph 3',
source='http://example.com/graphs.py?provider={{ obj.slug }}&foo=3'
) )
Graph.objects.bulk_create(graphs)
url = reverse('circuits-api:provider-graphs', kwargs={'pk': provider.pk}) url = reverse('circuits-api:provider-graphs', kwargs={'pk': provider.pk})
response = self.client.get(url, **self.header) response = self.client.get(url, **self.header)