ProPeler
Home
Console
Upload
information
Create File
Create Folder
About
Tools
:
/
opt
/
cpanel-ccs
/
contrib
/
performance
/
sqlusage
/
requests
/
Filename :
query.py
back
Copy
## # Copyright (c) 2012-2017 Apple Inc. All rights reserved. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. # You may obtain a copy of the License at # # http://www.apache.org/licenses/LICENSE-2.0 # # Unless required by applicable law or agreed to in writing, software # distributed under the License is distributed on an "AS IS" BASIS, # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. ## from caldavclientlibrary.protocol.url import URL from caldavclientlibrary.protocol.webdav.definitions import davxml, statuscodes from contrib.performance.sqlusage.requests.httpTests import HTTPTestBase from txweb2.dav.util import joinURL from pycalendar.datetime import DateTime from caldavclientlibrary.protocol.caldav.query import QueryVEVENTTimeRange from caldavclientlibrary.protocol.http.data.string import ResponseDataString ICAL = """BEGIN:VCALENDAR CALSCALE:GREGORIAN PRODID:-//Example Inc.//Example Calendar//EN VERSION:2.0 BEGIN:VTIMEZONE LAST-MODIFIED:20040110T032845Z TZID:US/Eastern BEGIN:DAYLIGHT DTSTART:20000404T020000 RRULE:FREQ=YEARLY;BYDAY=1SU;BYMONTH=4 TZNAME:EDT TZOFFSETFROM:-0500 TZOFFSETTO:-0400 END:DAYLIGHT BEGIN:STANDARD DTSTART:20001026T020000 RRULE:FREQ=YEARLY;BYDAY=-1SU;BYMONTH=10 TZNAME:EST TZOFFSETFROM:-0400 TZOFFSETTO:-0500 END:STANDARD END:VTIMEZONE BEGIN:VEVENT DTSTAMP:20051222T205953Z CREATED:20060101T150000Z DTSTART:%s DURATION:PT1H SUMMARY:event 1 UID:sync-collection-%d-ics END:VEVENT END:VCALENDAR """.replace("\n", "\r\n") class QueryTest(HTTPTestBase): """ A sync operation """ def __init__(self, label, sessions, logFilePath, logFilePrefix, count): super(QueryTest, self).__init__(label, sessions, logFilePath, logFilePrefix) self.count = count def prepare(self): """ Do some setup prior to the real request. """ # Add resources to create required number of changes self.start = DateTime.getNowUTC() self.start.setHHMMSS(12, 0, 0) self.end = self.start.duplicate() self.end.offsetHours(1) for i in range(self.count): href = joinURL(self.sessions[0].calendarHref, "tr-query-%d.ics" % (i + 1,)) self.sessions[0].writeData(URL(path=href), ICAL % (self.start.getText(), i + 1,), "text/calendar") def doRequest(self): """ Execute the actual HTTP request. """ props = ( davxml.getetag, davxml.getcontenttype, ) # Create CalDAV query request = QueryVEVENTTimeRange(self.sessions[0], self.sessions[0].calendarHref, self.start.getText(), self.end.getText(), props) result = ResponseDataString() request.setOutput(result) # Process it self.sessions[0].runSession(request) # If its a 207 we want to parse the XML if request.getStatusCode() == statuscodes.MultiStatus: pass else: raise RuntimeError("Query request failed: %s" % (request.getStatusCode(),)) def cleanup(self): """ Do some cleanup after the real request. """ # Remove created resources for i in range(self.count): href = joinURL(self.sessions[0].calendarHref, "tr-query-%d.ics" % (i + 1,)) self.sessions[0].deleteResource(URL(path=href))