File: TestClientRetry.cpp

package info (click to toggle)
ecflow 5.15.2-2
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 51,868 kB
  • sloc: cpp: 269,341; python: 22,756; sh: 3,609; perl: 770; xml: 333; f90: 204; ansic: 141; makefile: 70
file content (96 lines) | stat: -rw-r--r-- 2,958 bytes parent folder | download
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
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
/*
 * Copyright 2009- ECMWF.
 *
 * This software is licensed under the terms of the Apache Licence version 2.0
 * which can be obtained at http://www.apache.org/licenses/LICENSE-2.0.
 * In applying this licence, ECMWF does not waive the privileges and immunities
 * granted to it by virtue of its status as an intergovernmental organisation
 * nor does it submit to any jurisdiction.
 */

#include <boost/test/unit_test.hpp>

#include "ecflow/base/cts/task/CompleteCmd.hpp"
#include "ecflow/base/cts/user/CtsCmd.hpp"
#include "ecflow/client/ClientInvoker.hpp"
#include "ecflow/core/Environment.hpp"
#include "ecflow/test/scaffold/Naming.hpp"
#include "ecflow/test/scaffold/Provisioning.hpp"

using namespace std;
using namespace ecf;

BOOST_AUTO_TEST_SUITE(S_Client)

BOOST_AUTO_TEST_SUITE(T_ClientRetry)

const ClientToServerCmd& ping = CtsCmd(CtsCmd::Api::PING);
const ClientToServerCmd& task = CompleteCmd("/path/to/task", "password", "rid", 1);
const ClientToServerCmd& user = CtsCmd(CtsCmd::Api::RESTART_SERVER);

BOOST_AUTO_TEST_CASE(test_client_does_not_retry_ping_command) {
    ECF_NAME_THIS_TEST();

    ClientInvoker client;
    BOOST_CHECK(client.is_not_retrying(ping));
}

BOOST_AUTO_TEST_CASE(test_client_does_not_retry_user_command_by_default) {
    ECF_NAME_THIS_TEST();

    ClientInvoker client;
    BOOST_CHECK(client.is_not_retrying(user));
}

BOOST_AUTO_TEST_CASE(test_client_does_not_retry_when_denied_is_set) {
    ECF_NAME_THIS_TEST();

    WithTestEnvironmentVariable env_var(ecf::environment::ECF_DENIED, "true");
    ClientInvoker client;
    BOOST_CHECK(client.is_not_retrying(ping));
    BOOST_CHECK(client.is_not_retrying(task));
    BOOST_CHECK(client.is_not_retrying(user));
}

BOOST_AUTO_TEST_CASE(test_client_does_not_retry_when_testing_is_set) {
    ECF_NAME_THIS_TEST();

    ClientInvoker client;
    client.taskPath("/path/to/task"); // This effectively sets the testing mode
    BOOST_CHECK(client.is_not_retrying(ping));
    BOOST_CHECK(client.is_not_retrying(task));
    BOOST_CHECK(client.is_not_retrying(user));
}

BOOST_AUTO_TEST_CASE(test_client_retries_task_command_by_default) {
    ECF_NAME_THIS_TEST();

    ClientInvoker client;
    BOOST_CHECK(!client.is_not_retrying(task));
}

BOOST_AUTO_TEST_CASE(test_client_follows_host_file_policy) {
    ECF_NAME_THIS_TEST();

    {
        WithTestEnvironmentVariable env_var(ecf::environment::ECF_HOSTFILE_POLICY, "task");

        ClientInvoker client;
        BOOST_CHECK(client.is_not_retrying(ping));
        BOOST_CHECK(client.is_not_retrying(user));
        BOOST_CHECK(!client.is_not_retrying(task));
    }

    {
        WithTestEnvironmentVariable env_var(ecf::environment::ECF_HOSTFILE_POLICY, "all");

        ClientInvoker client;
        BOOST_CHECK(!client.is_not_retrying(ping));
        BOOST_CHECK(!client.is_not_retrying(user));
        BOOST_CHECK(!client.is_not_retrying(task));
    }
}

BOOST_AUTO_TEST_SUITE_END()

BOOST_AUTO_TEST_SUITE_END()