import unittest
import sh


class Qnand(unittest.TestCase):
    params = None
    __device = "10M"
    __resultlist = None  # resultlist is a python list of python dictionaries

    # varlist: device
    def __init__(self, testname, testfunc, varlist):
        self.params = varlist
        super(Qnand, self).__init__(testfunc)
        if "device" in varlist:
            self.__device = varlist["device"]
        else:
            raise Exception('device param inside Qnand must be defined')
        self._testMethodDoc = testname
        self.__resultlist = []

    def execute(self):
        try:
            p = sh.nandtest("-m", self.__device)
            # save result
            with open('/mnt/station_ramdisk/nand-nandtest.txt', 'w') as outfile:
                n = outfile.write(p.stdout.decode('ascii'))
            outfile.close()
            self.__resultlist.append(
                {
                    "description": "nandtest output",
                    "filepath": "/mnt/station_ramdisk/nand-nandtest.txt",
                    "mimetype": "text/plain"
                }
            )
        except sh.ErrorReturnCode as e:
            self.fail("failed: could not complete nandtest command")

    def getresults(self):
        return self.__resultlist

    def gettextresult(self):
        return ""