|
|
@@ -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)
|