1 files changed, 80 insertions, 0 deletions
diff --git a/test/mocks/emssimulator/swm/upgrade-post-check b/test/mocks/emssimulator/swm/upgrade-post-check
new file mode 100755
index 000000000..799afbccc
--- /dev/null
+++ b/test/mocks/emssimulator/swm/upgrade-post-check
@@ -0,0 +1,80 @@
+#!/usr/bin/python
+
+import sys
+import json
+
+import conf
+import ems_util
+
+
+def upgrade_postcheck(pnf_id, old_sw_version, target_sw_version, rule_name, tmp_file=None):
+    ne_info = ems_util.get_ne_info_from_db_by_id(pnf_id)
+
+    if not ne_info:
+        ret_value = {
+            "result": conf.RESULT_FAILURE,
+            "reason": "Can not find NE %s" % pnf_id
+        }
+        return ret_value
+
+    old_sw_version_in_db = ne_info.get("oldSwVersion", "")
+    current_sw_version_in_db = ne_info.get("currentSwVersion", "")
+
+    if old_sw_version != old_sw_version_in_db:
+        ret_value = {
+            "result": conf.RESULT_FAILURE,
+            "reason": "Old SW version %s in PNF is not matched with oldSwVersion %s" %
+                      (old_sw_version_in_db, old_sw_version)
+        }
+        return ret_value
+
+    if target_sw_version != current_sw_version_in_db:
+        ret_value = {
+            "result": conf.RESULT_FAILURE,
+            "reason": "Current SW version %s in PNF is not matched with targetSwVersion %s" %
+                      (current_sw_version_in_db, target_sw_version)
+        }
+        return ret_value
+
+    ne_info["checkStatus"] = conf.STATUS_POSTCHECKED
+    ems_util.update_ne_info(ne_info)
+
+    ret_value = {
+         "result": conf.RESULT_SUCCESS
+    }
+
+    return ret_value
+
+
+def main():
+    #  {{pnfId}} {{oldSwVersion}} {{targetSwVersion}} {{ruleName}} /tmp/tmp-{{Id}}
+
+    if len(sys.argv) < 5:
+        ret_value = {
+            "result": conf.RESULT_FAILURE,
+            "reason": "Missing parameters"
+        }
+        print json.dumps(ret_value)
+        sys.exit(conf.RET_CODE_FAILURE)
+
+    if len(sys.argv) >= 5:
+        pnf_id = sys.argv[1]
+        old_sw_version = sys.argv[2]
+        target_sw_version = sys.argv[3]
+        rule_name = sys.argv[4]
+        tmp_file = None
+
+    if len(sys.argv) >= 6:
+        tmp_file = sys.argv[5]
+
+    ret_value = upgrade_postcheck(pnf_id, old_sw_version, target_sw_version, rule_name, tmp_file)
+    print json.dumps(ret_value)
+
+    if ret_value["result"] == conf.RESULT_SUCCESS:
+        sys.exit(conf.RET_CODE_SUCCESS)
+    else:
+        sys.exit(conf.RET_CODE_FAILURE)
+
+
+if __name__ == '__main__':
+    main()