115 lines
3.8 KiB
Erlang
115 lines
3.8 KiB
Erlang
%% This Source Code Form is subject to the terms of the Mozilla Public
|
|
%% License, v. 2.0. If a copy of the MPL was not distributed with this
|
|
%% file, You can obtain one at https://mozilla.org/MPL/2.0/.
|
|
%%
|
|
%% Copyright (c) 2007-2025 Broadcom. All Rights Reserved. The term “Broadcom” refers to Broadcom Inc. and/or its subsidiaries. All rights reserved.
|
|
%%
|
|
|
|
-module(python_SUITE).
|
|
-compile(export_all).
|
|
-include_lib("common_test/include/ct.hrl").
|
|
|
|
all() ->
|
|
[
|
|
%% This must use a dedicated node as they mess with plugin configuration in incompatible ways
|
|
{group, tls},
|
|
{group, implicit_connect},
|
|
{group, main}
|
|
].
|
|
|
|
groups() ->
|
|
[
|
|
{main, [], [
|
|
main
|
|
]},
|
|
{implicit_connect, [], [
|
|
implicit_connect
|
|
]},
|
|
{tls, [], [
|
|
tls_connections
|
|
]}
|
|
].
|
|
|
|
init_per_suite(Config) ->
|
|
{ok, _} = rabbit_ct_helpers:exec(["pip", "install", "-r", requirements_path(Config),
|
|
"--target", deps_path(Config)]),
|
|
Config.
|
|
|
|
end_per_suite(Config) ->
|
|
ok = file:del_dir_r(deps_path(Config)),
|
|
Config.
|
|
|
|
init_per_group(_, Config) ->
|
|
Config1 = rabbit_ct_helpers:set_config(Config,
|
|
[
|
|
{rmq_nodename_suffix, ?MODULE},
|
|
{rmq_certspwd, "bunnychow"}
|
|
]),
|
|
rabbit_ct_helpers:log_environment(),
|
|
Config2 = rabbit_ct_helpers:run_setup_steps(
|
|
Config1,
|
|
rabbit_ct_broker_helpers:setup_steps()),
|
|
Config2.
|
|
|
|
end_per_group(_, Config) ->
|
|
rabbit_ct_helpers:run_teardown_steps(Config,
|
|
rabbit_ct_broker_helpers:teardown_steps()).
|
|
|
|
init_per_testcase(Test, Config) ->
|
|
rabbit_ct_helpers:testcase_started(Config, Test).
|
|
|
|
end_per_testcase(Test, Config) ->
|
|
rabbit_ct_helpers:testcase_finished(Config, Test).
|
|
|
|
|
|
main(Config) ->
|
|
rabbit_ct_broker_helpers:rpc(
|
|
Config, 0,
|
|
application, set_env, [rabbitmq_stomp, max_frame_size, 17 * 1024 * 1024]),
|
|
run(Config, filename:join("src", "main_runner.py")).
|
|
|
|
implicit_connect(Config) ->
|
|
run(Config, filename:join("src", "implicit_connect_runner.py")).
|
|
|
|
tls_connections(Config) ->
|
|
run(Config, filename:join("src", "tls_runner.py")).
|
|
|
|
|
|
run(Config, Test) ->
|
|
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),
|
|
AmqpPort = rabbit_ct_broker_helpers:get_node_config(Config, 0, tcp_port_amqp),
|
|
MgmtPort = rabbit_ct_broker_helpers:get_node_config(Config, 0, tcp_port_mgmt),
|
|
NodeName = rabbit_ct_broker_helpers:get_node_config(Config, 0, nodename),
|
|
os:putenv("AMQP_PORT", integer_to_list(AmqpPort)),
|
|
os:putenv("MGMT_PORT", integer_to_list(MgmtPort)),
|
|
os:putenv("STOMP_PORT", integer_to_list(StompPort)),
|
|
os:putenv("STOMP_PORT_TLS", integer_to_list(StompPortTls)),
|
|
os:putenv("RABBITMQ_NODENAME", atom_to_list(NodeName)),
|
|
os:putenv("SSL_CERTS_PATH", CertsDir),
|
|
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),
|
|
filename:dirname(Src).
|