|
@@ -163,17 +163,17 @@ class ExportTemplateTest(APITestCase):
|
|
|
|
|
|
|
|
super().setUp()
|
|
super().setUp()
|
|
|
|
|
|
|
|
- self.content_type = ContentType.objects.get_for_model(Device)
|
|
|
|
|
|
|
+ content_type = ContentType.objects.get_for_model(Device)
|
|
|
self.exporttemplate1 = ExportTemplate.objects.create(
|
|
self.exporttemplate1 = ExportTemplate.objects.create(
|
|
|
- content_type=self.content_type, name='Test Export Template 1',
|
|
|
|
|
|
|
+ content_type=content_type, name='Test Export Template 1',
|
|
|
template_code='{% for obj in queryset %}{{ obj.name }}\n{% endfor %}'
|
|
template_code='{% for obj in queryset %}{{ obj.name }}\n{% endfor %}'
|
|
|
)
|
|
)
|
|
|
self.exporttemplate2 = ExportTemplate.objects.create(
|
|
self.exporttemplate2 = ExportTemplate.objects.create(
|
|
|
- content_type=self.content_type, name='Test Export Template 2',
|
|
|
|
|
|
|
+ content_type=content_type, name='Test Export Template 2',
|
|
|
template_code='{% for obj in queryset %}{{ obj.name }}\n{% endfor %}'
|
|
template_code='{% for obj in queryset %}{{ obj.name }}\n{% endfor %}'
|
|
|
)
|
|
)
|
|
|
self.exporttemplate3 = ExportTemplate.objects.create(
|
|
self.exporttemplate3 = ExportTemplate.objects.create(
|
|
|
- content_type=self.content_type, name='Test Export Template 3',
|
|
|
|
|
|
|
+ content_type=content_type, name='Test Export Template 3',
|
|
|
template_code='{% for obj in queryset %}{{ obj.name }}\n{% endfor %}'
|
|
template_code='{% for obj in queryset %}{{ obj.name }}\n{% endfor %}'
|
|
|
)
|
|
)
|
|
|
|
|
|
|
@@ -194,7 +194,7 @@ class ExportTemplateTest(APITestCase):
|
|
|
def test_create_exporttemplate(self):
|
|
def test_create_exporttemplate(self):
|
|
|
|
|
|
|
|
data = {
|
|
data = {
|
|
|
- 'content_type': self.content_type.pk,
|
|
|
|
|
|
|
+ 'content_type': 'dcim.device',
|
|
|
'name': 'Test Export Template 4',
|
|
'name': 'Test Export Template 4',
|
|
|
'template_code': '{% for obj in queryset %}{{ obj.name }}\n{% endfor %}',
|
|
'template_code': '{% for obj in queryset %}{{ obj.name }}\n{% endfor %}',
|
|
|
}
|
|
}
|
|
@@ -205,7 +205,7 @@ class ExportTemplateTest(APITestCase):
|
|
|
self.assertHttpStatus(response, status.HTTP_201_CREATED)
|
|
self.assertHttpStatus(response, status.HTTP_201_CREATED)
|
|
|
self.assertEqual(ExportTemplate.objects.count(), 4)
|
|
self.assertEqual(ExportTemplate.objects.count(), 4)
|
|
|
exporttemplate4 = ExportTemplate.objects.get(pk=response.data['id'])
|
|
exporttemplate4 = ExportTemplate.objects.get(pk=response.data['id'])
|
|
|
- self.assertEqual(exporttemplate4.content_type_id, data['content_type'])
|
|
|
|
|
|
|
+ self.assertEqual(exporttemplate4.content_type, ContentType.objects.get_for_model(Device))
|
|
|
self.assertEqual(exporttemplate4.name, data['name'])
|
|
self.assertEqual(exporttemplate4.name, data['name'])
|
|
|
self.assertEqual(exporttemplate4.template_code, data['template_code'])
|
|
self.assertEqual(exporttemplate4.template_code, data['template_code'])
|
|
|
|
|
|
|
@@ -213,17 +213,17 @@ class ExportTemplateTest(APITestCase):
|
|
|
|
|
|
|
|
data = [
|
|
data = [
|
|
|
{
|
|
{
|
|
|
- 'content_type': self.content_type.pk,
|
|
|
|
|
|
|
+ 'content_type': 'dcim.device',
|
|
|
'name': 'Test Export Template 4',
|
|
'name': 'Test Export Template 4',
|
|
|
'template_code': '{% for obj in queryset %}{{ obj.name }}\n{% endfor %}',
|
|
'template_code': '{% for obj in queryset %}{{ obj.name }}\n{% endfor %}',
|
|
|
},
|
|
},
|
|
|
{
|
|
{
|
|
|
- 'content_type': self.content_type.pk,
|
|
|
|
|
|
|
+ 'content_type': 'dcim.device',
|
|
|
'name': 'Test Export Template 5',
|
|
'name': 'Test Export Template 5',
|
|
|
'template_code': '{% for obj in queryset %}{{ obj.name }}\n{% endfor %}',
|
|
'template_code': '{% for obj in queryset %}{{ obj.name }}\n{% endfor %}',
|
|
|
},
|
|
},
|
|
|
{
|
|
{
|
|
|
- 'content_type': self.content_type.pk,
|
|
|
|
|
|
|
+ 'content_type': 'dcim.device',
|
|
|
'name': 'Test Export Template 6',
|
|
'name': 'Test Export Template 6',
|
|
|
'template_code': '{% for obj in queryset %}{{ obj.name }}\n{% endfor %}',
|
|
'template_code': '{% for obj in queryset %}{{ obj.name }}\n{% endfor %}',
|
|
|
},
|
|
},
|
|
@@ -241,7 +241,7 @@ class ExportTemplateTest(APITestCase):
|
|
|
def test_update_exporttemplate(self):
|
|
def test_update_exporttemplate(self):
|
|
|
|
|
|
|
|
data = {
|
|
data = {
|
|
|
- 'content_type': self.content_type.pk,
|
|
|
|
|
|
|
+ 'content_type': 'dcim.device',
|
|
|
'name': 'Test Export Template X',
|
|
'name': 'Test Export Template X',
|
|
|
'template_code': '{% for obj in queryset %}{{ obj.name }}\n{% endfor %}',
|
|
'template_code': '{% for obj in queryset %}{{ obj.name }}\n{% endfor %}',
|
|
|
}
|
|
}
|