1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
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)
|
