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

Use assertHttpStatus for evaluating HTTP response codes

Browse Source
This commit is contained in:
Jeremy Stretch
2020-01-30 18:08:25 -05:00
parent 179abcc79d
commit 67fafb2b9d
7 changed files with 102 additions and 106 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
netbox/circuits/tests/test_views.py
View File

@ -13,7 +13,6 @@ class ProviderTestCase(TestCase):
@classmethod
def setUpTestData(cls):
Provider.objects.bulk_create([
Provider(name='Provider 1', slug='provider-1', asn=65001),
Provider(name='Provider 2', slug='provider-2', asn=65002),
@ -21,24 +20,21 @@ class ProviderTestCase(TestCase):
])
def test_provider_list(self):
url = reverse('circuits:provider_list')
params = {
"q": "test",
}
response = self.client.get('{}?{}'.format(url, urllib.parse.urlencode(params)))
self.assertEqual(response.status_code, 200)
self.assertHttpStatus(response, 200)
def test_provider(self):
provider = Provider.objects.first()
response = self.client.get(provider.get_absolute_url())
self.assertEqual(response.status_code, 200)
self.assertHttpStatus(response, 200)
def test_provider_import(self):
self.add_permissions('circuits.add_provider')
csv_data = (
"name,slug",
"Provider 4,provider-4",
@ -48,7 +44,7 @@ class ProviderTestCase(TestCase):
response = self.client.post(reverse('circuits:provider_import'), {'csv': '\n'.join(csv_data)})
self.assertEqual(response.status_code, 200)
self.assertHttpStatus(response, 200)
self.assertEqual(Provider.objects.count(), 6)
@ -71,7 +67,7 @@ class CircuitTypeTestCase(TestCase):
url = reverse('circuits:circuittype_list')
response = self.client.get(url)
self.assertEqual(response.status_code, 200)
self.assertHttpStatus(response, 200)
def test_circuittype_import(self):
self.add_permissions('circuits.add_circuittype')
@ -85,7 +81,7 @@ class CircuitTypeTestCase(TestCase):
response = self.client.post(reverse('circuits:circuittype_import'), {'csv': '\n'.join(csv_data)})
self.assertEqual(response.status_code, 200)
self.assertHttpStatus(response, 200)
self.assertEqual(CircuitType.objects.count(), 6)
@ -118,13 +114,13 @@ class CircuitTestCase(TestCase):
}
response = self.client.get('{}?{}'.format(url, urllib.parse.urlencode(params)))
self.assertEqual(response.status_code, 200)
self.assertHttpStatus(response, 200)
def test_circuit(self):
circuit = Circuit.objects.first()
response = self.client.get(circuit.get_absolute_url())
self.assertEqual(response.status_code, 200)
self.assertHttpStatus(response, 200)
def test_circuit_import(self):
self.add_permissions('circuits.add_circuit')
@ -138,5 +134,5 @@ class CircuitTestCase(TestCase):
response = self.client.post(reverse('circuits:circuit_import'), {'csv': '\n'.join(csv_data)})
self.assertEqual(response.status_code, 200)
self.assertHttpStatus(response, 200)
self.assertEqual(Circuit.objects.count(), 6)

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

@ -27,7 +27,7 @@ class RegionTestCase(TestCase):
url = reverse('dcim:region_list')
response = self.client.get(url)
self.assertEqual(response.status_code, 200)
self.assertHttpStatus(response, 200)
def test_region_import(self):
self.add_permissions('dcim.add_region')
@ -41,7 +41,7 @@ class RegionTestCase(TestCase):
response = self.client.post(reverse('dcim:region_import'), {'csv': '\n'.join(csv_data)})
self.assertEqual(response.status_code, 200)
self.assertHttpStatus(response, 200)
self.assertEqual(Region.objects.count(), 6)
@ -70,13 +70,13 @@ class SiteTestCase(TestCase):
}
response = self.client.get('{}?{}'.format(url, urllib.parse.urlencode(params)))
self.assertEqual(response.status_code, 200)
self.assertHttpStatus(response, 200)
def test_site(self):
site = Site.objects.first()
response = self.client.get(site.get_absolute_url())
self.assertEqual(response.status_code, 200)
self.assertHttpStatus(response, 200)
def test_site_import(self):
self.add_permissions('dcim.add_site')
@ -90,7 +90,7 @@ class SiteTestCase(TestCase):
response = self.client.post(reverse('dcim:site_import'), {'csv': '\n'.join(csv_data)})
self.assertEqual(response.status_code, 200)
self.assertHttpStatus(response, 200)
self.assertEqual(Site.objects.count(), 6)
@ -116,7 +116,7 @@ class RackGroupTestCase(TestCase):
url = reverse('dcim:rackgroup_list')
response = self.client.get(url)
self.assertEqual(response.status_code, 200)
self.assertHttpStatus(response, 200)
def test_rackgroup_import(self):
self.add_permissions('dcim.add_rackgroup')
@ -130,7 +130,7 @@ class RackGroupTestCase(TestCase):
response = self.client.post(reverse('dcim:rackgroup_import'), {'csv': '\n'.join(csv_data)})
self.assertEqual(response.status_code, 200)
self.assertHttpStatus(response, 200)
self.assertEqual(RackGroup.objects.count(), 6)
@ -153,7 +153,7 @@ class RackRoleTestCase(TestCase):
url = reverse('dcim:rackrole_list')
response = self.client.get(url)
self.assertEqual(response.status_code, 200)
self.assertHttpStatus(response, 200)
def test_rackrole_import(self):
self.add_permissions('dcim.add_rackrole')
@ -167,7 +167,7 @@ class RackRoleTestCase(TestCase):
response = self.client.post(reverse('dcim:rackrole_import'), {'csv': '\n'.join(csv_data)})
self.assertEqual(response.status_code, 200)
self.assertHttpStatus(response, 200)
self.assertEqual(RackRole.objects.count(), 6)
@ -198,7 +198,7 @@ class RackReservationTestCase(TestCase):
url = reverse('dcim:rackreservation_list')
response = self.client.get(url)
self.assertEqual(response.status_code, 200)
self.assertHttpStatus(response, 200)
class RackTestCase(TestCase):
@ -226,13 +226,13 @@ class RackTestCase(TestCase):
}
response = self.client.get('{}?{}'.format(url, urllib.parse.urlencode(params)))
self.assertEqual(response.status_code, 200)
self.assertHttpStatus(response, 200)
def test_rack(self):
rack = Rack.objects.first()
response = self.client.get(rack.get_absolute_url())
self.assertEqual(response.status_code, 200)
self.assertHttpStatus(response, 200)
def test_rack_import(self):
self.add_permissions('dcim.add_rack')
@ -246,7 +246,7 @@ class RackTestCase(TestCase):
response = self.client.post(reverse('dcim:rack_import'), {'csv': '\n'.join(csv_data)})
self.assertEqual(response.status_code, 200)
self.assertHttpStatus(response, 200)
self.assertEqual(Rack.objects.count(), 6)
@ -269,7 +269,7 @@ class ManufacturerTypeTestCase(TestCase):
url = reverse('dcim:manufacturer_list')
response = self.client.get(url)
self.assertEqual(response.status_code, 200)
self.assertHttpStatus(response, 200)
def test_manufacturer_import(self):
self.add_permissions('dcim.add_manufacturer')
@ -283,7 +283,7 @@ class ManufacturerTypeTestCase(TestCase):
response = self.client.post(reverse('dcim:manufacturer_import'), {'csv': '\n'.join(csv_data)})
self.assertEqual(response.status_code, 200)
self.assertHttpStatus(response, 200)
self.assertEqual(Manufacturer.objects.count(), 6)
@ -312,14 +312,14 @@ class DeviceTypeTestCase(TestCase):
}
response = self.client.get('{}?{}'.format(url, urllib.parse.urlencode(params)))
self.assertEqual(response.status_code, 200)
self.assertHttpStatus(response, 200)
def test_devicetype_export(self):
url = reverse('dcim:devicetype_list')
response = self.client.get('{}?export'.format(url))
self.assertEqual(response.status_code, 200)
self.assertHttpStatus(response, 200)
data = list(yaml.load_all(response.content, Loader=yaml.SafeLoader))
self.assertEqual(len(data), 3)
self.assertEqual(data[0]['manufacturer'], 'Manufacturer 1')
@ -329,7 +329,7 @@ class DeviceTypeTestCase(TestCase):
devicetype = DeviceType.objects.first()
response = self.client.get(devicetype.get_absolute_url())
self.assertEqual(response.status_code, 200)
self.assertHttpStatus(response, 200)
def test_devicetype_import(self):
@ -424,7 +424,7 @@ device-bays:
'format': 'yaml'
}
response = self.client.post(reverse('dcim:devicetype_import'), data=form_data, follow=True)
self.assertEqual(response.status_code, 200)
self.assertHttpStatus(response, 200)
dt = DeviceType.objects.get(model='TEST-1000')
@ -491,7 +491,7 @@ class DeviceRoleTestCase(TestCase):
url = reverse('dcim:devicerole_list')
response = self.client.get(url)
self.assertEqual(response.status_code, 200)
self.assertHttpStatus(response, 200)
def test_devicerole_import(self):
self.add_permissions('dcim.add_devicerole')
@ -505,7 +505,7 @@ class DeviceRoleTestCase(TestCase):
response = self.client.post(reverse('dcim:devicerole_import'), {'csv': '\n'.join(csv_data)})
self.assertEqual(response.status_code, 200)
self.assertHttpStatus(response, 200)
self.assertEqual(DeviceRole.objects.count(), 6)
@ -528,7 +528,7 @@ class PlatformTestCase(TestCase):
url = reverse('dcim:platform_list')
response = self.client.get(url)
self.assertEqual(response.status_code, 200)
self.assertHttpStatus(response, 200)
def test_platform_import(self):
self.add_permissions('dcim.add_platform')
@ -542,7 +542,7 @@ class PlatformTestCase(TestCase):
response = self.client.post(reverse('dcim:platform_import'), {'csv': '\n'.join(csv_data)})
self.assertEqual(response.status_code, 200)
self.assertHttpStatus(response, 200)
self.assertEqual(Platform.objects.count(), 6)
@ -581,13 +581,13 @@ class DeviceTestCase(TestCase):
}
response = self.client.get('{}?{}'.format(url, urllib.parse.urlencode(params)))
self.assertEqual(response.status_code, 200)
self.assertHttpStatus(response, 200)
def test_device(self):
device = Device.objects.first()
response = self.client.get(device.get_absolute_url())
self.assertEqual(response.status_code, 200)
self.assertHttpStatus(response, 200)
def test_device_import(self):
self.add_permissions('dcim.add_device')
@ -601,7 +601,7 @@ class DeviceTestCase(TestCase):
response = self.client.post(reverse('dcim:device_import'), {'csv': '\n'.join(csv_data)})
self.assertEqual(response.status_code, 200)
self.assertHttpStatus(response, 200)
self.assertEqual(Device.objects.count(), 6)
@ -639,7 +639,7 @@ class ConsolePortTestCase(TestCase):
url = reverse('dcim:consoleport_list')
response = self.client.get(url)
self.assertEqual(response.status_code, 200)
self.assertHttpStatus(response, 200)
def test_consoleport_import(self):
self.add_permissions('dcim.add_consoleport')
@ -653,7 +653,7 @@ class ConsolePortTestCase(TestCase):
response = self.client.post(reverse('dcim:consoleport_import'), {'csv': '\n'.join(csv_data)})
self.assertEqual(response.status_code, 200)
self.assertHttpStatus(response, 200)
self.assertEqual(ConsolePort.objects.count(), 6)
@ -691,7 +691,7 @@ class ConsoleServerPortTestCase(TestCase):
url = reverse('dcim:consoleserverport_list')
response = self.client.get(url)
self.assertEqual(response.status_code, 200)
self.assertHttpStatus(response, 200)
def test_consoleserverport_import(self):
self.add_permissions('dcim.add_consoleserverport')
@ -705,7 +705,7 @@ class ConsoleServerPortTestCase(TestCase):
response = self.client.post(reverse('dcim:consoleserverport_import'), {'csv': '\n'.join(csv_data)})
self.assertEqual(response.status_code, 200)
self.assertHttpStatus(response, 200)
self.assertEqual(ConsoleServerPort.objects.count(), 6)
@ -743,7 +743,7 @@ class PowerPortTestCase(TestCase):
url = reverse('dcim:powerport_list')
response = self.client.get(url)
self.assertEqual(response.status_code, 200)
self.assertHttpStatus(response, 200)
def test_powerport_import(self):
self.add_permissions('dcim.add_powerport')
@ -757,7 +757,7 @@ class PowerPortTestCase(TestCase):
response = self.client.post(reverse('dcim:powerport_import'), {'csv': '\n'.join(csv_data)})
self.assertEqual(response.status_code, 200)
self.assertHttpStatus(response, 200)
self.assertEqual(PowerPort.objects.count(), 6)
@ -795,7 +795,7 @@ class PowerOutletTestCase(TestCase):
url = reverse('dcim:poweroutlet_list')
response = self.client.get(url)
self.assertEqual(response.status_code, 200)
self.assertHttpStatus(response, 200)
def test_poweroutlet_import(self):
self.add_permissions('dcim.add_poweroutlet')
@ -809,7 +809,7 @@ class PowerOutletTestCase(TestCase):
response = self.client.post(reverse('dcim:poweroutlet_import'), {'csv': '\n'.join(csv_data)})
self.assertEqual(response.status_code, 200)
self.assertHttpStatus(response, 200)
self.assertEqual(PowerOutlet.objects.count(), 6)
@ -847,7 +847,7 @@ class InterfaceTestCase(TestCase):
url = reverse('dcim:interface_list')
response = self.client.get(url)
self.assertEqual(response.status_code, 200)
self.assertHttpStatus(response, 200)
def test_interface_import(self):
self.add_permissions('dcim.add_interface')
@ -861,7 +861,7 @@ class InterfaceTestCase(TestCase):
response = self.client.post(reverse('dcim:interface_import'), {'csv': '\n'.join(csv_data)})
self.assertEqual(response.status_code, 200)
self.assertHttpStatus(response, 200)
self.assertEqual(Interface.objects.count(), 6)
@ -911,7 +911,7 @@ class FrontPortTestCase(TestCase):
url = reverse('dcim:frontport_list')
response = self.client.get(url)
self.assertEqual(response.status_code, 200)
self.assertHttpStatus(response, 200)
def test_frontport_import(self):
self.add_permissions('dcim.add_frontport')
@ -925,7 +925,7 @@ class FrontPortTestCase(TestCase):
response = self.client.post(reverse('dcim:frontport_import'), {'csv': '\n'.join(csv_data)})
self.assertEqual(response.status_code, 200)
self.assertHttpStatus(response, 200)
self.assertEqual(FrontPort.objects.count(), 6)
@ -963,7 +963,7 @@ class RearPortTestCase(TestCase):
url = reverse('dcim:rearport_list')
response = self.client.get(url)
self.assertEqual(response.status_code, 200)
self.assertHttpStatus(response, 200)
def test_rearport_import(self):
self.add_permissions('dcim.add_rearport')
@ -977,7 +977,7 @@ class RearPortTestCase(TestCase):
response = self.client.post(reverse('dcim:rearport_import'), {'csv': '\n'.join(csv_data)})
self.assertEqual(response.status_code, 200)
self.assertHttpStatus(response, 200)
self.assertEqual(RearPort.objects.count(), 6)
@ -1019,7 +1019,7 @@ class DeviceBayTestCase(TestCase):
url = reverse('dcim:devicebay_list')
response = self.client.get(url)
self.assertEqual(response.status_code, 200)
self.assertHttpStatus(response, 200)
def test_devicebay_import(self):
self.add_permissions('dcim.add_devicebay')
@ -1033,7 +1033,7 @@ class DeviceBayTestCase(TestCase):
response = self.client.post(reverse('dcim:devicebay_import'), {'csv': '\n'.join(csv_data)})
self.assertEqual(response.status_code, 200)
self.assertHttpStatus(response, 200)
self.assertEqual(DeviceBay.objects.count(), 6)
@ -1074,7 +1074,7 @@ class InventoryItemTestCase(TestCase):
}
response = self.client.get('{}?{}'.format(url, urllib.parse.urlencode(params)))
self.assertEqual(response.status_code, 200)
self.assertHttpStatus(response, 200)
def test_inventoryitem_import(self):
self.add_permissions('dcim.add_inventoryitem')
@ -1088,7 +1088,7 @@ class InventoryItemTestCase(TestCase):
response = self.client.post(reverse('dcim:inventoryitem_import'), {'csv': '\n'.join(csv_data)})
self.assertEqual(response.status_code, 200)
self.assertHttpStatus(response, 200)
self.assertEqual(InventoryItem.objects.count(), 6)
@ -1154,13 +1154,13 @@ class CableTestCase(TestCase):
}
response = self.client.get('{}?{}'.format(url, urllib.parse.urlencode(params)))
self.assertEqual(response.status_code, 200)
self.assertHttpStatus(response, 200)
def test_cable(self):
cable = Cable.objects.first()
response = self.client.get(cable.get_absolute_url())
self.assertEqual(response.status_code, 200)
self.assertHttpStatus(response, 200)
def test_cable_import(self):
self.add_permissions('dcim.add_cable')
@ -1174,7 +1174,7 @@ class CableTestCase(TestCase):
response = self.client.post(reverse('dcim:cable_import'), {'csv': '\n'.join(csv_data)})
self.assertEqual(response.status_code, 200)
self.assertHttpStatus(response, 200)
self.assertEqual(Cable.objects.count(), 6)
@ -1228,4 +1228,4 @@ class VirtualChassisTestCase(TestCase):
url = reverse('dcim:virtualchassis_list')
response = self.client.get(url)
self.assertEqual(response.status_code, 200)
self.assertHttpStatus(response, 200)

46
netbox/ipam/tests/test_views.py
View File

@ -31,13 +31,13 @@ class VRFTestCase(TestCase):
}
response = self.client.get('{}?{}'.format(url, urllib.parse.urlencode(params)))
self.assertEqual(response.status_code, 200)
self.assertHttpStatus(response, 200)
def test_vrf(self):
vrf = VRF.objects.first()
response = self.client.get(vrf.get_absolute_url())
self.assertEqual(response.status_code, 200)
self.assertHttpStatus(response, 200)
def test_vrf_import(self):
self.add_permissions('ipam.add_vrf')
@ -51,7 +51,7 @@ class VRFTestCase(TestCase):
response = self.client.post(reverse('ipam:vrf_import'), {'csv': '\n'.join(csv_data)})
self.assertEqual(response.status_code, 200)
self.assertHttpStatus(response, 200)
self.assertEqual(VRF.objects.count(), 6)
@ -74,7 +74,7 @@ class RIRTestCase(TestCase):
url = reverse('ipam:rir_list')
response = self.client.get(url)
self.assertEqual(response.status_code, 200)
self.assertHttpStatus(response, 200)
def test_rir_import(self):
self.add_permissions('ipam.add_rir')
@ -88,7 +88,7 @@ class RIRTestCase(TestCase):
response = self.client.post(reverse('ipam:rir_import'), {'csv': '\n'.join(csv_data)})
self.assertEqual(response.status_code, 200)
self.assertHttpStatus(response, 200)
self.assertEqual(RIR.objects.count(), 6)
@ -117,13 +117,13 @@ class AggregateTestCase(TestCase):
}
response = self.client.get('{}?{}'.format(url, urllib.parse.urlencode(params)))
self.assertEqual(response.status_code, 200)
self.assertHttpStatus(response, 200)
def test_aggregate(self):
aggregate = Aggregate.objects.first()
response = self.client.get(aggregate.get_absolute_url())
self.assertEqual(response.status_code, 200)
self.assertHttpStatus(response, 200)
def test_aggregate_import(self):
self.add_permissions('ipam.add_aggregate')
@ -137,7 +137,7 @@ class AggregateTestCase(TestCase):
response = self.client.post(reverse('ipam:aggregate_import'), {'csv': '\n'.join(csv_data)})
self.assertEqual(response.status_code, 200)
self.assertHttpStatus(response, 200)
self.assertEqual(Aggregate.objects.count(), 6)
@ -160,7 +160,7 @@ class RoleTestCase(TestCase):
url = reverse('ipam:role_list')
response = self.client.get(url)
self.assertEqual(response.status_code, 200)
self.assertHttpStatus(response, 200)
def test_role_import(self):
self.add_permissions('ipam.add_role')
@ -174,7 +174,7 @@ class RoleTestCase(TestCase):
response = self.client.post(reverse('ipam:role_import'), {'csv': '\n'.join(csv_data)})
self.assertEqual(response.status_code, 200)
self.assertHttpStatus(response, 200)
self.assertEqual(Role.objects.count(), 6)
@ -203,13 +203,13 @@ class PrefixTestCase(TestCase):
}
response = self.client.get('{}?{}'.format(url, urllib.parse.urlencode(params)))
self.assertEqual(response.status_code, 200)
self.assertHttpStatus(response, 200)
def test_prefix(self):
prefix = Prefix.objects.first()
response = self.client.get(prefix.get_absolute_url())
self.assertEqual(response.status_code, 200)
self.assertHttpStatus(response, 200)
def test_prefix_import(self):
self.add_permissions('ipam.add_prefix')
@ -223,7 +223,7 @@ class PrefixTestCase(TestCase):
response = self.client.post(reverse('ipam:prefix_import'), {'csv': '\n'.join(csv_data)})
self.assertEqual(response.status_code, 200)
self.assertHttpStatus(response, 200)
self.assertEqual(Prefix.objects.count(), 6)
@ -252,13 +252,13 @@ class IPAddressTestCase(TestCase):
}
response = self.client.get('{}?{}'.format(url, urllib.parse.urlencode(params)))
self.assertEqual(response.status_code, 200)
self.assertHttpStatus(response, 200)
def test_ipaddress(self):
ipaddress = IPAddress.objects.first()
response = self.client.get(ipaddress.get_absolute_url())
self.assertEqual(response.status_code, 200)
self.assertHttpStatus(response, 200)
def test_ipaddress_import(self):
self.add_permissions('ipam.add_ipaddress')
@ -272,7 +272,7 @@ class IPAddressTestCase(TestCase):
response = self.client.post(reverse('ipam:ipaddress_import'), {'csv': '\n'.join(csv_data)})
self.assertEqual(response.status_code, 200)
self.assertHttpStatus(response, 200)
self.assertEqual(IPAddress.objects.count(), 6)
@ -301,7 +301,7 @@ class VLANGroupTestCase(TestCase):
}
response = self.client.get('{}?{}'.format(url, urllib.parse.urlencode(params)))
self.assertEqual(response.status_code, 200)
self.assertHttpStatus(response, 200)
def test_vlangroup_import(self):
self.add_permissions('ipam.add_vlangroup')
@ -315,7 +315,7 @@ class VLANGroupTestCase(TestCase):
response = self.client.post(reverse('ipam:vlangroup_import'), {'csv': '\n'.join(csv_data)})
self.assertEqual(response.status_code, 200)
self.assertHttpStatus(response, 200)
self.assertEqual(VLANGroup.objects.count(), 6)
@ -344,13 +344,13 @@ class VLANTestCase(TestCase):
}
response = self.client.get('{}?{}'.format(url, urllib.parse.urlencode(params)))
self.assertEqual(response.status_code, 200)
self.assertHttpStatus(response, 200)
def test_vlan(self):
vlan = VLAN.objects.first()
response = self.client.get(vlan.get_absolute_url())
self.assertEqual(response.status_code, 200)
self.assertHttpStatus(response, 200)
def test_vlan_import(self):
self.add_permissions('ipam.add_vlan')
@ -364,7 +364,7 @@ class VLANTestCase(TestCase):
response = self.client.post(reverse('ipam:vlan_import'), {'csv': '\n'.join(csv_data)})
self.assertEqual(response.status_code, 200)
self.assertHttpStatus(response, 200)
self.assertEqual(VLAN.objects.count(), 6)
@ -405,10 +405,10 @@ class ServiceTestCase(TestCase):
}
response = self.client.get('{}?{}'.format(url, urllib.parse.urlencode(params)))
self.assertEqual(response.status_code, 200)
self.assertHttpStatus(response, 200)
def test_service(self):
service = Service.objects.first()
response = self.client.get(service.get_absolute_url())
self.assertEqual(response.status_code, 200)
self.assertHttpStatus(response, 200)

6
netbox/netbox/tests/test_views.py
View File

@ -1,6 +1,6 @@
import urllib.parse
from django.test import TestCase
from utilities.testing import TestCase
from django.urls import reverse
@ -11,7 +11,7 @@ class HomeViewTestCase(TestCase):
url = reverse('home')
response = self.client.get(url)
self.assertEqual(response.status_code, 200)
self.assertHttpStatus(response, 200)
def test_search(self):
@ -21,4 +21,4 @@ class HomeViewTestCase(TestCase):
}
response = self.client.get('{}?{}'.format(url, urllib.parse.urlencode(params)))
self.assertEqual(response.status_code, 200)
self.assertHttpStatus(response, 200)

10
netbox/secrets/tests/test_views.py
View File

@ -28,7 +28,7 @@ class SecretRoleTestCase(TestCase):
url = reverse('secrets:secretrole_list')
response = self.client.get(url, follow=True)
self.assertEqual(response.status_code, 200)
self.assertHttpStatus(response, 200)
def test_secretrole_import(self):
self.add_permissions('secrets.add_secretrole')
@ -42,7 +42,7 @@ class SecretRoleTestCase(TestCase):
response = self.client.post(reverse('secrets:secretrole_import'), {'csv': '\n'.join(csv_data)})
self.assertEqual(response.status_code, 200)
self.assertHttpStatus(response, 200)
self.assertEqual(SecretRole.objects.count(), 6)
@ -97,13 +97,13 @@ class SecretTestCase(TestCase):
}
response = self.client.get('{}?{}'.format(url, urllib.parse.urlencode(params)), follow=True)
self.assertEqual(response.status_code, 200)
self.assertHttpStatus(response, 200)
def test_secret(self):
secret = Secret.objects.first()
response = self.client.get(secret.get_absolute_url(), follow=True)
self.assertEqual(response.status_code, 200)
self.assertHttpStatus(response, 200)
def test_secret_import(self):
self.add_permissions('secrets.add_secret')
@ -121,5 +121,5 @@ class SecretTestCase(TestCase):
response = self.client.post(reverse('secrets:secret_import'), {'csv': '\n'.join(csv_data)})
self.assertEqual(response.status_code, 200)
self.assertHttpStatus(response, 200)
self.assertEqual(Secret.objects.count(), 6)

10
netbox/tenancy/tests/test_views.py
View File

@ -25,7 +25,7 @@ class TenantGroupTestCase(TestCase):
url = reverse('tenancy:tenantgroup_list')
response = self.client.get(url, follow=True)
self.assertEqual(response.status_code, 200)
self.assertHttpStatus(response, 200)
def test_tenantgroup_import(self):
self.add_permissions('tenancy.add_tenantgroup')
@ -39,7 +39,7 @@ class TenantGroupTestCase(TestCase):
response = self.client.post(reverse('tenancy:tenantgroup_import'), {'csv': '\n'.join(csv_data)})
self.assertEqual(response.status_code, 200)
self.assertHttpStatus(response, 200)
self.assertEqual(TenantGroup.objects.count(), 6)
@ -68,13 +68,13 @@ class TenantTestCase(TestCase):
}
response = self.client.get('{}?{}'.format(url, urllib.parse.urlencode(params)), follow=True)
self.assertEqual(response.status_code, 200)
self.assertHttpStatus(response, 200)
def test_tenant(self):
tenant = Tenant.objects.first()
response = self.client.get(tenant.get_absolute_url(), follow=True)
self.assertEqual(response.status_code, 200)
self.assertHttpStatus(response, 200)
def test_tenant_import(self):
self.add_permissions('tenancy.add_tenant')
@ -88,5 +88,5 @@ class TenantTestCase(TestCase):
response = self.client.post(reverse('tenancy:tenant_import'), {'csv': '\n'.join(csv_data)})
self.assertEqual(response.status_code, 200)
self.assertHttpStatus(response, 200)
self.assertEqual(Tenant.objects.count(), 6)

20
netbox/virtualization/tests/test_views.py
View File

@ -25,7 +25,7 @@ class ClusterGroupTestCase(TestCase):
url = reverse('virtualization:clustergroup_list')
response = self.client.get(url)
self.assertEqual(response.status_code, 200)
self.assertHttpStatus(response, 200)
def test_clustergroup_import(self):
self.add_permissions('virtualization.add_clustergroup')
@ -39,7 +39,7 @@ class ClusterGroupTestCase(TestCase):
response = self.client.post(reverse('virtualization:clustergroup_import'), {'csv': '\n'.join(csv_data)})
self.assertEqual(response.status_code, 200)
self.assertHttpStatus(response, 200)
self.assertEqual(ClusterGroup.objects.count(), 6)
@ -62,7 +62,7 @@ class ClusterTypeTestCase(TestCase):
url = reverse('virtualization:clustertype_list')
response = self.client.get(url)
self.assertEqual(response.status_code, 200)
self.assertHttpStatus(response, 200)
def test_clustertype_import(self):
self.add_permissions('virtualization.add_clustertype')
@ -76,7 +76,7 @@ class ClusterTypeTestCase(TestCase):
response = self.client.post(reverse('virtualization:clustertype_import'), {'csv': '\n'.join(csv_data)})
self.assertEqual(response.status_code, 200)
self.assertHttpStatus(response, 200)
self.assertEqual(ClusterType.objects.count(), 6)
@ -109,13 +109,13 @@ class ClusterTestCase(TestCase):
}
response = self.client.get('{}?{}'.format(url, urllib.parse.urlencode(params)))
self.assertEqual(response.status_code, 200)
self.assertHttpStatus(response, 200)
def test_cluster(self):
cluster = Cluster.objects.first()
response = self.client.get(cluster.get_absolute_url())
self.assertEqual(response.status_code, 200)
self.assertHttpStatus(response, 200)
def test_cluster_import(self):
self.add_permissions('virtualization.add_cluster')
@ -129,7 +129,7 @@ class ClusterTestCase(TestCase):
response = self.client.post(reverse('virtualization:cluster_import'), {'csv': '\n'.join(csv_data)})
self.assertEqual(response.status_code, 200)
self.assertHttpStatus(response, 200)
self.assertEqual(Cluster.objects.count(), 6)
@ -161,13 +161,13 @@ class VirtualMachineTestCase(TestCase):
}
response = self.client.get('{}?{}'.format(url, urllib.parse.urlencode(params)))
self.assertEqual(response.status_code, 200)
self.assertHttpStatus(response, 200)
def test_virtualmachine(self):
virtualmachine = VirtualMachine.objects.first()
response = self.client.get(virtualmachine.get_absolute_url())
self.assertEqual(response.status_code, 200)
self.assertHttpStatus(response, 200)
def test_virtualmachine_import(self):
self.add_permissions('virtualization.add_virtualmachine')
@ -181,5 +181,5 @@ class VirtualMachineTestCase(TestCase):
response = self.client.post(reverse('virtualization:virtualmachine_import'), {'csv': '\n'.join(csv_data)})
self.assertEqual(response.status_code, 200)
self.assertHttpStatus(response, 200)
self.assertEqual(VirtualMachine.objects.count(), 6)