Add tests for extension workflow

tests_e2e/GuestAgentDcrTestExtension/.gitignore

@@ -0,0 +1,71 @@
+# Byte-compiled / optimized / DLL files
+__pycache__/
+*.py[cod]
+
+# Virtualenv
+py3env/
+
+# C extensions
+*.so
+
+# Distribution / packaging
+.Python
+env/
+build/
+develop-eggs/
+dist/
+downloads/
+eggs/
+lib/
+lib64/
+parts/
+sdist/
+var/
+*.egg-info/
+.installed.cfg
+*.egg
+
+# PyCharm
+.idea/
+
+# PyInstaller
+#  Usually these files are written by a python script from a template
+#  before PyInstaller builds the exe, so as to inject date/other infos into it.
+*.manifest
+*.spec
+
+# Installer logs
+pip-log.txt
+pip-delete-this-directory.txt
+
+# Unit test / coverage reports
+htmlcov/
+.tox/
+.coverage
+.cache
+nosetests.xml
+coverage.xml
+
+# Translations
+*.mo
+*.pot
+
+# Django stuff:
+*.log
+
+# Sphinx documentation
+docs/_build/
+
+# PyBuilder
+target/
+
+waagentc
+*.pyproj
+*.sln
+*.suo
+
+waagentc
+bin/waagent2.0c
+
+# rope project
+.ropeproject/

tests_e2e/GuestAgentDcrTestExtension/GuestAgentDcrTest.py

@@ -0,0 +1,123 @@
+#!/usr/bin/env python
+# pylint: disable=all
+from __future__ import print_function
+
+from Utils.WAAgentUtil import waagent
+import Utils.HandlerUtil as Util
+import sys
+import re
+import traceback
+import os
+import datetime
+
+ExtensionShortName = "GADcrTestExt"
+OperationFileName = "operations-{0}.log"
+
+
+def install():
+    operation = "install"
+    status = "success"
+    msg = "Installed successfully"
+
+    hutil = parse_context(operation)
+    hutil.log("Start to install.")
+    hutil.log(msg)
+    hutil.do_exit(0, operation, status, '0', msg)
+
+
+def enable():
+    # Global Variables definition
+    operation = "enable"
+    status = "success"
+    msg = "Enabled successfully."
+
+    # Operations.append(operation)
+    hutil = parse_context(operation)
+    hutil.log("Start to enable.")
+    public_settings = hutil.get_public_settings()
+    name = public_settings.get("name")
+    if name:
+        name = "Name: {0}".format(name)
+        hutil.log(name)
+        msg = "{0} {1}".format(msg, name)
+        print(name)
+    else:
+        hutil.error("The name in public settings is not provided.")
+    # msg = msg % ','.join(Operations)
+    hutil.log(msg)
+    hutil.do_exit(0, operation, status, '0', msg)
+
+
+def disable():
+    operation = "disable"
+    status = "success"
+    msg = "Disabled successfully."
+
+    # Operations.append(operation)
+    hutil = parse_context(operation)
+    hutil.log("Start to disable.")
+    # msg % ','.join(Operations)
+    hutil.log(msg)
+    hutil.do_exit(0, operation, status, '0', msg)
+
+
+def uninstall():
+    operation = "uninstall"
+    status = "success"
+    msg = "Uninstalled successfully."
+
+    # Operations.append(operation)
+    hutil = parse_context(operation)
+    hutil.log("Start to uninstall.")
+    # msg % ','.join(Operations)
+    hutil.log(msg)
+    hutil.do_exit(0, operation, status, '0', msg)
+
+
+def update():
+    operation = "update"
+    status = "success"
+    msg = "Updated successfully."
+
+    # Operations.append(operation)
+    hutil = parse_context(operation)
+    hutil.log("Start to update.")
+    # msg % ','.join(Operations)
+    hutil.log(msg)
+    hutil.do_exit(0, operation, status, '0', msg)
+
+
+def parse_context(operation):
+    hutil = Util.HandlerUtility(waagent.Log, waagent.Error)
+    hutil.do_parse_context(operation)
+    op_log = os.path.join(hutil.get_log_dir(), OperationFileName.format(hutil.get_extension_version()))
+    with open(op_log, 'a+') as oplog_handler:
+        oplog_handler.write("Date:{0}; Operation:{1}; SeqNo:{2}\n"
+                            .format(datetime.datetime.utcnow().strftime("%Y-%m-%dT%H:%M:%SZ"),
+                                    operation, hutil.get_seq_no()))
+    return hutil
+
+
+def main():
+    waagent.LoggerInit('/var/log/waagent.log', '/dev/stdout')
+    waagent.Log("%s started to handle." % (ExtensionShortName))
+
+    try:
+        for a in sys.argv[1:]:
+            if re.match("^([-/]*)(disable)", a):
+                disable()
+            elif re.match("^([-/]*)(uninstall)", a):
+                uninstall()
+            elif re.match("^([-/]*)(install)", a):
+                install()
+            elif re.match("^([-/]*)(enable)", a):
+                enable()
+            elif re.match("^([-/]*)(update)", a):
+                update()
+    except Exception as e:
+        err_msg = "Failed with error: {0}, {1}".format(e, traceback.format_exc())
+        waagent.Error(err_msg)
+
+
+if __name__ == '__main__':
+    main()

tests_e2e/GuestAgentDcrTestExtension/HandlerManifest.json

@@ -0,0 +1,14 @@
+[{
+  "name": "GuestAgentDcrTestExtension",
+  "version": 1.0,
+  "handlerManifest": {
+    "installCommand": "./GuestAgentDcrTest.py --install",
+    "uninstallCommand": "./GuestAgentDcrTest.py --uninstall",
+    "updateCommand": "./GuestAgentDcrTest.py --update",
+    "enableCommand": "./GuestAgentDcrTest.py --enable",
+    "disableCommand": "./GuestAgentDcrTest.py --disable",
+    "updateMode": "UpdateWithoutInstall",
+    "rebootAfterInstall": false,
+    "reportHeartbeat": false
+  }
+}]