Просмотр исходного кода

Optimize test_get_provider_graphs()

Jeremy Stretch 5 лет назад
Родитель
Сommit
86e5a09b01
1 измененных файлов с 6 добавлено и 15 удалено
  1. 6 15
      netbox/circuits/tests/test_api.py

+ 6 - 15
netbox/circuits/tests/test_api.py

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