TEST restructure

1 files changed, 57 insertions, 0 deletions

diff --git a/test-cli/test/helpers/plc.py b/test-cli/test/helpers/plc.py
new file mode 100644
index 0000000..3b00934
--- /dev/null
+++ b/test-cli/test/helpers/plc.py
@@ -0,0 +1,57 @@
+import sh
+from sh import flashcp
+from sh import flash_erase
+from sh import ErrorReturnCode
+from sh import Command
+from test.helpers.gpio import gpio
+
+
+class dcpPLC(object):
+    __nReset = None
+    __Phy = None
+    __Plc = None
+    __mtd_device = None
+    __factool = None
+    __myConfigTool = None
+
+    def __init__(self, plcfactool, mtd_device):
+        # default save firmware
+        self.__nRest = gpio('75', 'out', '0')
+        self.__nRest = gpio('69', 'out', '0')
+        self.__nRest = gpio('78', 'out', '1')
+        self.__factool = plcfactool
+        self.__mtd_device = mtd_device
+        self.__myConfigTool = Command(self.__factool)
+
+    def setSaveFirmwareMode(self):
+        self.__nRest = gpio('75', 'out', '0')
+        self.__nRest = gpio('69', 'out', '0')
+        self.__nRest = gpio('78', 'out', '1')
+
+    def setBootMode(self):
+        self.__nRest = gpio('78', 'out', '0')
+        self.__nRest = gpio('75', 'out', '1')
+        self.__nRest = gpio('69', 'out', '1')
+
+    def SaveFirmware(self, firmare):
+        self.setSaveFirmwareMode()
+        try:
+            flash_erase(self.__mtd_device)
+            flashcp(firmare, self.__mtd_device)
+        except ErrorReturnCode as Error:
+            return False, "plc flash firmware failed {} ".format(Error.exit_code)
+        return True, ''
+
+    def set_plc (self, var, value, password):
+        try:
+            self.__myConfigTool("-o", "SET", "-p", "{}".format(var), '{}'.format(value), "-w", "{}".format(password))
+        except ErrorReturnCode as Error:
+            return False, "set var failed {} {}".format(var,Error.exit_code)
+        return True, ''
+
+    def get_plc (self, var, value, password):
+        try:
+            self.__myConfigTool("-o", "GET", "-p", "{}".format(var), '{}'.format(value), "-w", "{}".format(password))
+        except ErrorReturnCode as Error:
+            return False, "set var failed {} {}".format(var,Error.exit_code)
+        return True, ''