U jS@sdZddlmZddlmZmZmZddlZddlmZm Z ddl m Z zddl m Z Wn ek rxddlm Z YnXddlmZmZmZmZmZdd lmZd d d d d dZejeegdddZejedddddedddeddddde ddgdddZddZddZdd Zd!d"Zejeeeegdd#d$Z d%d&Z!d'd(Z"eZ#e#$deddddde#$deddddd)e%e#&Z'eZ(e($d*ed+d,d,ed+d-d,fee#Z)ee'Z*ee(Z+ej,-d.e#e'e(e)e*e+gd/d0Z.d1d2Z/ejeddddd)eddd3edddd4de ddgdd5d6Z0d7d8Z1d9d:Z2d;d<Z3d=d>Z4eje1e4e2e3gdd?d@Z5dAdBZ6dCdDZ7ej,-dEd*dFgdGdHZ8dIdJdKdLdMZ9dIdJdKdNdOZ:dIdJdKdPdQZ;eje9e:e;gddRdSZe>$d*edVd,d,dWdXe>$dFedVd,d,eZ?e?$dFedVd,d,dWdXe?$d*edVd,d,eZ@e@$dFedVd,d,dWdXe@$d*edVd,d,dWdXe@$dYed,dZeZAeA$d*edVd,d,eA$dYed,d[eZBeB$d*edVd,d,dWdXeB$d\edVd,d,eZCeC$d\edVd,d,dWdXeC$d*edVd,d,eZDeD$d\edVd,d,dWdXeD$d*edVd,d,dWdXeD$dYed,dZeZEeE$d*edVd,d,eE$dYed,d[ej,-d]e>d^fe?d^fe@d_feAd`feBdafeCdafeDdbfeEd`fgej,-dEdcddgdedfZFdgdhZGdidjZHdkdlZIdmdnZJeZKeZLeL$dYed,d[eZMeZNeN$dYed,d[ej,-doeKeLeMeNgej,-dEdcdddSgdpdqZOej,-drePed,dZedVdWdsdWdXed,dZfgej,-dtedcfeddfed*fedFfedcfeddfed*fedcfeddfed*fed\fg dudvZQdwdxdydzZRej,-drePdedVdWdsdWdXed,dZfed{d-d-edd-d-gd|d}ZSd~dZTddZUej,-dd*edVdWdsdWdXfdFedVdWdsdWdXfdYed,dZfgZVeVddZWeVddZXej,-dEd*dFdYgddZYddZZej,-dEd*dddcgej,-dcedVdWdsdWdXedVdWdsedVdWdsdWdXe ddgddZ[ej,-dEdcddgddZ\ddZ]ej,-dde%ddfdeddZdfdeddZdfgddZ^ddZ_ddZ`ddZaddZbej,-dddde%de%de%de%de%dfffddeddZfdffdeddZfddffgddZcddZded,d[ZeddZfddZgej,-dddde%dfffddeddZfdffdeddZfddffgej,-dSeddZeegej,-dd,dgddZhdS)z8This tests the properties of components and their types.) annotations)datedatetime timedeltaN)IncompleteComponentInvalidCalendar)Alarm)ZoneInfo)EventJournalTodo vDDDTypes vDatetime vDurationz Event | Todostr) componentpropreturncCst|tr|dkrdS|S)z]Translate the end property. This allows us to run the same tests on Event and Todo. DTENDDUE) isinstancer upper)rrrW/tmp/pip-unpacked-wheel-g40nyxg1/icalendar/tests/test_issue_662_component_properties.pyrsr)paramscCs|S)zThe event to test.paramrequestrrrstart_end_component$sr i z Europe/Paris)tzinfocCs|||j|jS)zStart of the event.r)rset_component_startr rrrdtstart*s r'cCs2t|}t||d<||t||dS)*Create the event with the __init__ method.r'N)dictr clearupdatetype)rstartdrrr_set_component_start_init7s r/cCs ||_dSz+Create the event with the dtstart property.N)DTSTARTrr-rrr_set_component_dtstart?sr3cCs ||_dSr0)r-r2rrr_set_component_start_attrDsr4cCs@|d||}t|||t||dS)z)Create the event with the start property.r'N)addto_icaldecodeprintr*r+r, from_ical)rr-icsrrr_set_component_start_icsIs   r;cCs|jSzCreate a new event.rrrrrr&Rs r&cCs|j|kstdSzTest the start of events.N)r1AssertionErrorr'r rrrtest_component_dtstart_sr@cCs|j|kstdSr=)r-r>r?rrrtest_event_startdsrAr1i invalid_eventc Cs<tt |jW5QRXtt |jW5QRXdS)"Check that we get the right error.N)pytestraisesrr-r1)rErrrtest_multiple_dtstartts  rIc Cs.|jdksttt |jW5QRXdS)zDTSTART is optional. The following is REQUIRED if the component appears in an iCalendar object that doesn't specify the "METHOD" property; otherwise, it is OPTIONAL; in any case, it MUST NOT occur more than once. N)r1r>rGrHrr-r rrrtest_no_dtstarts  rKcCs|||j|jS)zend of the event.r)rset_component_endr rrrdtends rOcCs8t|}t||t|d<||t||dS)r(rON)r)r rr*r+r,)rendr.rrr_set_component_end_initsrQcCst|t|d|dS))Create the event with the dtend property.rN)setattrrrrPrrr_set_component_end_propertysrUcCs ||_dS)rRN)rPrTrrr_set_component_end_attrsrVcCsF|t|d||}t|||t||dS)z'Create the event with the end property.rN) r5rr6r7r8r*r+r,r9)rrPr:rrr_set_component_end_icss  rWcCs|jSr<rrrrrrNs rNcCs t|d}t|||kstdS)Test the end of events.rN)rgetattrr>)rOr attrrrrtest_component_end_propertys r[cCs|j|kstdS)rXN)rPr>)rOr rrrtest_component_endsr\rZrcCs4t||}t||t||dks&tt||dSN)rdelattrrYr>)r r'rOrZrrrtest_delete_attrs  r_r reventdurationcCst||d<dSzSet the vDDDTypes valueDURATIONN)r r`rrr_set_duration_vdddtypessrecCs|d|dSrc)r5r`rrr_set_duration_addsrfcCst||d<dSrcrr`rrr_set_duration_vdurationsrgcCs.t|trtddntdd}||||S)z... events have a DATE value type for the "DTSTART" property ... If such a "VEVENT" has a "DURATION" property, it MUST be specified as a "dur-day" or "dur-week" value. rChoursrDdays)rrrr)r r'rrbrrrrbs  rbcCs*|j|j}||kst|j|ks&tdS)z'Check calculation of end with duration.N)rPr-r>rb)r r'rbZdurrrrtest_start_and_durations  rl rdrjrhr)invalid_componentmessagezDDTSTART and DTEND must be of the same type, either date or datetime.z)rprqrZerrrtest_invalid_event$s( rucCs@t}tddddd|_|j|jks(t|jtddksrbrrarrrtest_event_duration_zeroQsrxcCsFt}tddd|_|j|jtddks.t|jtddksBtdS)z For cases where a "VEVENT" calendar component specifies a "DTSTART" property with a DATE value type but no "DTEND" nor "DURATION" property, the event's duration is taken to be one day rmrnrvrCrjNr rr-rPrr>rbrwrrrtest_event_duration_one_day^srzcCs@t}tddddd|_|j|jks(t|jtddksrbr)todorrrtest_todo_duration_zeroksr|cCsFt}tddd|_|j|jtddks.t|jtddksBtdS)aThe end is at the end of the day, excluding midnight. RFC 5545: The following is an example of a "VTODO" calendar component that needs to be completed before May 1st, 2007. On midnight May 1st, 2007 this to-do would be considered overdue. rmrnrvrCrjNryrwrrrtest_todo_duration_one_dayssr}incomplete_event_endc Cs$ttt||W5QRXdS)z)Test that the end throws the right error.N)rGrHrrY)r~rZrrrtest_incomplete_events r invalid_valuerv) ComponentrZc CsN|}tt}t|||W5QRX|jjddt|jdksJtdS)z^Check that we get the right error. - other types that vDDDTypes accepts - object rzUse datetime or date, not .N) rGrH TypeErrorrSrrrsr,__name__r>)rrZrrrtrrrtest_set_invalid_starts  rr))r.cCs |||<dSr]r)r.keyrrrrrsetitemsric CsJ||d<tt }|jW5QRX|jjddt|jdksFtdS)rFrdrz"DURATION must be a timedelta, not rN) rGrHrrdrrrsr,rr>)r rrtrrrtest_check_invalid_durations   rc Cst|d}tddddd|_tdd|_t||tdddddd |ksLt|jd ksZtt||}|tdddddks|td S) z+Setting the end should not break the event.rrmrnrvrorCrjrdN)rrr1rrdrSr>rY)r rrPrrr)test_setting_the_end_deletes_the_durations    rc Csvt|d}tddddd|_t||tdddddtdd|_||ksLtt||d ks^t|jtddksrtd S) z0Setting the duration should not break the event.rrmrnrvrorrCrjN)rrr1rSrrdr>rY)r rrrr%test_setting_duration_deletes_the_ends   r)rZrrcCsPt||}t|||||ks"tt|||ks4tt||d||ksLtdS)z(Setting attributes to None deletes them.N)rrSr>rYr rZrrrrr"test_setting_to_none_deletes_values     rcCsBt||}t|||tddt|||t|||ks>tdS)z'Setting attributes twice replaces them.rCrjN)rrSrrYr>rrrrtest_setting_a_value_twices  rc Cs6t||}d||<ttt||W5QRXdS)zSpecial case for None.N)rrGrHrrY)r rZrrrtest_invalid_none s  rcCs"tdd|_|`|jdkstdS)zTest the del command.rCrjN)rrdr>rJrrrtest_delete_durations rcCsRt}t|||||jks t|j|ks.t|j|ksr-rPrbr)r-rZjrrrtest_journal_starts  rc Cs\t}tddddd|_d|_|jdks,td|ks8tttt||W5QRXdS)z Delete the start of the journal.irvr#r$rMNr1) r rr-r1r>rGrHrrY)rZrrrrtest_delete_journal_start0s  rcCsht}tddd|_tddd|_tddd|jks8t|jtdddksNt|jtdddksdtdS)NrmrC)r rr1r>r-rP)rrrr*setting_twice_does_not_duplicate_the_entry<s r)filetriggerrelatedZrfc_5545_absolute_alarm_exampleZ19970317T133000ZSTARTZ rfc_5545_endENDZ start_datecCs(||}|j|kst|j|ks$tdSGet the trigger property.N)TRIGGERr>TRIGGER_RELATED)alarmsrrralarmrrrtest_get_alarm_trigger_propertyEsrcCs8t}tdd|_|jtddks&t|jdks4tdS)Set the alarm trigger.rCrhrN)rrrr>rarrrtest_set_alarm_triggerXs rcCs>t}tdd|_d|_|jtddks,t|jdks:tdS)rrCrhrN)rrrrr>rrrrtest_set_alarm_trigger_related`s  rcCstjdkstdS)zThe default is startrN)rrr>rrrr test_get_related_without_triggerisrc Cs:tt}t}d|_W5QRX|jjddks6tdS)z)TRIGGER must be set to set the parameter.rrz)rtrrrr'test_cannot_set_related_without_triggerns  r)rtriggersrZ19970317T134500ZZ19970317T140000ZZ19970317T141500ZZ19970317T143000ZcCs0||}tt|jt||j|ks,tdSr)r8tuplerr>)rrrrrrrtest_get_alarm_triggersysrcCstjdkstdS)z)An alarm with no trigger has no triggers.)rrrN)rrr>rrrrtest_triggers_emtpy_alarmsrcCs,t}t|_t|_|jtfddfks(tdS)Check incomplete values.rN)rh1rrdrr>rrrr"test_triggers_emtpy_with_no_repeatsrcCs,t}t|_d|_|jtfddfks(tdS)rrnrN)rrrREPEATrr>rrrr$test_triggers_emtpy_with_no_durationsrrepeatrc Cs||}||_||_t||jD]v\}}|s>|dks$tq$t|d|ksRt|d|dksftt|dd|ddD]\}}|||kstqq$dS)rrrCrNr)copyrrdziprr>len) rrrrbrrexpectedxyrrr test_get_alarm_triggers_repeateds  "r)i__doc__ __future__rrrrrGZicalendar.errorrrZ icalendar.calrZzoneinfor ImportErrorZbackports.zoneinfoZ icalendarr r r r rZicalendar.proprrZfixturer r'r/r3r4r;r&r@rAZinvalid_start_event_1r5r9r6Zinvalid_start_event_2Zinvalid_start_event_3Zinvalid_start_todo_1Zinvalid_start_todo_2Zinvalid_start_todo_3markZ parametrizerIrKrOrQrUrVrWrNr[r\r_rerfrgrbrlZinvalid_event_end_1Zinvalid_event_end_2Zinvalid_event_end_3Zinvalid_event_end_4Zinvalid_todo_end_1Zinvalid_todo_end_2Zinvalid_todo_end_3Zinvalid_todo_end_4rurxrzr|r}Zincomplete_event_1Zincomplete_event_2Zincomplete_todo_1Zincomplete_todo_2robjectrrrrrZ valid_valuesrrrrrrrrrrrrrrrrrrrrrrs$                  %