1#! /usr/bin/env python3
2
3from hwdata import PCI, PNP, USB
4
5# Test PCI IDs
6pci_vendor_id = '1af4'
7pci_device_id = '1003'
8
9pci = PCI()
10
11pci_vendor = pci.get_vendor(pci_vendor_id)
12print("PCI Vendor: %s" % pci_vendor)
13assert (pci_vendor == "Red Hat, Inc.")
14
15pci_device = pci.get_device(pci_vendor_id, pci_device_id)
16print("PCI Device: %s" % pci_device)
17assert (pci_device == "Virtio console")
18
19# Test USB IDs
20usb_vendor_id = '1d6b'
21usb_device_id = '0001'
22
23usb = USB()
24
25usb_vendor = usb.get_vendor(usb_vendor_id)
26print("USB Vendor: %s" % usb_vendor)
27assert (usb_vendor == "Linux Foundation")
28
29usb_device = usb.get_device(usb_vendor_id, usb_device_id)
30print("USB Device: %s" % usb_device)
31assert (usb_device == "1.1 root hub")
32
33# Test PNP IDs
34pnp_id = 'RHT'
35
36pnp = PNP()
37
38pnp_vendor = pnp.get_vendor(pnp_id)
39print("PNP Vendor: %s" % pnp_vendor)
40assert (pnp_vendor == "Red Hat, Inc.")
41