1 files changed, 72 insertions, 0 deletions
diff --git a/deps/rabbitmq_stomp/test/python_SUITE.erl b/deps/rabbitmq_stomp/test/python_SUITE.erl
new file mode 100644
index 0000000000..9613b25032
--- /dev/null
+++ b/deps/rabbitmq_stomp/test/python_SUITE.erl
@@ -0,0 +1,72 @@
+%% 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-2020 VMware, Inc. or its affiliates.  All rights reserved.
+%%
+
+-module(python_SUITE).
+-compile(export_all).
+-include_lib("common_test/include/ct.hrl").
+
+all() ->
+    [
+    common,
+    ssl,
+    connect_options
+    ].
+
+init_per_testcase(TestCase, Config) ->
+    Suffix = rabbit_ct_helpers:testcase_absname(Config, TestCase, "-"),
+    Config1 = rabbit_ct_helpers:set_config(Config,
+                                           [{rmq_certspwd, "bunnychow"},
+                                            {rmq_nodename_suffix, Suffix}]),
+    rabbit_ct_helpers:log_environment(),
+    Config2 = rabbit_ct_helpers:run_setup_steps(
+        Config1,
+        rabbit_ct_broker_helpers:setup_steps()),
+    DataDir = ?config(data_dir, Config2),
+    PikaDir = filename:join([DataDir, "deps", "pika"]),
+    StomppyDir = filename:join([DataDir, "deps", "stomppy"]),
+    rabbit_ct_helpers:make(Config2, PikaDir, []),
+    rabbit_ct_helpers:make(Config2, StomppyDir, []),
+    Config2.
+
+end_per_testcase(_, Config) ->
+    rabbit_ct_helpers:run_teardown_steps(Config,
+        rabbit_ct_broker_helpers:teardown_steps()).
+
+
+common(Config) ->
+    run(Config, filename:join("src", "test.py")).
+
+connect_options(Config) ->
+    run(Config, filename:join("src", "test_connect_options.py")).
+
+ssl(Config) ->
+    run(Config, filename:join("src", "test_ssl.py")).
+
+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),
+    AmqpPort = rabbit_ct_broker_helpers:get_node_config(Config, 0, tcp_port_amqp),
+    NodeName = rabbit_ct_broker_helpers:get_node_config(Config, 0, nodename),
+    PythonPath = os:getenv("PYTHONPATH"),
+    os:putenv("PYTHONPATH", filename:join([DataDir, "deps", "pika","pika"])
+                            ++":"++
+                            filename:join([DataDir, "deps", "stomppy", "stomppy"])
+                            ++ ":" ++
+                            PythonPath),
+    os:putenv("AMQP_PORT", integer_to_list(AmqpPort)),
+    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),
+    {ok, _} = rabbit_ct_helpers:exec([filename:join(DataDir, Test)]).
+
+
+cur_dir() ->
+    {ok, Src} = filelib:find_source(?MODULE),
+    filename:dirname(Src).