o .j!9@sdZddlZddlZddlZddlZddlZddlmZddlmZm Z ddl m Z ddl m Z ddlmZddlZddlmZdd lmZmZmZmZmZmZmZmZmZmZmZmZdd lm Z dd l!m"Z"dd l!m#Z$d dl%m&Z&zddl'Z'Wn e(ydZ'Ynwe'duZ)e)re'j*e' dgZ+ddgZ,dgZ-ngZ+gZ,gZ-GdddZ.e e/j0Z1e1dZ2e1dZ3e1dZ4e1dZ5e1dZ6e1dZ7ej8deegdddZ9ej8ddZ:ej8dd d!dZ;ej8dd d"dZej8dd d%dZ?ej8e+edej@eAde j*gd&d'd(Z*ej8e,d)dd*dgd&d+d,ZBd-ZCd.ZDd/d0eEe2Fe3Fe4FDZGd1d0eEe2Fe3Fe4FDZHd2d0eEe2Fe3Fe4FDZId3efd4d5ZJej8eGd&d3efd6d7ZKej8eHd&d3efd8d9ZLej8eId&d3efd:d;ZMdZOej8d?d@ZPej8dAdBZQej8dCdDZRej8dEdFZSej8dGdHZTej8dIdJZUej8dKdLZVeVZWej8dMdNZXej8dOdPZYej8dQdRZZej8dSdTZ[ej8dd d3ee"ddffdUdVZ#ej8e-dWgd&dXdYZ\ej8d3e]fdZd[Z^ej8d3e]fd\d]Z_ej8d3e]fd^d_Z`ej8d3e]fd`daZadbdcZbGdddedeeZcdfdgZddhdiZeedjdkdldmdndoZfefjge j*dpZhdqZidrZjej8d3e]fdsdtZkej8dudvdwdxZlej8dydzZmej8e&jnd&d{ejod3e]fd|d}Zpej8e d&d~dZqej8ddZrej8d3ejsfddZtej8ddZudS)zTest configurationN) Generator)datetimetimezone)PathMock)ZoneInfo)tz) Alarm AvailabilityCalendar ComponentComponentFactoryEvent LazyCalendarTimezoneTodo TypesFactoryprop vUTCOffset)PROPERTY_NAMES)TZPtzp) timezone_idsUTCcCst||SN)pytzrlocalizedttznamer#M/home/thesage/.local/lib/python3.10/site-packages/icalendar/tests/conftest.py1r%rc@seZdZdZddgZ ddedeefddZd d Z d d Z d dZ ddZ ddZ eddZeejdedeefddZdS) DataSourcezFA collection of parsed ICS elements (e.g calendars, timezones, events).ics.jcalFdata_source_folderparsercCs||_||_||_dSr)_parser_data_source_folder _multiple)selfr*r+multipler#r#r$__init__?s zDataSource.__init__csfddjDS)z(Return all the files that could be used.cs"g|] }|jjvr|jqSr#)suffixlower extensionsstem).0pr/r#r$ Hs z#DataSource.keys..)r-iterdirr8r#r8r$keysFs zDataSource.keyscs|j}d|vr|dd\}}d|g}|D]}|||j}|r'nq|s9t|dd|ddd|dkrF|n|fdd}t|j |t |t ||S) z;Parse a file and return the result stored in the attribute..rz' does not exist with these extensions: z, Nr)csBr |jj|jdn|j}|jr|S|_|_|_|S)Nr0)r, from_icalr. from_jcalraw_icsraw_jcal source_file)r/sourcer@rArBr#r$ get_parsedhs z*DataSource.__getitem__..get_parsed) r4rsplitr-is_fileAttributeErrorjoin read_text read_bytessetattr __class__propertygetattr)r/ attributer4 extension source_pathrEr#rDr$ __getitem__Ns,    zDataSource.__getitem__cCs|ddd}||vS)zkey in self.keys()r<rr)rFr;r/keyr#r#r$ __contains__xs zDataSource.__contains__cCs||Srr#rTr#r#r$ __getattr__}szDataSource.__getattr__cC t|jSr)repr__dict__r8r#r#r$__repr__s zDataSource.__repr__cCs|j|j|jddS)z'Return a list of all components parsed.Tr=)rMr-r,r8r#r#r$r0szDataSource.multiplecCs td|j|fi}|||S)zEParse all files in a folder and return a DataSource with the results.r')typename)clsr*r+new_clsr#r#r$ from_folders zDataSource.from_folderN)F)__name__ __module__ __qualname____doc__r4rr\r r1r;rSrVrWr[rNr0 classmethod functoolscacher`r#r#r#r$r':s& * r' calendars timezoneseventsalarmsavailabilitiestodosmodule)scopeparamscCstt|jS)zReturn the data source for calendar files. We have two ways of parsing those: - Immediately - Lazily with LazyCalendar )r'r`CALENDARS_FOLDERparam)rrequestr#r#r$rhcC tttS)z*Return the data source for calendar files.)r'r`rqrrr#r#r$lazy_calendarss rvrocCrur)r'r`TIMEZONES_FOLDERrrr#r#r$ri cCrur)r'r` EVENTS_FOLDERrrr#r#r$rjrycCrur)r'r` ALARMS_FOLDERr rr#r#r$rkrycCrur)r'r`AVAILABILITIES_FOLDERr rr#r#r$rlrycCrur)r'r` TODOS_FOLDERrrr#r#r$rmry)rpcC|jSrrrrsrr#r#r$utcsrcCs|jt|dSNtzinfo)replacer gettzr r#r#r$r%scCs|jt|dSr)rrr r#r#r$r%r&cCr~rrrr#r#r$ in_timezonesr fuzz_testcase)zbig_bad_calendar.icszissue_104_broken_calendar.icsz!issue_1050_multiple_calendars.icszsmall_bad_calendar.icsz multiple_calendar_components.icszpr_480_summary_with_colon.icszparsing_error_in_UTC_offset.icszparsing_error.icscCs0g|]}|jtvr|jdvrt|jvr|jqS))r(r)r]BROKEN_SOURCE_FILESr2FUZZ_TESTCASES_BROKEN_CALENDARSr6filer#r#r$r9   r9cC0g|]}|jtvr|jdkrt|jvr|jqS)r(rrr#r#r$r9rcCr)r)rrr#r#r$r9rreturncCsD|j}td||||fD] }||vr||Sq td|d)Nz example file:zCould not find file r<)rrprint ValueError)rhrirjrsrBdatar#r#r$get_source_files  rcCt||||S)zqAn example file. source_file.raw_ics - bytes if .ics file source_file.raw_jcal - str if .jcal file rrrhrirjrsr#r#r$rB rtrBcCr)z[An example .ica file. source_file.raw_ics - bytes source_file.raw_jcal - None rrr#r#r$ics_filercCr)zZAn example .jcal file. source_file.raw_ics - None source_file.raw_jcal - str rrr#r#r$ jcal_filerrcCsg|] }tt|vr|qSr#)rstr)r6rUr#r#r$r9)s  cCr~)zClusterfuzz calendars.rrsr#r#r$fuzz_v1_calendar_path0rcCtSzReturn a new types factory.)rr#r#r#r$ types_factory6rrcCrrrr#r#r#r$component_factory<rrccs"d|jd<dV|jddS)zMap x_sometime to timetimez X-SOMETIMEN) types_mappop)rr#r#r$ x_sometimeBs rcCr)zReturn a new component factory.rr#r#r#r$factoryJrrccsdt_dVdt_dS)NTF)rignore_exceptionsr#r#r#r$vUTCOffset_ignore_exceptionsPs rcCt}d|_|S)zReturn an event component.VEVENTr r]rcr#r#r$event_componentWrcCr)Return an empty component.)r rr#r#r$r_rrcCr)r VCALENDARrrr#r#r$calendar_componenthrrccstj}dt_dV|t_dS)z*Remove validation for new() of components.FN)r _validate_new)valuer#r#r$dont_validate_newps  rcCs>tdd}d|_|jdddd|jdd dd|||S) zBReturn an event with some values and add it to calendar_component.zA brief history of time)summaryrdtend20000102T000000r)encodedtstart20000101T000000)r r]add add_component)rrer#r#r$filled_event_componentys  rcCst}d|d<|S)NzChair, Table, "Room: 42" resources)r rr#r#r$calendar_with_resourcessrccst|tVtdS)zThe timezone provider.N)_tzpuse use_defaulttzp_namer#r#r$rs  rzoneinfocCrX)zThis is annother timezone provider. The purpose here is to cross test: pytz <-> zoneinfo. tzp as parameter makes sure we test the cross product. )rrrrr#r#r$ other_tzps rcCs|sJ|S)z+Skip tests that are not running under pytz.) uses_pytz)rrr#r#r$ pytz_only rcCs|sJ|S)z/Skip tests that are not running under zoneinfo.) uses_zoneinfo)rrsrr#r#r$ zoneinfo_onlyrrcC|dksJ|S)zDo not run tests with pytz.rr#rr#r#r$no_pytzrrcCr)zDo not run tests with zoneinfo.rr#rr#r#r$ no_zoneinforrcCsd|jvrGtdg}d|jvrdg}d|jvrt}d|jvr'd|jvr'JddD]}d||jvr<||vr<||q)|jd|dd 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_rnrwN) fixturenamesPYTZ_TZPremove parametrize)metafunc tzp_namesr]r#r#r$pytest_generate_testss      rc@seZdZdZddZdS)DoctestZoneInfoz+Constent ZoneInfo representation for tests.cCsd|jdS)Nz ZoneInfo(key=))rUr8r#r#r$r[szDoctestZoneInfo.__repr__N)rarbrcrdr[r#r#r#r$rs rcGs$dd|D}tdd|DdS)z doctest printcSs$g|]}t|tr|dn|qS)zUTF-8) isinstancebytesdecoder6argr#r#r$r9s$z!doctest_print..css,|]}t|ddddVqdS)z   N)rstriprrr#r#r$ s*z doctest_print..N)r)argsr#r#r$ doctest_printsrcOs"|dkrtSt|g|Ri|S)zIReplace the import mechanism to skip the whole doctest if we import pytz.r)r __import__)r]rkwr#r#r$doctest_importsri rz$d755cef5-2311-46ed-a0e1-6733c9e15c63z#-//collective//icalendar//7.0.0//ENcCtS)z!The UID that tests always create.) UID_DEFAULTr#r#r#r$test_uidsrT)autousecs>tjtdd|tdfdd|tdtdddS) zYCreate a fixed test environment for the functions that are time and randomness dependent.)versionuuid4csSrr#r#uidr#r$r%zfixed_env.._utc_nowcSrr)NOW_UTCr#r#r#r$r%rN)uuidUUIDrrLr staticmethod) monkeypatchr#rr$ fixed_envsrcCs`|tjdt|tdtddlm}||dtdddl m }||dt d t iS) z,Modify the environment to make doctests run.rrr)ZONEINFOrr)calendarDEFAULT_PRODIDr) setitemsysmodulesrrLricalendar.timezone.zoneinforr icalendar.calrrr)rrrr#r#r$env_for_doctests  r rscCr~)zxReturn a timezone id to be used with pytz or zoneinfo. This goes through all the different timezones possible. rrr#r#r$tzid sr cCr~)z@Names of property types that occur as a property of a component.rrr#r#r$ v_prop_namerr cCs tt|S)z7Property types that occur as a property of a component.)rOr)r r#r#r$v_props r cCs |dS)Nr)examples)r r#r#r$v_prop_example"ryrcCr)zA mock.rr#r#r#r$mock'rr)vrdrf itertoolsrrrcollections.abcrrrpathlibr unittest.mockrrpytestdateutilr icalendarr r r r rrrrrrrricalendar.tests.dataricalendar.timezonerrrrr ImportErrorHAS_PYTZrPYTZ_UTCPYTZ_IN_TIMEZONErr'__file__parentHERErqrxrzr{r|r}fixturerhrvrirjrkrlrmrrrrrchainr: SOURCE_FILES ICS_FILES JCAL_FILESrrBrrFUZZ_V1rrrrrrrrcomprrrrrrrrrrrrrrNOWrrrrrrr TZIDSFixtureRequestr r r  VPROPERTYrrr#r#r#r$s8     8       V