Board: Client Test Suite Initial Commit

1 files changed, 60 insertions, 0 deletions

diff --git a/scripts/PYTHON_SOFTWARE/test/tests/qusb.py b/scripts/PYTHON_SOFTWARE/test/tests/qusb.py
new file mode 100644
index 0000000..44490bc
--- /dev/null
+++ b/scripts/PYTHON_SOFTWARE/test/tests/qusb.py
@@ -0,0 +1,60 @@
+from test.helpers.syscmd import SysCommand
+import unittest
+
+class Qusb(unittest.TestCase):
+
+    def __init__(self, testname, testfunc, devLabel, numPorts):
+        super(Qusb, self).__init__(testfunc)
+        self.__numPorts = numPorts
+        self._testMethodDoc = testname
+        self.__devLabel = devLabel
+        if testname=="USBOTG":
+            self.__usbFileName = "/this_is_an_usb_otg"
+            self.__usbtext = "USBOTG"
+        elif testname=="SATA":
+            self.__usbFileName = "/this_is_a_sata"
+            self.__usbtext = "SATA"
+        else:
+            self.__usbFileName = "/this_is_an_usb_host"
+            self.__usbtext = "USBHOST"
+        self.__numUsbFail=[]
+
+    def execute(self):
+        str_cmd= "lsblk -o LABEL"
+        lsblk_command = SysCommand("lsblk", str_cmd)
+        if lsblk_command.execute() == 0:
+            self.__raw_out = lsblk_command.getOutput()
+            if self.__raw_out == "":
+                return -1
+            lines = lsblk_command.getOutput().splitlines()
+            host_list=[]
+            for i in range(len(lines)):
+                if str(lines[i].decode('ascii'))==self.__devLabel:
+                    host_list.append(i)
+            if len(host_list)==int(self.__numPorts):
+                str_cmd = "lsblk -o MOUNTPOINT"
+                lsblk_command = SysCommand("lsblk", str_cmd)
+                if lsblk_command.execute() == 0:
+                    self.__raw_out = lsblk_command.getOutput()
+                    if self.__raw_out == "":
+                        print("failed: no command output")
+                        self.fail("failed: no command output")
+                    else:
+                        lines = lsblk_command.getOutput().splitlines()
+                        for i in range(len(host_list)):
+                            file_path=str(lines[host_list[i]].decode('ascii')) + self.__usbFileName
+                            usb_file = open(file_path, 'r')
+                            read=usb_file.read()
+                            if read.find(self.__usbtext)!=-1:
+                                print(file_path + "  -->  OK!")
+                            else:
+                                self.fail("failed: could not read from usb {}".format(lines[host_list[i]].decode('ascii')))
+                                self.__numUsbFail.append(host_list[i])
+                            usb_file.close()
+                else:
+                    self.fail("failed: couldn't execute lsblk command")
+
+            else:
+                self.fail("failed: reference and real usb host devices number mismatch")
+        else:
+            self.fail("failed: couldn't execute lsblk command")