2016-08-02 112 views
1

我是硒的新手,我想在弹出警报时采取警报框的截图。 我写的代码如下:无法在使用python的硒中截取警报框截图

import unittest 
from selenium import webdriver 
from selenium.webdriver.common.keys import Keys 
from selenium.common.exceptions import NoAlertPresentException 
import time 


class SearchXSS(unittest.TestCase): 

def setUp(self): 
    self.driver = webdriver.Firefox() 

def test_search_in_python_org(self): 
    driver = self.driver 
    driver.get("http://testfire.net/") 
    ele = driver.find_element_by_name("txtSearch") 
    ele.send_keys("<script>alert(document.cookie)</script>") 
    ele.send_keys(Keys.RETURN) 
    try: 
     time.sleep(1) 
     alert = driver.switch_to_alert() 
     driver.get_screenshot_as_file('screenshot.png') 
     alert.accept() 
    except NoAlertPresentException as e: 
     print "no alert to accept " 
    fo.close() 


def tearDown(self): 
    self.driver.quit() 

if __name__=="__main__": 
    unittest.main() 

我面临的问题,同时采取screenshot.The错误信息如下

====================================================================== ERROR: test_search_in_python_org (main.SearchXSS) ---------------------------------------------------------------------- Traceback (most recent call last): File "/home/user/programs/sele/testx.py", line 22, in test_search_in_python_org driver.get_screenshot_as_file('screenshot.png') File "/usr/local/lib/python2.7/dist-packages/selenium/webdriver/remote/webdriver.py", line 798, in get_screenshot_as_file png = self.get_screenshot_as_png() File "/usr/local/lib/python2.7/dist-packages/selenium/webdriver/remote/webdriver.py", line 817, in get_screenshot_as_png return base64.b64decode(self.get_screenshot_as_base64().encode('ascii'))
File "/usr/local/lib/python2.7/dist-packages/selenium/webdriver/remote/webdriver.py", line 827, in get_screenshot_as_base64 return self.execute(Command.SCREENSHOT)['value'] File "/usr/local/lib/python2.7/dist-packages/selenium/webdriver/remote/webdriver.py", line 236, in execute self.error_handler.check_response(response) File "/usr/local/lib/python2.7/dist-packages/selenium/webdriver/remote/errorhandler.py", line 191, in check_response raise exception_class(message, screen, stacktrace, value['alert'].get('text')) UnexpectedAlertPresentException: Alert Text: amSessionId=92632533494 Message: Unexpected modal dialog (text: amSessionId=92632533494) Stacktrace: at fxdriver.modals.closeUnhandledAlert/< (file:///tmp/tmpdnEJbt/extensions/[email protected]/components/prompt-service.js:4745) at (http://testfire.net/search.aspx?txtSearch=%3Cscript%3Ealert%28document.cookie%29%3C%2Fscript%3E:80)

---------------------------------------------------------------------- Ran 1 test in 10.346s

FAILED (errors=1) [Finished in 10.4s with exit code 1]

帮我出这个问题。

回答

1

你不能把screenshotalert中,您需要处理alert第一装置acceptdismiss然后去,因为当硒去采取截图与警报目前采取的截图,它总是抛出UnexpectedAlertPresentException因为你得到, 使用selenium不可能使用alert框进行屏幕截图。所以,你需要如下操作: -

alert = driver.switch_to_alert() 
alert.accept() 
driver.get_screenshot_as_file('screenshot.png') 

如果你想采取截图与alert,你应该尝试一些其他的图书馆借了一些不同的方法,在java中有Robot类礼物能够采取screenshot在这种类型的场景,但我不确定什么是在python等效。

+0

谢谢Saurabh,但我想采取'alert'框的截图是这样的意图.. – user3513995

+0

@ user3513995看看是否有任何其他库提供的python取得屏幕截图..而在Java中有一个“机器人”类在这种情况下截取屏幕截图可能是相当于python的这个东西... :) –