1 // ANNA - Anna is Not Nothingness Anymore //
3 // (c) Copyright 2005-2015 Eduardo Ramos Testillano & Francisco Ruiz Rayo //
5 // See project site at http://redmine.teslayout.com/projects/anna-suite //
6 // See accompanying file LICENSE or copy at http://www.teslayout.com/projects/public/anna.LICENSE //
13 #include <anna/xml/Compiler.hpp>
14 #include <anna/xml/Node.hpp>
15 #include <anna/core/tracing/Logger.hpp>
16 #include <anna/app/functions.hpp>
17 #include <anna/timex/Engine.hpp>
18 #include <anna/diameter/helpers/base/functions.hpp>
19 #include <anna/diameter.comm/ClientSession.hpp>
20 #include <anna/diameter.comm/ServerSession.hpp>
23 #include <TestManager.hpp>
24 #include <TestClock.hpp>
25 #include <Launcher.hpp>
26 #include <RealmNode.hpp>
34 ///////////////////////////////////////////////////////////////////////////////////////////////////
35 void TestManager::StatSummary::newTCState(const TestCase::State::_v beginState, const TestCase::State::_v endState) throw() {
37 if ((beginState == TestCase::State::Initialized)&&(endState == TestCase::State::Initialized)) { // special case (new test case provisioning)
43 case TestCase::State::Initialized: a_initializedTcs--; break;
44 case TestCase::State::InProgress: a_inprogressTcs--; break;
45 case TestCase::State::Failed: a_failedTcs--; break;
46 case TestCase::State::Success: a_sucessTcs--; break;
50 case TestCase::State::Initialized: a_initializedTcs++; break;
51 case TestCase::State::InProgress: a_inprogressTcs++; break;
52 case TestCase::State::Failed: a_failedTcs++; break;
53 case TestCase::State::Success: a_sucessTcs++; break;
58 void TestManager::StatSummary::clear() throw() {
65 anna::xml::Node *TestManager::StatSummary::asXML(anna::xml::Node* parent) const throw() {
66 anna::xml::Node* result = parent->createChild("StatSummary");
68 anna::xml::Node* tcs = result->createChild("TestCasesCounts");
69 tcs->createAttribute("Total", a_initializedTcs + a_inprogressTcs + a_failedTcs + a_sucessTcs);
70 tcs->createAttribute("Initialized", a_initializedTcs);
71 tcs->createAttribute("InProgress", a_inprogressTcs);
72 tcs->createAttribute("Failed", a_failedTcs);
73 tcs->createAttribute("Success", a_sucessTcs);
77 ///////////////////////////////////////////////////////////////////////////////////////////////////
81 TestManager::TestManager() :
82 anna::timex::TimeEventObserver("TestManager") {
83 a_timeController = NULL;
84 a_reportsDirectory = "./";
85 a_dumpSuccessfulReports = false;
86 a_dumpFailedReports = false;
87 a_dumpHexMessages = false;
88 a_synchronousAmount = 1;
89 a_poolRepeats = 0; // repeat disabled by default
91 a_inProgressLimit = UINT_MAX; // no limit
94 a_currentTestIt = a_testPool.end();
97 void TestManager::registerSessionId(const std::string &sessionId, const TestCase *testCase) throw(anna::RuntimeException) {
99 std::map<std::string /* session id's */, TestCase*>::const_iterator it = a_sessionIdTestCaseMap.find(sessionId);
100 if (it != a_sessionIdTestCaseMap.end()) { // found
101 unsigned int id = it->second->getId();
102 if (id != testCase->getId()) {
103 throw anna::RuntimeException(anna::functions::asString("There is another test case (id = %llu) which registered such sessionId: %s", id, sessionId.c_str()), ANNA_FILE_LOCATION);
107 a_sessionIdTestCaseMap[sessionId] = const_cast<TestCase*>(testCase);
111 TestTimer* TestManager::createTimer(TestCaseStep* testCaseStep, const anna::Millisecond &timeout, const TestTimer::Type::_v type)
112 throw(anna::RuntimeException) {
113 TestTimer* result(NULL);
115 if(a_timeController == NULL)
116 throw anna::RuntimeException("You must invoke 'setTimerController' with a not NULL timex engine", ANNA_FILE_LOCATION);
118 anna::Guard guard(a_timeController, "TestManager::createTimer"); // avoid interblocking
119 result = a_timers.create();
120 result->setType(type);
121 result->setId((anna::timex::TimeEvent::Id) testCaseStep);
122 result->setObserver(this);
123 result->setContext(testCaseStep);
124 result->setTimeout(timeout);
127 std::string msg("TestManager::createTimer | ");
128 msg += result->asString();
129 anna::Logger::debug(msg, ANNA_FILE_LOCATION);
132 a_timeController->activate(result);
136 void TestManager::cancelTimer(TestTimer* timer)
142 std::string msg("TestManager::cancel | ");
143 msg += timer->asString();
144 anna::Logger::debug(msg, ANNA_FILE_LOCATION);
148 if(a_timeController == NULL)
149 a_timeController = anna::app::functions::component <anna::timex::Engine> (ANNA_FILE_LOCATION);
151 a_timeController->cancel(timer);
152 } catch(anna::RuntimeException& ex) {
157 //------------------------------------------------------------------------------------------
158 // Se invoca automaticamente desde anna::timex::Engine
159 //------------------------------------------------------------------------------------------
160 void TestManager::release(anna::timex::TimeEvent* timeEvent)
162 TestTimer* timer = static_cast <TestTimer*>(timeEvent);
163 timer->setContext(NULL);
164 a_timers.release(timer);
167 bool TestManager::configureTTPS(int testTicksPerSecond) throw() {
169 if (testTicksPerSecond == 0) {
171 a_timeController->cancel(a_clock);
172 LOGDEBUG(anna::Logger::debug("Testing timer clock stopped !", ANNA_FILE_LOCATION));
175 LOGDEBUG(anna::Logger::debug("No testing timer started yet !", ANNA_FILE_LOCATION));
179 else if (testTicksPerSecond < 0) {
180 LOGWARNING(anna::Logger::warning("Invalid 'ttps' provided", ANNA_FILE_LOCATION));
184 anna::Millisecond admlTimeInterval = anna::Millisecond(1000 / testTicksPerSecond);
185 a_synchronousAmount = 1;
187 if (admlTimeInterval < anna::Millisecond(1)) {
188 LOGWARNING(anna::Logger::warning("Not allowed to configure more than 1000 events per second for for triggering testing system", ANNA_FILE_LOCATION));
192 Launcher& my_app = static_cast <Launcher&>(anna::app::functions::getApp());
193 const anna::Millisecond &admlMinResolution = my_app.getADMLMinResolution();
195 if (admlTimeInterval < admlMinResolution) {
196 int maximumObtained = 1000 / (int)admlMinResolution;
197 a_synchronousAmount = ceil((double)testTicksPerSecond/maximumObtained);
199 admlTimeInterval = anna::Millisecond(a_synchronousAmount * 1000 / testTicksPerSecond);
202 if (a_synchronousAmount > 1) {
204 std::string msg = anna::functions::asString("Desired testing time trigger rate (%d events per second) requires more than one sending per event (%d every %lld milliseconds). Consider launch more instances with lower rate (for example %d ADML processes with %d ttps), or configure %d or more sockets to the remote endpoints to avoid burst sendings",
207 admlTimeInterval.getValue(),
209 1000/admlTimeInterval,
210 a_synchronousAmount);
212 anna::Logger::warning(msg, ANNA_FILE_LOCATION);
217 a_clock->setTimeout(admlTimeInterval);
220 a_clock = new TestClock("Testing clock", admlTimeInterval, this); // clock
223 if (!a_clock->isActive()) a_timeController->activate(a_clock);
228 bool TestManager::gotoTestCase(unsigned int id) throw() {
229 test_pool_it it = a_testPool.find(id);
230 if (it != a_testPool.end()) {
231 a_currentTestIt = it;
238 TestCase *TestManager::findTestCase(unsigned int id) const throw() { // id = -1 provides current test case triggered
240 if (!tests()) return NULL;
241 test_pool_it it = ((id != -1) ? a_testPool.find(id) : a_currentTestIt);
242 if (it != a_testPool.end()) return const_cast<TestCase*>(it->second);
246 TestCase *TestManager::getTestCase(unsigned int id) throw() {
248 test_pool_nc_it it = a_testPool.find(id);
249 if (it != a_testPool.end()) return it->second;
251 TestCase *result = new TestCase(id);
252 a_testPool[id] = result;
256 bool TestManager::clearPool() throw() {
257 if (!tests()) return false;
258 for (test_pool_it it = a_testPool.begin(); it != a_testPool.end(); it++) delete it->second;
259 // TODO: stop the possible command threads or there will be a core dump
262 a_sessionIdTestCaseMap.clear();
263 a_currentTestIt = a_testPool.end();
265 configureTTPS(0); // stop
266 a_statSummary.clear();
270 bool TestManager::resetPool(bool hard) throw() {
271 bool result = false; // any reset
273 if (!tests()) return result;
274 for (test_pool_nc_it it = a_testPool.begin(); it != a_testPool.end(); it++) {
275 if (it->second->reset(hard))
278 //a_sessionIdTestCaseMap.clear();
282 bool TestManager::tick() throw() {
283 LOGDEBUG(anna::Logger::debug("New test clock tick !", ANNA_FILE_LOCATION));
284 return execTestCases(a_synchronousAmount);
287 bool TestManager::execTestCases(int sync_amount) throw() {
290 LOGWARNING(anna::Logger::warning("Testing pool is empty. You need programming", ANNA_FILE_LOCATION));
294 // Synchronous sendings per tick:
295 int count = sync_amount;
297 if (!nextTestCase()) return false; // stop the clock
304 bool TestManager::nextTestCase() throw() {
308 // Limit for in-progress test cases:
309 if (getInProgressCount() >= a_inProgressLimit) {
310 LOGDEBUG(anna::Logger::debug(anna::functions::asString("TestManager next case ignored (over in-progress count limit: %llu)", a_inProgressLimit), ANNA_FILE_LOCATION));
311 return true; // wait next tick to release OTA test cases
315 if (a_currentTestIt == a_testPool.end())
316 a_currentTestIt = a_testPool.begin();
321 if (a_currentTestIt == a_testPool.end()) {
322 if ((a_poolCycle > a_poolRepeats) && (a_poolRepeats != -1)) {
323 LOGWARNING(anna::Logger::warning("Testing pool cycle completed. No remaining repeat cycles left. Suspending", ANNA_FILE_LOCATION));
329 std::string nolimit = (a_poolRepeats != -1) ? "":" [no limit]";
330 anna::Logger::warning(anna::functions::asString("Testing pool cycle %d completed (repeats configured: %d%s). Restarting for the %s cycle", a_poolCycle, a_poolRepeats, nolimit.c_str(), (a_poolRepeats == a_poolCycle) ? "last":"next"), ANNA_FILE_LOCATION);
333 //a_currentTestIt = a_testPool.begin();
334 return true; // avoids infinite loop: if the cycle takes less time than test cases completion, below reset never will turns state
335 // into Initialized and this while will be infinite. It is preferable to wait one tick when the cycle is completed.
339 // Soft reset to initialize already finished (in previous cycle) test cases:
340 a_currentTestIt->second->reset(false);
342 // Process test case:
343 LOGDEBUG(anna::Logger::debug(anna::functions::asString("Processing test case id = %llu, currently '%s' state", a_currentTestIt->first, TestCase::asText(a_currentTestIt->second->getState())), ANNA_FILE_LOCATION));
344 if (a_currentTestIt->second->getState() != TestCase::State::InProgress) {
345 a_currentTestIt->second->process();
346 return true; // is not probably to reach still In-Progress test cases from previous cycles due to the whole
347 // time for complete the test cases pool regarding the single test case lifetime. You shouldn't
348 // forget to programm a test case timeout with a reasonable value
353 TestCase *TestManager::getTestCaseFromSessionId(const anna::DataBlock &message, std::string &sessionId) throw() {
355 sessionId = anna::diameter::helpers::base::functions::getSessionId(message);
357 catch (anna::RuntimeException &ex) {
359 LOGWARNING(anna::Logger::warning("Cannot get the Session-Id from received DataBlock in order to identify the Test Case", ANNA_FILE_LOCATION));
362 std::map<std::string /* session id's */, TestCase*>::const_iterator sessionIdIt = a_sessionIdTestCaseMap.find(sessionId);
363 if (sessionIdIt != a_sessionIdTestCaseMap.end())
364 return sessionIdIt->second;
366 LOGWARNING(anna::Logger::warning(anna::functions::asString("Cannot identify the Test Case for received Session-Id: %s", sessionId.c_str()), ANNA_FILE_LOCATION));
370 void TestManager::receiveMessage(const anna::DataBlock &message, RealmNode *realm, const anna::diameter::comm::ClientSession *clientSession) throw(anna::RuntimeException) {
373 if (!tests()) return;
375 // Identify the test case:
376 std::string sessionId;
377 TestCase *tc = getTestCaseFromSessionId(message, sessionId);
380 // Work with Test case:
381 TestStepWait *tsw = tc->searchNextWaitConditionFulfilled(message, true /* comes from entity */);
382 if (!tsw) { // store as 'uncovered'
383 std::string hint = "Uncovered condition for received message from entity over Session-Id '"; hint += sessionId; hint += "':";
386 static anna::diameter::codec::Message codecMsg;
387 codecMsg.decode(message);
388 hint += "\n"; hint += codecMsg.asXMLString();
390 // // Realm checking:
391 // std::string messageOR = message.getAvp(anna::diameter::helpers::base::AVPID__Origin_Realm)->getDiameterIdentity()->getValue();
392 // if (messageOR != realm->getName()) {
393 // LOGWARNING(anna::Logger::warning(anna::functions::asString("Received message from realm '%s', with different Origin-Realm: %s", realm->getName().c_str(), messageOR.c_str()), ANNA_FILE_LOCATION));
396 catch (anna::RuntimeException &ex) {
398 hint += "\n"; hint += ex.asString();
400 hint += "\n"; hint += clientSession->asString();
402 tc->addDebugSummaryHint(hint);
405 tsw->setClientSession(const_cast<anna::diameter::comm::ClientSession*>(clientSession));
410 void TestManager::receiveMessage(const anna::DataBlock &message, RealmNode *realm, const anna::diameter::comm::ServerSession *serverSession) throw(anna::RuntimeException) {
413 if (!tests()) return;
415 // Identify the test case:
416 std::string sessionId;
417 TestCase *tc = getTestCaseFromSessionId(message, sessionId);
420 // Work with Test case:
421 TestStepWait *tsw = tc->searchNextWaitConditionFulfilled(message, false /* comes from client */);
422 if (!tsw) { // store as 'uncovered'
423 std::string hint = "Uncovered condition for received message from client over Session-Id '"; hint += sessionId; hint += "':";
426 static anna::diameter::codec::Message codecMsg;
427 codecMsg.decode(message);
428 hint += "\n"; hint += codecMsg.asXMLString();
430 // // Realm checking:
431 // std::string messageOR = message.getAvp(anna::diameter::helpers::base::AVPID__Origin_Realm)->getDiameterIdentity()->getValue();
432 // if (messageOR != realm->getName()) {
433 // LOGWARNING(anna::Logger::warning(anna::functions::asString("Received message from realm '%s', with different Origin-Realm: %s", realm->getName().c_str(), messageOR.c_str()), ANNA_FILE_LOCATION));
436 catch (anna::RuntimeException &ex) {
438 hint += "\n"; hint += ex.asString();
440 hint += "\n"; hint += serverSession->asString();
442 tc->addDebugSummaryHint(hint);
445 tsw->setServerSession(const_cast<anna::diameter::comm::ServerSession*>(serverSession));
450 anna::xml::Node* TestManager::asXML(anna::xml::Node* parent) const
452 anna::xml::Node* result = parent->createChild("TestManager");
454 int poolSize = a_testPool.size();
455 result->createAttribute("NumberOfTestCases", poolSize);
456 if (a_poolRepeats) result->createAttribute("PoolRepeats", a_poolRepeats);
457 else result->createAttribute("PoolRepeats", "disabled");
458 result->createAttribute("PoolCycle", a_poolCycle);
459 a_statSummary.asXML(result);
460 if (a_inProgressLimit == UINT_MAX)
461 result->createAttribute("InProgressLimit", "<no limit>");
463 result->createAttribute("InProgressLimit", a_inProgressLimit);
464 result->createAttribute("DumpSuccessfulReports", (a_dumpSuccessfulReports ? "yes":"no"));
465 result->createAttribute("DumpFailedReports", (a_dumpFailedReports ? "yes":"no"));
466 result->createAttribute("DumpHexMessages", (a_dumpHexMessages ? "yes":"no"));
467 result->createAttribute("ReportsDirectory", a_reportsDirectory);
469 result->createAttribute("AsynchronousSendings", a_synchronousAmount);
470 int ticksPerSecond = (a_synchronousAmount * 1000) / a_clock->getTimeout();
471 result->createAttribute("TicksPerSecond", ticksPerSecond);
473 if (a_currentTestIt != a_testPool.end()) {
474 result->createAttribute("CurrentTestCaseId", (*a_currentTestIt).first);
477 anna::xml::Node* testCases = result->createChild("TestCases");
478 for (test_pool_it it = a_testPool.begin(); it != a_testPool.end(); it++) {
479 if (((*it).second->getState() == TestCase::State::Success) && (!getDumpSuccessfulReports())) continue;
480 if (((*it).second->getState() == TestCase::State::Failed) && (!getDumpFailedReports())) continue;
481 (*it).second->asXML(testCases);
488 std::string TestManager::asXMLString() const throw() {
489 anna::xml::Node root("root");
490 return anna::xml::Compiler().apply(asXML(&root));