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

Introduce create_test_virtualmachine()

Browse Source
This commit is contained in:
jeremystretch
2021-11-04 13:22:21 -04:00
parent ff3edc9889
commit 06f1d15283
1 changed files with 12 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -7,6 +7,7 @@ from django.utils.text import slugify
from dcim.models import Device, DeviceRole, DeviceType, Manufacturer, Site from dcim.models import Device, DeviceRole, DeviceType, Manufacturer, Site
from extras.models import Tag from extras.models import Tag
from virtualization.models import Cluster, ClusterType, VirtualMachine
def post_data(data): def post_data(data):
@ -46,6 +47,17 @@ def create_test_device(name):
return device return device
def create_test_virtualmachine(name):
"""
Convenience method for creating a VirtualMachine.
"""
cluster_type, _ = ClusterType.objects.get_or_create(name='Cluster Type 1', slug='cluster-type-1')
cluster, _ = Cluster.objects.get_or_create(name='Cluster 1', type=cluster_type)
virtual_machine = VirtualMachine.objects.create(name=name, cluster=cluster)
return virtual_machine
def create_test_user(username='testuser', permissions=None): def create_test_user(username='testuser', permissions=None):
""" """
Create a User with the given permissions. Create a User with the given permissions.