aboutsummaryrefslogtreecommitdiffstats
path: root/src/laas_dashboard/model_test.py
diff options
context:
space:
mode:
Diffstat (limited to 'src/laas_dashboard/model_test.py')
-rw-r--r--src/laas_dashboard/model_test.py110
1 files changed, 110 insertions, 0 deletions
diff --git a/src/laas_dashboard/model_test.py b/src/laas_dashboard/model_test.py
new file mode 100644
index 0000000..ba3ef35
--- /dev/null
+++ b/src/laas_dashboard/model_test.py
@@ -0,0 +1,110 @@
+##############################################################################
+# Copyright (c) 2020 Sawyer Bergeron, Parker Berberian, Sean Smith, and others.
+#
+# All rights reserved. This program and the accompanying materials
+# are made available under the terms of the Apache License, Version 2.0
+# which accompanies this distribution, and is available at
+# http://www.apache.org/licenses/LICENSE-2.0
+##############################################################################
+
+
+from resource_inventory.models import (
+ ResourceProfile,
+ ResourceQuery,
+ Image,
+ DiskProfile,
+ CpuProfile,
+ RamProfile,
+ InterfaceProfile,
+)
+
+
+def rp_has_all_components():
+ """
+ Check that every ResourceProfile has an InterfaceProfile,
+ DiskProfile, CpuProfile, and RamProfile.
+ """
+
+ result = True
+
+ for rp in ResourceProfile.objects.all():
+ ip = InterfaceProfile.objects.filter(host=rp).exists()
+ dp = DiskProfile.objects.filter(host=rp).exists()
+ cp = CpuProfile.objects.filter(host=rp).exists()
+ ram = RamProfile.objects.filter(host=rp).exists()
+
+ if not ip:
+ print("No InterfaceProfiles for host", rp.name)
+ result = False
+
+ if not dp:
+ print("No DiskProfile for host", rp.name)
+ result = False
+
+ if not cp:
+ print("No CpuProfile for host", rp.name)
+ result = False
+
+ if not ram:
+ print("No RamProfile for host", rp.name)
+ result = False
+
+ return result
+
+
+def ip_for_all_ifaces():
+ """
+ Check that every InterfaceProfile for a Resource has
+ an Interface.
+ """
+
+ result = True
+
+ for res in ResourceQuery.filter():
+ iface_set = res.get_interfaces()
+ iface_profile_set = InterfaceProfile.objects.filter(host=res.profile)
+
+ # find out what profiles we have
+ curr_profiles = [iface.profile for iface in iface_set]
+ missing_profiles = set(iface_profile_set) - set(curr_profiles)
+
+ if missing_profiles:
+ print('No interface for profiles', missing_profiles, 'for host', res.name)
+ result = False
+
+ return result
+
+
+def rp_has_image():
+ """
+ Make sure every ResourceProfile has an Image.
+ """
+
+ result = True
+
+ rp_set = ResourceProfile.objects.all()
+ image_set = Image.objects.all()
+ image_profiles = set([image.host_type for image in image_set])
+
+ for rp in rp_set:
+ if rp not in image_profiles:
+ print("ResourceProfile", rp.name, "has no image associated with it.")
+ result = False
+ return result
+
+
+def run_test(test):
+ print('RUNNING TEST', test)
+ result = test()
+ if result:
+ print(test, 'WAS A SUCCESS!')
+ else:
+ print(test, 'FAILED')
+ print('============================================')
+
+
+def run_tests():
+ tests = [rp_has_all_components, ip_for_all_ifaces, rp_has_image]
+
+ for test in tests:
+ run_test(test)