1 files changed, 84 insertions, 0 deletions
diff --git a/src/VBox/ValidationKit/testdriver/vboxcon.py b/src/VBox/ValidationKit/testdriver/vboxcon.py
new file mode 100755
index 00000000..6c1159f8
--- /dev/null
+++ b/src/VBox/ValidationKit/testdriver/vboxcon.py
@@ -0,0 +1,84 @@
+# -*- coding: utf-8 -*-
+# $Id: vboxcon.py $
+
+"""
+VirtualBox Constants.
+
+See VBoxConstantWrappingHack for details.
+"""
+
+__copyright__ = \
+"""
+Copyright (C) 2010-2019 Oracle Corporation
+
+This file is part of VirtualBox Open Source Edition (OSE), as
+available from http://www.virtualbox.org. This file is free software;
+you can redistribute it and/or modify it under the terms of the GNU
+General Public License (GPL) as published by the Free Software
+Foundation, in version 2 as it comes in the "COPYING" file of the
+VirtualBox OSE distribution. VirtualBox OSE is distributed in the
+hope that it will be useful, but WITHOUT ANY WARRANTY of any kind.
+
+The contents of this file may alternatively be used under the terms
+of the Common Development and Distribution License Version 1.0
+(CDDL) only, as it comes in the "COPYING.CDDL" file of the
+VirtualBox OSE distribution, in which case the provisions of the
+CDDL are applicable instead of those of the GPL.
+
+You may elect to license modified versions of this file under the
+terms and conditions of either the GPL or the CDDL or both.
+"""
+__version__ = "$Revision: 127855 $"
+
+
+# Standard Python imports.
+import sys
+
+
+class VBoxConstantWrappingHack(object):                                         # pylint: disable=R0903
+    """
+    This is a hack to avoid the self.oVBoxMgr.constants.MachineState_Running
+    ugliness that forces one into the right margin...  Anyone using this module
+    can get to the constants easily by:
+
+        from testdriver import vboxcon
+        if self.o.machine.state == vboxcon.MachineState_Running:
+            do stuff;
+
+    For our own convenience there's a vboxcon attribute set up in vbox.py,
+    class TestDriver which is the basis for the VirtualBox testcases. It takes
+    care of setting things up properly through the global variable
+    'goHackModuleClass' that refers to the instance of this class(if we didn't
+    we'd have to use testdriver.vboxcon.MachineState_Running).
+    """
+    def __init__(self, oWrapped):
+        self.oWrapped = oWrapped;
+        self.oVBoxMgr = None;
+        self.fpApiVer = 99.0;
+
+    def __getattr__(self, sName):
+        # Our self.
+        try:
+            return getattr(self.oWrapped, sName)
+        except AttributeError:
+            # The VBox constants.
+            if self.oVBoxMgr is None:
+                raise;
+            try:
+                return getattr(self.oVBoxMgr.constants, sName);
+            except AttributeError:
+                # Do some compatability mappings to keep it working with
+                # older versions.
+                if self.fpApiVer < 3.3:
+                    if sName == 'SessionState_Locked':
+                        return getattr(self.oVBoxMgr.constants, 'SessionState_Open');
+                    if sName == 'SessionState_Unlocked':
+                        return getattr(self.oVBoxMgr.constants, 'SessionState_Closed');
+                    if sName == 'SessionState_Unlocking':
+                        return getattr(self.oVBoxMgr.constants, 'SessionState_Closing');
+                raise;
+
+
+goHackModuleClass = VBoxConstantWrappingHack(sys.modules[__name__]);                         # pylint: disable=C0103
+sys.modules[__name__] = goHackModuleClass;
+