fix python pep dependency thing

Author: deadtrickster

.gitignore

@@ -86,6 +86,7 @@ rabbitmq-server-*.tar.xz
 rabbitmq-server-*.zip
 
 traces*
+deps/rabbitmq_stomp/test/python_SUITE_data/src/deps
 callgrand*
 
 /user.bazelrc

deps/rabbitmq_stomp/test/python_SUITE.erl

@@ -30,6 +30,17 @@ groups() ->
         ]}
     ].
 
+init_per_suite(Config) ->
+    DataDir = ?config(data_dir, Config),
+    {ok, _} = rabbit_ct_helpers:exec(["pip", "install", "-r", requirements_path(Config),
+                                                        "--target", deps_path(Config)]),
+    Config.
+
+end_per_suite(Config) ->
+    DataDir = ?config(data_dir, Config),
+    ok = file:del_dir_r(deps_path(Config)),
+    Config.
+
 init_per_group(_, Config) ->
     Config1 = rabbit_ct_helpers:set_config(Config,
                                            [
@@ -64,7 +75,6 @@ tls_connections(Config) ->
 
 
 run(Config, Test) ->
-    DataDir = ?config(data_dir, Config),
     CertsDir = rabbit_ct_helpers:get_config(Config, rmq_certsdir),
     StompPort = rabbit_ct_broker_helpers:get_node_config(Config, 0, tcp_port_stomp),
     StompPortTls = rabbit_ct_broker_helpers:get_node_config(Config, 0, tcp_port_stomp_tls),
@@ -75,8 +85,26 @@ run(Config, Test) ->
     os:putenv("STOMP_PORT_TLS", integer_to_list(StompPortTls)),
     os:putenv("RABBITMQ_NODENAME", atom_to_list(NodeName)),
     os:putenv("SSL_CERTS_PATH", CertsDir),
-    {ok, _} = rabbit_ct_helpers:exec([filename:join(DataDir, Test)]).
+    run_python(Config, Test).
+
+run_python(Config, What) ->
+    DataDir = ?config(data_dir, Config),
+    os:putenv("PYTHONPATH", python_path(Config)),
+    {ok, _} = rabbit_ct_helpers:exec([filename:join(DataDir, What)]).
+
+deps_path(Config) ->
+    DataDir = ?config(data_dir, Config),
+    filename:join([DataDir, "src", "deps"]).
+
+requirements_path(Config) ->
+    DataDir = ?config(data_dir, Config),
+    filename:join([DataDir, "src", "requirements.txt"]).
 
+python_path(Config) ->
+    case os:getenv("PYTHONPATH") of
+        false -> deps_path(Config);
+        P -> deps_path(Config) ++ ":" ++ P
+    end.
 
 cur_dir() ->
     {ok, Src} = filelib:find_source(?MODULE),

deps/rabbitmq_stomp/test/python_SUITE_data/src/main_runner.py

@@ -1,14 +1,5 @@
 #!/usr/bin/env python3
 
-import sys
-import subprocess
-
-# implement pip as a subprocess:
-subprocess.check_call([sys.executable, '-m', 'pip', 'install',
-                       'stomp.py==8.1.0'])
-subprocess.check_call([sys.executable, '-m', 'pip', 'install',
-                       'pika==1.1.0'])
-
 import test_runner
 
 if __name__ == '__main__':

deps/rabbitmq_stomp/test/python_SUITE_data/src/test_runner.py

@@ -7,15 +7,6 @@
 ## Copyright (c) 2007-2023 VMware, Inc. or its affiliates.  All rights reserved.
 ##
 
-import sys
-import subprocess
-
-# implement pip as a subprocess:
-subprocess.check_call([sys.executable, '-m', 'pip', 'install',
-                       'stomp.py==8.1.0'])
-subprocess.check_call([sys.executable, '-m', 'pip', 'install',
-                       'pika==1.1.0'])
-
 import unittest
 import sys
 import os

deps/rabbitmq_stomp/test/python_SUITE_data/src/tls_runner.py

@@ -7,15 +7,6 @@
 ## Copyright (c) 2007-2023 VMware, Inc. or its affiliates.  All rights reserved.
 ##
 
-import sys
-import subprocess
-
-# implement pip as a subprocess:
-subprocess.check_call([sys.executable, '-m', 'pip', 'install',
-                       'stomp.py==8.1.0'])
-subprocess.check_call([sys.executable, '-m', 'pip', 'install',
-                       'pika==1.1.0'])
-
 import test_runner
 import test_util