1 files changed, 78 insertions, 0 deletions
diff --git a/sources/pyside6/tests/init_paths.py b/sources/pyside6/tests/init_paths.py
new file mode 100644
index 000000000..e69ec1c15
--- /dev/null
+++ b/sources/pyside6/tests/init_paths.py
@@ -0,0 +1,78 @@
+# Copyright (C) 2022 The Qt Company Ltd.
+# SPDX-License-Identifier: LicenseRef-Qt-Commercial OR GPL-3.0-only WITH Qt-GPL-exception-1.0
+
+import os
+import sys
+
+SRC_DIR = os.path.dirname(os.path.abspath(__file__))
+sys.path.append(os.path.join(os.path.dirname(os.path.dirname(SRC_DIR)),
+                             'shiboken6', 'tests'))
+from shiboken_paths import (get_dir_env_var, get_build_dir, add_python_dirs,
+                            add_lib_dirs, shiboken_paths)
+
+
+def _get_qt_dir():
+    """Retrieve the location of Qt."""
+    result = None
+    try:
+        result = get_dir_env_var('QT_DIR')
+    except Exception:
+        # This is just a convenience for command line invocation independent
+        # of ctest. Normally, QT_DIR should be set to avoid repetitive patch
+        # searches.
+        print('The environment variable QT_DIR is not set; falling back to path search',
+              file=sys.stderr)
+        qmake = 'qmake.exe' if sys.platform == 'win32' else 'qmake'
+        for path in os.environ.get('PATH').split(os.pathsep):
+            if path and os.path.isfile(os.path.join(path, qmake)):
+                result = os.path.dirname(path)
+                break
+        if not result:
+            raise ValueError('Unable to locate Qt. Please set the environment variable QT_DIR')
+        print(f"Qt as determined by path search: {result}", file=sys.stderr)
+    return result
+
+
+def _get_qt_lib_dir():
+    """Retrieve the location of the Qt library directory."""
+    sub_dir = 'bin' if sys.platform == 'win32' else 'lib'
+    return os.path.join(_get_qt_dir(), sub_dir)  # For testbinding
+
+
+def _init_test_paths(shiboken_tests=False, testbindings_module=False):
+    """Sets the correct import paths (Python modules and C++ library paths)
+       for PySide tests and shiboken6 tests using depending on the environment
+       variables BUILD_DIR and QT_DIR pointing to the build directory and
+       Qt directory, respectively."""
+    src_dir = os.path.dirname(os.path.abspath(__file__))
+
+    python_dirs = [os.path.join(src_dir, 'util')]  # Helper module
+
+    pyside_build_dir = os.path.join(get_build_dir(), 'pyside6')
+    python_dirs.append(pyside_build_dir)   # for PySide6
+    lib_dirs = [os.path.join(pyside_build_dir, 'libpyside'),
+                os.path.join(pyside_build_dir, 'libpysideqml')]
+
+    if testbindings_module:
+        python_dirs.append(os.path.join(pyside_build_dir,
+                                        'tests', 'pysidetest'))
+        lib_dirs.append(_get_qt_lib_dir())
+
+    shiboken_path_tuple = shiboken_paths(shiboken_tests)
+    python_dirs.extend(shiboken_path_tuple[0])
+    lib_dirs.extend(shiboken_path_tuple[1])
+
+    add_python_dirs(python_dirs)
+    add_lib_dirs(lib_dirs)
+
+
+def init_test_paths(testbindings_module=False):
+    """Sets the correct import paths for PySide6 tests, optionally including
+       testbindings."""
+    _init_test_paths(False, testbindings_module)
+
+
+def init_all_test_paths():
+    """Sets the correct import paths for PySide6 and shiboken6 tests
+       (for registry checking only)."""
+    _init_test_paths(True, True)