Added unit tests for DURATION in iCalendarEvent/ObjectDataTests.

[xestiacalendar/.git] / source / objects / calendarevent / CalendarEvent.cpp

diff --git a/source/objects/calendarevent/CalendarEvent.cpp b/source/objects/calendarevent/CalendarEvent.cpp
index 0623177..50b535b 100644 (file)
--- a/source/objects/calendarevent/CalendarEvent.cpp
+++ b/source/objects/calendarevent/CalendarEvent.cpp
@@ -807,4 +807,27 @@ void CalendarEventObject::ProcessData(){
                
        }
        
+       if (DateTimeEndProcessed == false){
+               
+               // Process the data from DURATION if DTEND 
+               // hasn't already been processed.
+               
+               DataReceived = ProcessTextVectors(&ObjectName, &ObjectData, false, "DURATION");
+               
+               if (DataReceived.begin() != DataReceived.end()){
+       
+                       try {
+                               DurationDataTokens = DataReceived.begin()->first.substr(9);
+                       }
+               
+                       catch(const out_of_range &oor){
+                               // Do nothing as there is no data.
+                       }               
+               
+                       DurationData = DataReceived.begin()->second;
+               
+               }
+               
+       }
+       
 }