o .jD@sdZddlmZddlZddlmZmZmZddlm Z er$ddlm Z deddfd d Z deddfd d Z Gd ddZ ejddZejddZddZddZddZddZddZejdgddd Zd!d"Zd#d$Zd%d&Zd'd(Zd)d*d+d,d-d.Zejd/d0d1gejd2gd3ejd4d0d1gd5d6Zd7d8Zd9d:Z d;d<Z!ejd=gd>ejd4d0d1gd?d@Z"dAdBZ#dCdDZ$dS)EzcTest lazy parsing of calendar components. See https://github.com/collective/icalendar/issues/1050 ) TYPE_CHECKINGN)AlarmEvent LazyCalendarLazySubcomponent)CalendarcalendarreturncCs,t|tsJ|jsJ|sJdS)z,Assert that the calendar is a lazy calendar.N isinstancer_subcomponentsis_lazyr ra/home/thesage/.local/lib/python3.10/site-packages/icalendar/tests/test_issue_1050_lazy_parsing.pyassert_is_lazy_calendarrcCs,t|tsJ|jrJ|rJdS)z.Assert that the calendar is a parsed calendar.Nr rrrrassert_is_parsed_calendarrrc@seZdZddZddZdS) MockParsercCs||_d|_dS)Nr) _component_calls)self componentrrr__init__ s zMockParser.__init__cCs|jd7_|jgS)zReturn the component.)rr)rrrrparse$szMockParser.parseN)__name__ __module__ __qualname__rrrrrrrs rcCstdS)z-A mock parser that returns a fixed component.mock_component)rrrrr mock_parser*sr!cCs td|S)z.A lazy subcomponent that uses the mock parser.VEVENTr)r!rrrlazy_subcomponent0s r#cCs,|rJ|j|us J|jdksJdS)z&The component is initially not parsed.rN) is_parsed_parserr)r!r#rrr1test_lazy_subcomponent_is_not_parsed_when_created6s r&cCs2|}|dks J|jdksJ|sJdS)z!We retrieve the component inside.r rN)rrr$r#r!rrrr(test_lazy_subcomponent_returns_component=s r(cCs.|}|}||usJ|jdksJdS)z)Parsing twice returns the same component.rN)rr)r#r! component1 component2rrr)test_parsing_twice_returns_same_componentEs r+cCs&|}|dks J|jdusJdS)z8Test that we can remove references to the content lines.r N)rr%r'rrr test_when_parsed_parser_is_freedMs r,cCst|jtsJdS)z7LazyCalendar.from_ical returns a LazyCalendar instance.N)r emptyr)lazy_calendarsrrr%test_lazy_calendar_returns_a_calendarTsr/component_name)r"VTODOVALARMz X-COMPONENT VTIMEZONEcCs4td|d|}|j|ksJ|rJdS)zLLazyCalendar.from_ical() returns components as Calendar and as Component do.zBEGIN:z END:N)r from_icalnamer)r0rrrr1test_non_calendar_components_are_parsed_correctlyYsr6cCs|j}t|dS)z,Test that the BigCalendar is initially lazy.N)examplerr.r rrr#test_big_calendar_is_initially_lazycs r9cCs|j}|tt|dS)z4Test that we can add a component to a lazy calendar.N)r- add_componentrrr8rrr*test_adding_a_component_to_a_lazy_calendaris  r;cCs8|j}|j}t|dksJt||jj|ksJdS)z:Test that we can get the subcomponents of a lazy calendar.N)r7 subcomponentslenr)r. calendarsr r=rrr1test_getting_the_subcomponents_of_a_lazy_calendarps r@cCs*|j}|j}|j}||usJt|dS)zCTest that parsing a lazy calendar twice does not change the result.N)r7r=r)r.r subcomponents1subcomponents2rrr"test_parsing_a_lazy_calendar_twiceys   rC timezoneseventstodosjournalsavailabilities)r3r"r1VJOURNAL VAVAILABILITYwalkTF)r5parsed))r3r3)r"r3r")r1r3r1)rIr3rI)rJr3rJ fully_parsedcCsv|j}|r|j|r||nt|t|}t|dks Jd|dj|ks)Jt|tr7|s9t ||dSdSdS)zCheck that only timezones are parsed. Skipping FREEBUSY for now is ok as they are for requests and need the whole calendar anyway. rz8We only have one component of each type in the calendar.rN) issue_1050_all_componentsr=rKgetattr name2getterr>r5r rassert_only_parsed)rKr?r5rLrRr componentsrrrtest_accessing_timezonessrXcCsT|sJddd|jjD}|t|ks(Jdd|dd|ddS) z6Check that only certain types of components are parsedz(The calendar should not be fully parsed.cSsh|] }|r|jqSr)r$r5).0lcrrr s z%assert_only_parsed..zWe expect only z, z to be parsed, not .N)rr _componentssetjoin)r rLparsed_in_calendarrrrrVsrVcCs.|j}|}t|dksJ|rJdS)zWalk goes through them all. N)rSrKr>rr?r rWrrr#test_walk_parses_the_whole_calendarsrccCs6|jd}t|dksJ|d}|jdksJdS)z%Walking and finding the alarm inside.r2rrN)rSrKr>r5)r?alarmsalarmrrrtest_walk_and_find_an_alarms rf)uidrL))zevent-1rN)z calendar-1rM)ztodo-1rO)z journal-1rP)zavailability-1rQ)zalarm-1rNcCs`|j}|r|j||}t|dksJ|dj|ksJt|tr,|s.t||dSdSdS)Nrr)rSr=with_uidr>rgr rrV)r?rgrLrRr rWrrrtest_get_with_uids  ricCs$|j}|d}t|dksJdS)Nznon-existent-uidrrSrhr>rbrrrtest_absent_uids rkcCs |jd}t|dksJdS)zCheck that even if the UID is present somewhere the component is not returned. See https://github.com/collective/icalendar/pull/1201/changes#r2869287074 rerNrj)r?rWrrr/test_uid_is_present_but_not_in_the_uid_propertys rl)%__doc__typingrpytest icalendarrrricalendar.parser.ical.lazyrrrrrfixturer!r#r&r(r+r,r/mark parametrizer6r9r;r@rCrUrXrVrcrfrirkrlrrrrsf