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

Move create_test_device() to testing utils

Browse Source
This commit is contained in:
jeremystretch
2021-04-12 16:07:03 -04:00
parent 608bf30bda
commit b4b68c0b00
2 changed files with 16 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
netbox/utilities/testing/utils.py
View File

@ -4,6 +4,8 @@ from contextlib import contextmanager
from django.contrib.auth.models import Permission, User
from dcim.models import Device, DeviceRole, DeviceType, Manufacturer, Site
def post_data(data):
"""
@ -29,6 +31,19 @@ def post_data(data):
return ret
def create_test_device(name):
"""
Convenience method for creating a Device (e.g. for component testing).
"""
site, _ = Site.objects.get_or_create(name='Site 1', slug='site-1')
manufacturer, _ = Manufacturer.objects.get_or_create(name='Manufacturer 1', slug='manufacturer-1')
devicetype, _ = DeviceType.objects.get_or_create(model='Device Type 1', manufacturer=manufacturer)
devicerole, _ = DeviceRole.objects.get_or_create(name='Device Role 1', slug='device-role-1')
device = Device.objects.create(name=name, site=site, device_type=devicetype, device_role=devicerole)
return device
def create_test_user(username='testuser', permissions=None):
"""
Create a User with the given permissions.