Introduce test_ust_python_agent_vs_tools.py

[deliverable/lttng-ivc.git] / lttng_ivc / utils / project.py

diff --git a/lttng_ivc/utils/project.py b/lttng_ivc/utils/project.py
index 71d17e2a14d79c48c3b790be177b5305a8257d1c..bd389a59b4fb6b7257f63757b6074390fc098dad 100644 (file)
--- a/lttng_ivc/utils/project.py
+++ b/lttng_ivc/utils/project.py
@@ -6,6 +6,7 @@ import logging
 import lttng_ivc.settings as Settings
 
 from lttng_ivc.utils.utils import sha256_checksum
+from lttng_ivc.utils.utils import find_dir
 
 _logger = logging.getLogger('project')
 
@@ -321,6 +322,15 @@ class Lttng_ust(Project):
         classpath = ":".join([jul_path, Settings.log4j_class_path, '.'])
         self.add_special_env_variable("CLASSPATH", classpath)
 
+    def install(self):
+        super(Lttng_ust, self).install()
+        python_path = find_dir(self.installation_path, "lttngust")
+        if python_path:
+            # Fetch the parent of lttngust folder
+            python_path = os.path.dirname(python_path)
+            self.add_special_env_variable("PYTHONPATH", python_path)
+
+
 
 class Lttng_tools(Project):
     def __init__(self, label, git_path, sha1, tmpdir):