From 06f1d152838da9c8f433e5570a6af71b6091b19a Mon Sep 17 00:00:00 2001
From: jeremystretch <jstretch@ns1.com>
Date: Thu, 4 Nov 2021 13:22:21 -0400
Subject: [PATCH] Introduce create_test_virtualmachine()

---
 netbox/utilities/testing/utils.py | 12 ++++++++++++
 1 file changed, 12 insertions(+)

diff --git a/netbox/utilities/testing/utils.py b/netbox/utilities/testing/utils.py
index bec3bc762..466b5e22b 100644
--- a/netbox/utilities/testing/utils.py
+++ b/netbox/utilities/testing/utils.py
@@ -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.