mirror of
https://github.com/netbox-community/netbox.git
synced 2024-05-10 07:54:54 +00:00
Introduce create_test_virtualmachine()
This commit is contained in:
@ -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.
|
||||
|
Reference in New Issue
Block a user