1 files changed, 44 insertions, 0 deletions
diff --git a/test-cli/test/tasks/flashmemory.py b/test-cli/test/tasks/flashmemory.py
new file mode 100644
index 0000000..ad9d92a
--- /dev/null
+++ b/test-cli/test/tasks/flashmemory.py
@@ -0,0 +1,44 @@
+import sh
+import os
+
+class NAND_Flasher:
+    __Name = None
+    __varlist = None
+    __xmlObj = None
+    __igepflashPath = None
+    __lastError = ''
+
+    def __init__(self, varlist):
+        self.__varlist = varlist
+        self.__xmlObj = varlist['xml']
+        self.__Name = varlist.get('name_flashnand', self.__xmlObj.getKeyVal('NAND_Flasher', 'name', 'NAND_Flasher'))
+        self.__igepflashPath = self.__xmlObj.getKeyVal('NAND_Flasher', 'igep_flash', '/usr/bin/igep-flash')
+        self.__ImagePath = varlist.get('flashimagepath', '')
+        self.__SkipNandtest = varlist.get('skipnandtest', self.__xmlObj.getKeyVal('NAND_Flasher', 'skipnandtest', '1'))
+
+    def getTaskName(self):
+        return self.__Name
+
+    def getError(self):
+        return self.__lastError
+
+    def Execute(self):
+        try:
+            self.__lastError = ''
+            if not os.path.isfile(self.__ImagePath):
+                self.__lastError('Not Image Found: {}'.format(self.__ImagePath))
+                return False, None
+            if self.__SkipNandtest == '1':
+                p = sh.bash('{}'.format(self.__igepflashPath), "--skip-nandtest", "--image", self.__ImagePath)
+            else:
+                p = sh.bash('{}'.format(self.__igepflashPath), "--image", self.__ImagePath)
+            if p.exit_code != 0:
+                return False, None
+        except OSError as e:
+            self.__lastError('Exception: {}'.format(e.strerror))
+            return False, None
+        except Exception as Error:
+            self.__lastError('Exception: {}'.format(Error))
+            return False, None
+        return True, None
+