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 extras.models import Tag
from virtualization.models import Cluster, ClusterType, VirtualMachine
def post_data(data):
@ -46,6 +47,17 @@ def create_test_device(name):
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):
"""
Create a User with the given permissions.