From f170a579de2718761ba087ef78e8f6bd3e4accb1 Mon Sep 17 00:00:00 2001 From: Jeremy Stretch Date: Mon, 13 Jan 2020 15:35:01 -0500 Subject: [PATCH] Add test for DeviceType YAML export --- netbox/dcim/tests/test_views.py | 12 ++++++++++++ 1 file changed, 12 insertions(+) diff --git a/netbox/dcim/tests/test_views.py b/netbox/dcim/tests/test_views.py index 87f7b8489..856862a3e 100644 --- a/netbox/dcim/tests/test_views.py +++ b/netbox/dcim/tests/test_views.py @@ -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()