i]4l dZddlZddlZddlZddlZddlmZddlmZmZddl m Z ddl m Z ddlm Z ddlZddlmZdd lmZmZmZmZmZmZmZmZmZmZmZdd lmZdd lm Z dd lm!Z"d dl#m$Z$ ddl%Z%n #e&$rdZ%YnwxYwe%duZ'e're%j(e%jdgZ)dgZ*dgZ+ngZ)gZ*gZ+GddZ,e e-j.Z/e/dz Z0e/dz Z1e/dz Z2e/dz Z3e/dz Z4e/dz Z5ej6ddZ7ej6ddZ8ej6ddZ9ej6ddZ:ej6dd Z;ej6dd!Z<ej6e)e dej=ej>dej(gz"d#Z(ej6e*d$d%gz"d&Z?d'Z@d(ZAd)ejBe0Ce1Ce2CDZDd*ejBe0Ce1Ce2CDZEd+ejBe0Ce1Ce2CDZFd,efd-ZGej6eD"d,efd.ZHej6eE"d,efd/ZIej6eF"d,efd0ZJd1e0CDZKej6eK"d2ZLej6d3ZMej6d4ZNej6d5ZOej6d6ZPej6d7ZQej6d8ZRej6d9ZSeSZTej6d:ZUej6d;ZVej6d<ZWej6d=ZXej6dd,ee ddffd>Z!ej6e+d?gz"d@ZYej6d,eZfdAZ[ej6d,eZfdBZ\ej6d,eZfdCZ]ej6d,eZfdDZ^dEZ_GdFdGe Z`dHZadIZbedJdKdLdMdNdOZcecdej(PZedQZfej6d,eZfdRZgej6dSTdUZhej6dVZiej6e$jj"dWejkd,eZfdXZlej6e"dYZmej6dZZnej6d,ejofd[Zpej6d\ZqdS)]zTest configurationN) Generator)datetimetimezone)PathMock)ZoneInfo)tz) Alarm AvailabilityCalendar ComponentComponentFactoryEventTimezoneTodo TypesFactoryprop vUTCOffset)PROPERTY_NAMES)TZPtzp) timezone_idsUTCcPtj||SN)pytzrlocalizedttznames S/root/projects/butler/venv/lib/python3.11/site-packages/icalendar/tests/conftest.pyr%/s4=0099"==rcZeZdZdZddgZdefdZdZdZdZ d Z d Z e d Z d S) DataSourcezFA collection of parsed ICS elements (e.g calendars, timezones, events).ics.jcaldata_source_folderc"||_||_dSr)_parser_data_source_folder)selfr+parsers r$__init__zDataSource.__init__=s #5   r&cNfdjDS)z(Return all the files that could be used.c^g|])}|jjv"|j*S)suffixlower extensionsstem).0pr/s r$ z#DataSource.keys..Cs>   x~~4?22 F222r&)r.iterdirr/s`r$keyszDataSource.keysAs=    -5577    r&cF|j}d|vr|dd\}}d|zg}|D]'}||z}|j|z }|rn(|s(t |dd|d|dkr+|}tj|}d}n+| }| |}d}t|ts||_ ||_||_||j|<|S)z;Parse a file and return the result stored in the attribute..rz' does not exist with these extensions: z, r*N)r7rsplitr.is_fileAttributeErrorjoin read_textr from_jcal read_bytesr- isinstancelistraw_icsraw_jcal source_file__dict__) r/ attributer7 extensionrL source_pathrKsourcerJs r$ __getitem__zDataSource.__getitem__IsY_ )  #,#3#3C#;#; Iy /*J#  I#i/K2[@K""$$  ""$$  ]]TYYzEZEZ]]]    ",,..H(22FGG!,,..G\\'**FH&$'' -$FN&FO!,F #) i  r&cf|ddd}||vS)zkey in self.keys()r@rr)rAr>r/keys r$ __contains__zDataSource.__contains__is-jja  #diikk!!r&c||Srr4rTs r$ __getattr__zDataSource.__getattr__ns Cyr&c*t|jSr)reprrMr=s r$__repr__zDataSource.__repr__qsDM"""r&c@jfdS)z'Return a list of all components parsed.c2|dS)NT)multiple)r-)datar/s r$r%z%DataSource.multiple..xs4<<t<3T3Tr&) __class__r.r=s`r$r^zDataSource.multiplets-~~  $&T&T&T&T   r&N)__name__ __module__ __qualname____doc__r7rr1r>rRrVrXr[propertyr^r4r&r$r(r(8sPP'"J646666   @""" ###  X   r&r( calendars timezoneseventsalarmsavailabilitiestodosmodulescopec@tttjSr)r(CALENDARS_FOLDERr from_icalrs r$rfrf &(: ; ;;r&c@tttjSr)r(TIMEZONES_FOLDERrrqrs r$rgrgrrr&c@tttjSr)r( EVENTS_FOLDERrrqrs r$rhrh mU_ 5 55r&c@tttjSr)r( ALARMS_FOLDERr rqrs r$ririrwr&c@tttjSr)r(AVAILABILITIES_FOLDERr rqrs r$rjrjs +\-C D DDr&c@tttjSr)r( TODOS_FOLDERrrqrs r$rkrks lDN 3 33r&)paramsc|jSrparamrequestrs r$utcrs  =r&cR|tj|SNtzinfo)replacer gettzr!s r$r%r%s2::RXf-=-=:>>r&cH|t|Sr)rr r!s r$r%r%s2::Xf-=-=:>>r&c|jSrrrs r$ in_timezoners  =r& fuzz_testcase)zbig_bad_calendar.icszissue_104_broken_calendar.icszsmall_bad_calendar.icsz multiple_calendar_components.icszpr_480_summary_with_colon.icszparsing_error_in_UTC_offset.icszparsing_error.icscfg|].}|jtvr|jdvrt|jv'|j/S))r)r*nameBROKEN_SOURCE_FILESr5FUZZ_TESTCASES_BROKEN_CALENDARSr9files r$r;r;sQ  y+++ ((('ty88  I 988r&cjg|]0}|jtvr |jdkrt|jv)|j1S)r)rrs r$r;r;sP     y+++ v'ty88  I 988r&cjg|]0}|jtvr |jdkrt|jv)|j1S)r*rrs r$r;r;sP  y+++ w'ty88  I 988r&returnc|j}td||||fD]}||vr ||cStd|d)Nz example file:zCould not find file r@)rprint ValueError)rfrgrhrrLr_s r$get_source_filersj-K /;'''9f,%% $   $ $ $ $  :K::: ; ;;r&c&t||||S)zqAn example file. source_file.raw_ics - bytes if .ics file source_file.raw_jcal - str if .jcal file rrrfrgrhrs r$rLrLs 9i A AAr&c&t||||S)z[An example .ica file. source_file.raw_ics - bytes source_file.raw_jcal - None rrs r$ics_filer 9i A AAr&c&t||||S)zZAn example .jcal file. source_file.raw_ics - None source_file.raw_jcal - str rrs r$ jcal_filerrr&c>g|]}tt|v|Sr4)rstr)r9rUs r$r;r;s3    &#c((22222r&c|jS)zClusterfuzz calendars.rrs r$fuzz_v1_calendar_pathr  =r&ctSzReturn a new types factory.)rr4r&r$ types_factoryrs >>r&ctSrrr4r&r$component_factoryr   r&c#ZKd|jd<dV|jddS)zMap x_sometime to timetimez X-SOMETIMEN) types_mappop)rs r$ x_sometimer!s;-3ML) EEE -----r&ctS)zReturn a new component factory.rr4r&r$factoryr)rr&c#BKdt_dVdt_dS)NTF)rignore_exceptionsr4r&r$vUTCOffset_ignore_exceptionsr/s%#'J  EEE#(J   r&c0t}d|_|S)zReturn an event component.VEVENTrrrcs r$event_componentr6s  A AF Hr&ctS)Return an empty component.)rrs r$rr>s ;;r&c0t}d|_|S)r VCALENDARrrs r$calendar_componentrGs  A AF Hr&c#ZKtj}dt_dV|t_dS)z*Remove validation for new() of components.FN)r _validate_new)values r$dont_validate_newrOs/  #E#I EEE#Ir&ctd}d|_|ddd|dd d|||S) zBReturn an event with some values and add it to calendar_component.zA brief history of time)summaryrdtend20000102T000000r)encodedtstart20000101T000000)rradd add_component)rres r$filled_event_componentrXsg 3444A AFEE'$QE///EE)&qE111$$Q''' Hr&c,t}d|d<|S)NzChair, Table, "Room: 42" resources)r rs r$calendar_with_resourcesrcs A/AkN Hr&c#jKtj|tVtjdS)zThe timezone provider.N)_tzpuse use_defaulttzp_names r$rrjs6 HX JJJr&zoneinfoc*t|jS)zThis is annother timezone provider. The purpose here is to cross test: pytz <-> zoneinfo. tzp as parameter makes sure we test the cross product. )rrrs r$ other_tzprrs w}  r&c2|sJ|S)z+Skip tests that are not running under pytz.) uses_pytz)rrs r$ pytz_onlyr|s ==?? Or&c2|sJ|S)z/Skip tests that are not running under zoneinfo.) uses_zoneinfo)rrrs r$ zoneinfo_onlyrs!      Or&c|dksJ|S)zDo not run tests with pytz.rr4rs r$no_pytzrs v     Or&c|dksJ|S)zDo not run tests with zoneinfo.rr4rs r$ no_zoneinfors z ! ! ! ! Or&c&d|jvrtdgz}d|jvrdg}d|jvrt}d|jvrd|jvr JddD]'}d||jvr||vr||(|d|d d Sd S) aParametrize without skipping: tzp_name will be parametrized according to the use of - pytz_only - zoneinfo_only - no_pytz - no_zoneinfo See https://docs.pytest.org/en/6.2.x/example/parametrize.html#deferring-the-setup-of-parametrized-resources rrrrz,Use pytz_only or zoneinfo_only but not both!)rrno_rlrmN) fixturenamesPYTZ_TZPremove parametrize)metafunc tzp_namesrs r$pytest_generate_testsrsX*** |+ h3 3 3# I (/ / / I x4 4 4x4444 9544( ' 'DT||x4449J9J  &&&Z(CCCCC+*r&ceZdZdZdZdS)DoctestZoneInfoz+Constent ZoneInfo representation for tests.cd|jdS)Nz ZoneInfo(key=))rUr=s r$r[zDoctestZoneInfo.__repr__s,tx,,,,r&N)rarbrcrdr[r4r&r$rrs)55-----r&rcDd|D}td|DdS)z doctest printcfg|].}t|tr|dn|/S)zUTF-8)rHbytesdecoder9args r$r;z!doctest_print..s8 S S Ss:c5#9#9 BCJJw   s S S Sr&c3K|]M}t|ddddVNdS)z   N)rstriprrs r$ z doctest_print..sR W W3CHHNN   $ $VT 2 2 : :4 F F W W W W W Wr&N)r)argss r$ doctest_printrs6 S Sd S S SD W WRV W W WXXXXr&c<|dkrtSt|g|Ri|S)zIReplace the import mechanism to skip the whole doctest if we import pytz.r)r __import__)rrkws r$doctest_importr s. v~~ d (T ( ( (R ( ((r&i rz$d755cef5-2311-46ed-a0e1-6733c9e15c63ctS)z!The UID that tests always create.) UID_DEFAULTr4r&r$test_uidrs  r&T)autousectjtd|tdfd|tdt ddS)zYCreate a fixed test environment for the functions that are time and randomness dependent.)versionuuid4cSrr4)uidsr$r%zfixed_env..ssr&_utc_nowctSr)NOW_UTCr4r&r$r%zfixed_env..sGr&N)uuidUUIDrsetattrr staticmethod) monkeypatchrs @r$ fixed_envr!s_ )K + + +Cg{{{{333 :|OO/L/LMMMMMr&c|tjdt|tdt ddlm}||dtjddtiS)z,Modify the environment to make doctests run.rr r)ZONEINFOrrr) setitemsysmodulesrrricalendar.timezone.zoneinfor#r r)r r#s r$env_for_doctestr(sv Z:::*o>>>444444%):5)A)ABBB ] ##r&rc|jS)zxReturn a timezone id to be used with pytz or zoneinfo. This goes through all the different timezones possible. rrs r$tzidr*s =r&c|jS)z@Names of property types that occur as a property of a component.rrs r$ v_prop_namer,rr&c,tt|S)z7Property types that occur as a property of a component.)getattrr)r,s r$v_propr/s 4 % %%r&c6|dS)Nr)examples)r/s r$v_prop_exampler2s ??  Q r&ctS)zA mock.rr4r&r$mockr4s  66Mr&)rrd itertoolsr%rrcollections.abcrrrpathlibr unittest.mockrr pytestdateutilr icalendarr r r rrrrrrrricalendar.tests.dataricalendar.timezonerrrrr ImportErrorHAS_PYTZrPYTZ_UTCPYTZ_IN_TIMEZONErr(__file__parentHERErprtrvryr{r}fixturerfrgrhrirjrkrrrrrchainr< SOURCE_FILES ICS_FILES JCAL_FILESrrLrrFUZZ_V1rrrrrrrrcomprrrrrrrrrrrrrr NOWrrrrr!r(TZIDSFixtureRequestr*r,r/ VPROPERTYr2r4r4r&r$rQs  %%%%%%''''''''                           0/////""""""******KKKK DDD t    eH >=xHHHHA A A A A A A A H tH~~+%+%x x //g~ h<< <h<< <h66 6h66 6hEE Eh44 4 xxN N >>>> #2   ""$4$<$<$>$> @U@U@W@W      ""$4$<$<$>$> @U@U@W@W       ""$4$<$<$>$> @U@U@W@W A?