def test_init_methods(self):
"""Checks that the variables are correctly set"""
# Processor class
processor = Processor("x1", "x2")
self.assertEquals("x1", processor.vendor_id)
self.assertEquals("x2", processor.model_name)
# CPU class
cpu = CPU("Intel", "Xeon", "x86_64", "e6333", "2600Mhz", "yes", 2,
"cache", "111")
self.assertEquals("Intel", cpu.vendor_id)
self.assertEquals("Xeon", cpu.model_name)
self.assertEquals("x86_64", cpu.arch)
self.assertEquals("2600Mhz", cpu.speed)
self.assertEquals("yes", cpu.fpu)
self.assertEquals(2, cpu.cores)
self.assertEquals("cache", cpu.cache)
self.assertEquals("111", cpu.flags)
# GPU
gpu = GPU("AMD", "Raedon")
self.assertEquals("AMD", gpu.vendor_id)
self.assertEquals("Raedon", gpu.model_name)
self.assertEquals([], gpu.memory)
# MCP
mcp = MCP("Intel", "Phi")
self.assertEquals("Intel", mcp.vendor_id)
self.assertEquals("Phi", mcp.model_name)
self.assertEquals([], mcp.memory)
"asset_tag":
dmi.get("System")[0].get("UUID", "Unknown")
}
device_data["device_type"] = utils.goc_device_type(
netbox,
dmi.get("System")[0].get("Product Name", "Unknown"),
device_data["manufacturer"])
# Get or Create device
device = utils.goc_device(netbox, device_data)
print(" => %s " % device.name)
# All device get with dmidecode
item_types = {
"BIOS": lambda item_data: Bios(item_data),
"Baseboard": lambda item_data: BaseBord(item_data),
"Processor": lambda item_data: Processor(item_data),
"Memory Device": lambda item_data: MemoryDevice(item_data),
}
for item_type, item_func in item_types.items():
print(item_type)
for item in dmi.get(item_type):
item_func(item).create(device, netbox)
# Get disks
lsblk_command = "lsblk -SJ -o KNAME,VENDOR,MODEL,SERIAL,REV,SIZE,STATE"
proc = subprocess.Popen(
["ssh", os.environ.get("INVENTORY_HOST"), "sudo"] +
lsblk_command.split(" "),
stderr=subprocess.STDOUT,
stdout=subprocess.PIPE)
stdout, stderr = proc.communicate()