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 114 115
|
/*
* 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 <cstdlib>
#include <fstream>
#include <iostream>
#include <boost/test/unit_test.hpp>
#include "ServerTestHarness.hpp"
#include "TestFixture.hpp"
#include "ecflow/attribute/VerifyAttr.hpp"
#include "ecflow/core/Converter.hpp"
#include "ecflow/core/PrintStyle.hpp"
#include "ecflow/core/Timer.hpp"
#include "ecflow/node/Defs.hpp"
#include "ecflow/node/Family.hpp"
#include "ecflow/node/Suite.hpp"
#include "ecflow/node/Task.hpp"
#include "ecflow/test/scaffold/Naming.hpp"
using namespace std;
using namespace ecf;
BOOST_AUTO_TEST_SUITE(S_Test)
BOOST_AUTO_TEST_SUITE(T_RequeueNode)
BOOST_AUTO_TEST_CASE(test_requeue_node) {
ECF_NAME_THIS_TEST();
DurationTimer timer;
TestClean clean_at_start_and_end;
// Create the defs file corresponding to the text below
// ECF_HOME variable is automatically added by the test harness.
// ECF_INCLUDE variable is automatically added by the test harness.
// SLEEPTIME variable is automatically added by the test harness.
// ECF_CLIENT_EXE_PATH variable is automatically added by the test harness.
// This is substituted in sms includes
// Allows test to run without requiring installation
// # Note: we have to use relative paths, since these tests are relocatable
// suite test_repeat_integer
// repeat integer VAR 0 1 1 # run at 0, 1 2 times
// edit SLEEPTIME 1
// edit ECF_INCLUDE $ECF_HOME/includes
// family family
// repeat integer VAR 0 2 1 # run at 0, 1 2 times
// task t<n>
// ....
// endfamily
// endsuite
// Each task/job should be run *4* times, according to the repeats Mimics nested loops
Defs theDefs;
{
suite_ptr suite = theDefs.add_suite("test_requeue_node");
suite->addRepeat(RepeatInteger("VAR", 0, 1, 1)); // repeat suite 2 times
suite->addVerify(VerifyAttr(NState::COMPLETE, 2));
family_ptr fam = suite->add_family("family");
fam->addRepeat(RepeatInteger("VAR", 0, 1, 1)); // repeat family 2 times
fam->addVerify(VerifyAttr(NState::COMPLETE, 4));
int taskSize = 2; // on linux 1024 tasks take ~4 seconds for job submission
for (int i = 0; i < taskSize; i++) {
task_ptr task = fam->add_task("t" + ecf::convert_to<std::string>(i));
task->addVerify(VerifyAttr(NState::COMPLETE, 4)); // task should complete 4 times
}
}
// The test harness will create corresponding directory structure
// and populate with standard sms files.
ServerTestHarness serverTestHarness;
serverTestHarness.run(theDefs, ServerTestHarness::testDataDefsLocation("test_requeue_node.def"));
// Now re-queue the whole suite
TestFixture::client().set_throw_on_error(true);
TestFixture::client().requeue("/test_requeue_node");
// Wait for test to finish
int timeout = 30;
bool verifyAttrInServer = false;
defs_ptr serverDefs = serverTestHarness.testWaiter(theDefs, timeout, verifyAttrInServer);
BOOST_REQUIRE_MESSAGE(serverDefs.get(), " Failed to return server after re-queue");
// cout << "Printing Defs \n";
// std::cout << *serverDefs.get();
// since we requeue, each task should have completed 8 times
std::vector<Task*> taskVec;
serverDefs->getAllTasks(taskVec);
for (Task* t : taskVec) {
for (const VerifyAttr& v : t->verifys()) {
if (v.state() == NState::COMPLETE) {
BOOST_CHECK_MESSAGE(v.actual() == 8,
"Expected task " << t->absNodePath()
<< " to complete 8 times, due to reque, but it completed only "
<< v.actual() << " times\n");
}
}
}
cout << timer.duration() << " update-calendar-count(" << serverTestHarness.serverUpdateCalendarCount() << ")\n";
}
BOOST_AUTO_TEST_SUITE_END()
BOOST_AUTO_TEST_SUITE_END()
|