summaryrefslogtreecommitdiff
path: root/deps/rabbitmq_stomp/test/python_SUITE.erl
blob: 9613b25032c39887b820d7f3cd40b902484048c6 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
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).