iLO-manager first main commitgit status

venv/lib/python3.10/site-packages/tests/pmemtests/functional/ilomock.py

@@ -0,0 +1,97 @@
+"""ilomock module"""
+from __future__ import absolute_import
+
+import os
+import subprocess
+import sys
+
+import requests
+
+JAVA_BIN = os.getenv('JAVA_BIN', 'java')
+ILOMOCK_JAR = os.getenv('ILOMOCK_JAR')
+
+os.environ['NO_PROXY'] = 'localhost'
+
+# ignore SSL warnings
+# requests.packages.urllib3.disable_warnings()
+
+
+class MockServerError(Exception):
+    """An error occurred starting the mock server."""
+
+
+class MockServer(object):
+    """Wrapper for ilomock."""
+
+    def __init__(self, profile_dir=None, port=0):
+        """Initializes a mock server.
+
+        :param profile_dir: path to the server profile to use
+        :param port: port to listen on
+        """
+        self.profile_dir = profile_dir
+        self.port = port
+        self.url = None
+        self.proc = None
+
+    def start(self):
+        """Starts the mock server.
+
+        :raises MockServerError: if the server failed to start
+        """
+        if not ILOMOCK_JAR:
+            raise MockServerError("ILOMOCK_JAR must be set")
+
+        sys.stdout.write("Starting mock server on port {} with profile {}".format(self.port, self.profile_dir))
+
+        args = [JAVA_BIN, '-jar', ILOMOCK_JAR, '-p', str(self.port)]
+        if self.profile_dir:
+            args.extend(['-s', self.profile_dir])
+        if not os.path.exists(JAVA_BIN):
+            raise IOError("JAVA_BIN location does not exist: {}".format(JAVA_BIN))
+        proc = subprocess.Popen(args, stdout=subprocess.PIPE, stderr=subprocess.STDOUT)
+
+        # Block until the server starts or fails
+        output = []
+        for line in iter(proc.stdout.readline, ''):
+            output.append(line)
+            prefix = "Listening on port "
+            if line.startswith(prefix):
+                self.proc = proc
+                self.port = int(line[len(prefix):])
+                self.url = "https://localhost:{}".format(self.port)
+                return
+
+        proc.communicate()
+        output = ''.join(output)
+        raise MockServerError("Failed to start ilomock: {}\n{}".format(proc.returncode, output))
+
+    def stop(self):
+        """Stops the running mock server."""
+        if self.proc:
+            self.proc.kill()
+            self.proc = None
+            self.url = None
+
+    def __enter__(self):
+        """Starts the mock server and returns it."""
+        self.start()
+        return self
+
+    def __exit__(self, exc_type, exc_value, traceback):
+        """Ensures the running mock server stops."""
+        self.stop()
+
+    def patch(self, path, data):
+        """Patches a resource in the mock server."""
+        payload = {'path': path, 'data': data}
+        resp = requests.post(self.url + '/admin/patch', json=payload, verify=False)
+        assert resp.status_code == 200, "Failed to patch mock server: {}\n{}"\
+                                        .format(resp.status_code, resp.text)
+
+    def load(self, profile_dir):
+        """Loads a new server profile while the mock server is running."""
+        config = {'serverProfile': profile_dir}
+        resp = requests.post(self.url + '/admin/load', json=config, verify=False)
+        assert resp.status_code == 200, "Failed to load mock server: {}\n{}"\
+                                        .format(resp.status_code, resp.text)