def step_NavigateRootFolder(self):
with timed.measure("%s, is navigating to root folder."%(self.username)) as t:
try:
r = self.browser.click_link(text=self.ssInfo[0][self.ssHeader.index('FOLDER')])
s = self.browser.open(r)
cc.traffRec(wai(), len(s.read()))
except: cc.errorRec("Couldn't find root folder in search results.")
cc.rec(wai(), t)
def step_NavigateDocumentProfile(self):
line = random.choice(self.ssInfo)
doc = line[self.ssHeader.index('DOCUMENT NAME')]
with timed.measure('%s, is navigating to %s profile page.'%(self.username, doc)) as t:
try:
r = self.browser.click_link(text=doc)
s = self.browser.open(r)
cc.traffRec(wai(), len(s.rread()))
except: cc.errorRec("%s couldn't find expected document."%(self.username))
cc.rec(wai(), t)
def step_DownloadDocument(self):
with timed.measure('%s, is downloading document.'%(self.username)) as t:
try:
#determine file extension listed on the current page
fileExtension = re.search('File extension:</span> <b>.(\w+)</b>', self.browser.response().read()).groups()[0]
r = self.browser.click_link(text='Download')
self.browser.open(r)
a = self.browser.response().read()
cc.traffRec(wai(), len(a))
with open('%s\\%s.%s'%(tempFilesFolder, str(uuid.uuid4()), fileExtension), 'wb') as f:
f.write(a)
except: cc.errorRec("%s couldn't find download link in doc profile."%(self.username))
cc.rec(wai(), t)
