14 #define BOOST_TEST_DYN_LINK
15 #define BOOST_TEST_MAIN
16 #define BOOST_TEST_MODULE DemandGenerationTest
17 #include <boost/test/unit_test.hpp>
19 #include <stdair/stdair_basic_types.hpp>
20 #include <stdair/basic/BasConst_General.hpp>
21 #include <stdair/basic/BasLogParams.hpp>
22 #include <stdair/basic/BasDBParams.hpp>
23 #include <stdair/basic/BasFileMgr.hpp>
24 #include <stdair/basic/ProgressStatusSet.hpp>
25 #include <stdair/bom/EventStruct.hpp>
26 #include <stdair/bom/BookingRequestStruct.hpp>
27 #include <stdair/service/Logger.hpp>
32 #include <trademgen/config/trademgen-paths.hpp>
34 namespace boost_utf = boost::unit_test;
37 std::ofstream utfReportStream (
"DemandGenerationTestSuite_utfresults.xml");
42 struct UnitTestConfig {
45 boost_utf::unit_test_log.set_stream (utfReportStream);
46 #if BOOST_VERSION_MACRO >= 105900
47 boost_utf::unit_test_log.set_format (boost_utf::OF_XML);
49 boost_utf::unit_test_log.set_format (boost_utf::XML);
51 boost_utf::unit_test_log.set_threshold_level (boost_utf::log_test_units);
61 typedef std::pair<stdair::Count_T, stdair::Count_T> NbOfEventsPair_T;
62 typedef std::map<
const stdair::DemandStreamKeyStr_T,
63 NbOfEventsPair_T> NbOfEventsByDemandStreamMap_T;
69 void testDemandGenerationHelper (
const unsigned short iTestFlag,
70 const stdair::Filename_T& iDemandInputFilename,
71 const stdair::DemandGenerationMethod& iDemandGenerationMethod,
72 const bool isBuiltin) {
75 const stdair::RandomSeed_T lRandomSeed = stdair::DEFAULT_RANDOM_SEED;
78 std::ostringstream oStr;
79 oStr <<
"DemandGenerationTestSuite_" << iTestFlag <<
".log";
80 const stdair::Filename_T lLogFilename (oStr.str());
83 std::ofstream logOutputFile;
85 logOutputFile.open (lLogFilename.c_str());
86 logOutputFile.clear();
89 const stdair::BasLogParams lLogParams (stdair::LOG::DEBUG, logOutputFile);
101 NbOfEventsByDemandStreamMap_T lNbOfEventsMap;
104 stdair::Count_T lRefExpectedNbOfEvents (0);
105 stdair::Count_T lRefActualNbOfEvents (0);
108 if (isBuiltin ==
true) {
111 trademgenService.buildSampleBom();
113 lNbOfEventsMap.insert (NbOfEventsByDemandStreamMap_T::
114 value_type (
"SIN-BKK 2010-Feb-08 Y",
115 NbOfEventsPair_T (4, 60)));
116 lNbOfEventsMap.insert (NbOfEventsByDemandStreamMap_T::
117 value_type (
"BKK-HKG 2010-Feb-08 Y",
118 NbOfEventsPair_T (4, 60)));
119 lNbOfEventsMap.insert (NbOfEventsByDemandStreamMap_T::
120 value_type (
"SIN-HKG 2010-Feb-08 Y",
121 NbOfEventsPair_T (4, 60)));
124 lRefExpectedNbOfEvents = 180;
125 lRefActualNbOfEvents = 186;
131 trademgenService.parseAndLoad (lDemandFilePath);
133 lNbOfEventsMap.insert (NbOfEventsByDemandStreamMap_T::
134 value_type (
"SIN-HND 2010-Feb-08 Y",
135 NbOfEventsPair_T (1, 10)));
136 lNbOfEventsMap.insert (NbOfEventsByDemandStreamMap_T::
137 value_type (
"SIN-HND 2010-Feb-09 Y",
138 NbOfEventsPair_T (1, 10)));
139 lNbOfEventsMap.insert (NbOfEventsByDemandStreamMap_T::
140 value_type (
"SIN-BKK 2010-Feb-08 Y",
141 NbOfEventsPair_T (1, 10)));
142 lNbOfEventsMap.insert (NbOfEventsByDemandStreamMap_T::
143 value_type (
"SIN-BKK 2010-Feb-09 Y",
144 NbOfEventsPair_T (1, 10)));
147 lRefExpectedNbOfEvents = 40;
148 lRefActualNbOfEvents = 40;
153 const stdair::Count_T& lExpectedNbOfEventsToBeGenerated =
154 trademgenService.getExpectedTotalNumberOfRequestsToBeGenerated();
156 BOOST_CHECK_EQUAL (lRefExpectedNbOfEvents,
157 std::floor (lExpectedNbOfEventsToBeGenerated));
159 BOOST_CHECK_MESSAGE (lRefExpectedNbOfEvents ==
160 std::floor (lExpectedNbOfEventsToBeGenerated),
161 "Expected total number of requests to be generated: "
162 << lExpectedNbOfEventsToBeGenerated
164 << std::floor (lExpectedNbOfEventsToBeGenerated)
165 <<
"). Reference value: " << lRefExpectedNbOfEvents);
172 const stdair::Count_T& lActualNbOfEventsToBeGenerated =
173 trademgenService.generateFirstRequests(iDemandGenerationMethod);
176 STDAIR_LOG_DEBUG (
"Expected number of events: "
177 << lExpectedNbOfEventsToBeGenerated <<
", actual: "
178 << lActualNbOfEventsToBeGenerated);
181 BOOST_CHECK_GE (lRefActualNbOfEvents, lActualNbOfEventsToBeGenerated - 10.0);
182 BOOST_CHECK_LE (lRefActualNbOfEvents, lActualNbOfEventsToBeGenerated + 10.0);
184 BOOST_CHECK_MESSAGE (lRefActualNbOfEvents<=lActualNbOfEventsToBeGenerated+10.0
185 &&lRefActualNbOfEvents>=lActualNbOfEventsToBeGenerated-10.0,
186 "Actual total number of requests to be generated: "
187 << lExpectedNbOfEventsToBeGenerated
189 << std::floor (lExpectedNbOfEventsToBeGenerated)
190 <<
"). Reference value: " << lRefActualNbOfEvents);
193 const bool isQueueDone = trademgenService.isQueueDone();
194 BOOST_REQUIRE_MESSAGE (isQueueDone ==
false,
195 "The event queue should not be empty.");
204 stdair::Count_T idx = 1;
205 while (trademgenService.isQueueDone() ==
false) {
208 stdair::EventStruct lEventStruct;
209 stdair::ProgressStatusSet lPPS = trademgenService.popEvent (lEventStruct);
212 STDAIR_LOG_DEBUG (
"Poped event: '" << lEventStruct.describe() <<
"'.");
215 const stdair::BookingRequestStruct& lPoppedRequest =
216 lEventStruct.getBookingRequest();
219 STDAIR_LOG_DEBUG (
"Poped booking request: '"
220 << lPoppedRequest.describe() <<
"'.");
223 const stdair::DemandGeneratorKey_T& lDemandStreamKey =
224 lPoppedRequest.getDemandGeneratorKey();
227 const NbOfEventsByDemandStreamMap_T::iterator itNbOfEventsMap =
228 lNbOfEventsMap.find (lDemandStreamKey);
229 BOOST_REQUIRE_MESSAGE (itNbOfEventsMap != lNbOfEventsMap.end(),
230 "The demand stream key '" << lDemandStreamKey
231 <<
"' is not expected in that test");
242 const NbOfEventsPair_T& lNbOfEventsPair = itNbOfEventsMap->second;
243 stdair::Count_T lCurrentNbOfEvents = lNbOfEventsPair.first;
244 const stdair::Count_T& lExpectedTotalNbOfEvents = lNbOfEventsPair.second;
247 const bool stillHavingRequestsToBeGenerated = trademgenService.
248 stillHavingRequestsToBeGenerated (lDemandStreamKey, lPPS,
249 iDemandGenerationMethod);
257 if (lCurrentNbOfEvents == 1) {
263 const stdair::ProgressStatus& lDemandStreamProgressStatus =
264 lPPS.getSpecificGeneratorStatus();
265 const stdair::Count_T& lNbOfRequests =
266 lDemandStreamProgressStatus.getExpectedNb();
268 BOOST_CHECK_EQUAL (lNbOfRequests, lExpectedTotalNbOfEvents);
269 BOOST_CHECK_MESSAGE (lNbOfRequests == lExpectedTotalNbOfEvents,
270 "[" << lDemandStreamKey
271 <<
"] Total number of requests to be generated: "
272 << lNbOfRequests <<
"). Expected value: "
273 << lExpectedTotalNbOfEvents);
277 STDAIR_LOG_DEBUG (
"=> [" << lDemandStreamKey <<
"][" << lCurrentNbOfEvents
278 <<
"/" << lExpectedTotalNbOfEvents
279 <<
"] is now processed. "
280 <<
"Still generate events for that demand stream? "
281 << stillHavingRequestsToBeGenerated);
285 if (stillHavingRequestsToBeGenerated ==
true) {
286 const stdair::BookingRequestPtr_T lNextRequest_ptr =
287 trademgenService.generateNextRequest (lDemandStreamKey,
288 iDemandGenerationMethod);
289 assert (lNextRequest_ptr != NULL);
296 const stdair::Duration_T lDuration =
297 lNextRequest_ptr->getRequestDateTime()
298 - lPoppedRequest.getRequestDateTime();
299 BOOST_REQUIRE_GT (lDuration.total_milliseconds(), 0);
300 BOOST_REQUIRE_MESSAGE (lDuration.total_milliseconds() > 0,
301 "[" << lDemandStreamKey
302 <<
"] The date-time of the generated event ("
303 << lNextRequest_ptr->getRequestDateTime()
304 <<
") is lower than the date-time "
305 <<
"of the current event ("
306 << lPoppedRequest.getRequestDateTime() <<
")");
309 STDAIR_LOG_DEBUG (
"[" << lDemandStreamKey <<
"][" << lCurrentNbOfEvents
310 <<
"/" << lExpectedTotalNbOfEvents
311 <<
"] Added request: '" << lNextRequest_ptr->describe()
312 <<
"'. Is queue done? "
313 << trademgenService.isQueueDone());
316 ++lCurrentNbOfEvents;
317 itNbOfEventsMap->second = NbOfEventsPair_T (lCurrentNbOfEvents,
318 lExpectedTotalNbOfEvents);
327 if (iDemandGenerationMethod == stdair::DemandGenerationMethod::STA_ORD) {
329 BOOST_CHECK_GE (idx, lRefActualNbOfEvents - 10.0);
330 BOOST_CHECK_LE (idx, lRefActualNbOfEvents + 10.0);
331 BOOST_CHECK_MESSAGE (idx >= lRefActualNbOfEvents - 10.0
332 && idx <= lRefActualNbOfEvents + 10.0,
333 "The total actual number of events is "
334 << lRefActualNbOfEvents <<
", but " << idx
335 <<
" events have been generated");
340 trademgenService.reset();
343 STDAIR_LOG_DEBUG (
"End of the simulation");
346 logOutputFile.close();
354 BOOST_GLOBAL_FIXTURE (UnitTestConfig);
357 BOOST_AUTO_TEST_SUITE (master_test_suite)
362 BOOST_AUTO_TEST_CASE (trademgen_simple_simulation_test) {
365 const stdair::Filename_T lInputFilename (STDAIR_SAMPLE_DIR
"/demand01.csv");
368 const stdair::DemandGenerationMethod lDemandGenerationMethod (stdair::DemandGenerationMethod::STA_ORD);
371 const bool isBuiltin =
false;
372 BOOST_CHECK_NO_THROW (testDemandGenerationHelper(0,
374 lDemandGenerationMethod,
382 BOOST_AUTO_TEST_CASE (trademgen_missing_input_file_test) {
385 const stdair::Filename_T lInputFilename (STDAIR_SAMPLE_DIR
"/missingFile.csv");
388 const stdair::DemandGenerationMethod lDemandGenerationMethod (stdair::DemandGenerationMethod::STA_ORD);
391 const bool isBuiltin =
false;
392 BOOST_CHECK_THROW (testDemandGenerationHelper(1,
394 lDemandGenerationMethod,
403 BOOST_AUTO_TEST_CASE (trademgen_default_bom_simulation_test) {
406 const stdair::DemandGenerationMethod lDemandGenerationMethod (stdair::DemandGenerationMethod::STA_ORD);
409 const bool isBuiltin =
true;
410 BOOST_CHECK_NO_THROW (testDemandGenerationHelper(2,
412 lDemandGenerationMethod,
420 BOOST_AUTO_TEST_CASE (trademgen_poisson_process_test) {
423 const stdair::DemandGenerationMethod lDemandGenerationMethod (stdair::DemandGenerationMethod::POI_PRO);
426 const bool isBuiltin =
true;
427 BOOST_CHECK_NO_THROW (testDemandGenerationHelper(3,
429 lDemandGenerationMethod,
435 BOOST_AUTO_TEST_SUITE_END()
class holding the services related to Travel Demand Generation.