mirror/netbox-community-netbox
1
0
Fork 0
mirror of https://github.com/netbox-community/netbox.git synced 2024-05-10 07:54:54 +00:00
Code Activity

Add test for DeviceType YAML export

Browse Source
This commit is contained in:
Jeremy Stretch
2020-01-13 15:35:01 -05:00
parent 0dad9f8901
commit f170a579de
1 changed files with 12 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
netbox/dcim/tests/test_views.py
View File

@@ -1,5 +1,6 @@
import urllib.parse
import yaml
from django.test import Client, TestCase
from django.urls import reverse
@@ -327,6 +328,17 @@ class DeviceTypeTestCase(TestCase):
response = self.client.get('{}?{}'.format(url, urllib.parse.urlencode(params)))
self.assertEqual(response.status_code, 200)
def test_devicetype_export(self):
url = reverse('dcim:devicetype_list')
response = self.client.get('{}?export'.format(url))
self.assertEqual(response.status_code, 200)
data = list(yaml.load_all(response.content, Loader=yaml.SafeLoader))
self.assertEqual(len(data), 3)
self.assertEqual(data[0]['manufacturer'], 'Manufacturer 1')
self.assertEqual(data[0]['model'], 'Device Type 1')
def test_devicetype(self):
devicetype = DeviceType.objects.first()