File: TestAvisoAttr.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 (113 lines) | stat: -rw-r--r-- 3,696 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
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
/*
 * 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 <iostream>
#include <string>

#include <boost/test/unit_test.hpp>

#include "ecflow/node/AvisoAttr.hpp"
#include "ecflow/node/Defs.hpp"
#include "ecflow/node/Family.hpp"
#include "ecflow/node/Task.hpp"
#include "ecflow/node/parser/DefsStructureParser.hpp"
#include "ecflow/test/scaffold/Naming.hpp"
#include "ecflow/test/scaffold/Provisioning.hpp"

BOOST_AUTO_TEST_SUITE(U_Node)

BOOST_AUTO_TEST_SUITE(T_AvisoAttr)

BOOST_AUTO_TEST_CASE(can_run_aviso_attribute_with_variable_substitution) {
    ECF_NAME_THIS_TEST();

    using namespace ecf;

    WithTestFile schema(NamedTestFile{"schema.aviso.json"}, R"(
      {
        "version":0.1,
        "payload":"location",
        "mars":{
          "endpoint":[
            {
              "engine":[
                "etcd_rest",
                "etcd_grpc"
              ],
              "base":"/ec/mars",
              "stem":"date={date},class={class},expver={expver},domain={domain},time={time},stream={stream},step={step}"
            },
            {
              "engine":[
                "file_based"
              ],
              "base":"/tmp/aviso/mars",
              "stem":"{class}/{expver}/{domain}/{date}/{time}/{stream}/{step}"
            }
          ]
        }
      })");

    std::string definition = R"(
        suite s1
          family f1
            edit CLASS 'od'
            edit ECF_AVISO_URL 'https://example.com/aviso'
            edit ECF_AVISO_SCHEMA 'schema.aviso.json'
            edit ECF_AVISO_AUTH ''
            edit ECF_AVISO_POLLING '30'
            task t1
              aviso --name A --listener '{ "event": "mars", "request": { "class": "%CLASS%" } }'
          endfamily
    )";

    Defs defs;
    DefsStructureParser parser(&defs, definition, true);

    std::string errorMsg, warningMsg;
    bool parsedOK = parser.doParse(errorMsg, warningMsg);
    BOOST_CHECK_MESSAGE(parsedOK, "Failed to parse definition: " << errorMsg);

    const auto& suites = defs.suiteVec();
    BOOST_CHECK_EQUAL(suites.size(), static_cast<size_t>(1));

    const auto& families = suites[0]->familyVec();
    BOOST_CHECK_EQUAL(families.size(), static_cast<size_t>(1));

    const auto& tasks = families[0]->taskVec();
    BOOST_CHECK_EQUAL(tasks.size(), static_cast<size_t>(1));

    const auto& avisos = tasks[0]->avisos();
    BOOST_CHECK_EQUAL(avisos.size(), static_cast<size_t>(1));

    const auto& aviso = avisos[0];
    BOOST_CHECK_EQUAL(aviso.name(), "A");
    BOOST_CHECK_EQUAL(aviso.listener(), R"('{ "event": "mars", "request": { "class": "%CLASS%" } }')");
    BOOST_CHECK_EQUAL(aviso.url(), "%ECF_AVISO_URL%");
    BOOST_CHECK_EQUAL(aviso.schema(), "%ECF_AVISO_SCHEMA%");
    BOOST_CHECK_EQUAL(aviso.auth(), "%ECF_AVISO_AUTH%");
    BOOST_CHECK_EQUAL(aviso.polling(), "%ECF_AVISO_POLLING%");
    BOOST_CHECK_EQUAL(aviso.active(), "");
    BOOST_CHECK_EQUAL(aviso.reason(), "''");

    aviso.start();
    // Ensure that the variable substitution is done correctly
    BOOST_CHECK_EQUAL(aviso.active(), R"({ "event": "mars", "request": { "class": "od" } })");

    // This is the first multithreaded test in ecFlow!

    aviso.finish();
    // Ensure that, after finishing, the active listener is cleared
    BOOST_CHECK_EQUAL(aviso.active(), R"()");
}

BOOST_AUTO_TEST_SUITE_END()

BOOST_AUTO_TEST_SUITE_END()