3

我有一个使用 selenium IDE 导出到 python 的 selenium webdriver 测试用例。然后想添加一些额外的功能，所以在 selenium 测试用例之间插入了自定义代码。

现在的问题是 - 如果我犯了错误或自定义编写的 python 代码返回错误 - selenium 测试用例仍然继续执行

而如果出现此类错误，我希望停止执行 selenium 测试用例。通过简单的谷歌搜索复制了下面的场景，然后是使用shutil模块的日志移动代码

from selenium import webdriver
from selenium.webdriver.common.by import By
from selenium.webdriver.support.ui import Select
from selenium.common.exceptions import NoSuchElementException
import unittest, time, re
import os, shutil

class SeleniumException(unittest.TestCase):
    def setUp(self):
        self.driver = webdriver.Chrome()
        self.driver.implicitly_wait(30)
        self.base_url = "https://www.google.co.in/"
        self.verificationErrors = []

        self.attachment = 1
        self.file_source_location = "F:\\python_test\\logfiles\\"
        self.file_move_location = "F:\\logfiles_back\\"

    def test_selenium_exception(self):
        driver = self.driver
        driver.get(self.base_url + "/")
        driver.find_element_by_id("gbqfq").clear()
        driver.find_element_by_id("gbqfq").send_keys("Check this out")
        if self.attachment:
            try:
                for contents in os.listdir(self.file_source_location):
                    src_file = os.path.join(self.file_source_location, contents)
                    dst_file = os.path.join(self.file_move_location, contents)
                    shutil.move(src_file, dst_file)
                print'files_moved_success'
                driver.find_element_by_link_text("check out - definition of check out by the Free Online Dictionary ...").click()
            except Exception as e:
                print e

    def is_element_present(self, how, what):
        try: self.driver.find_element(by=how, value=what)
        except NoSuchElementException, e: return False
        return True

    def tearDown(self):
        self.driver.quit()
        self.assertEqual([], self.verificationErrors)

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

在此代码中 - 如果从 logfiles 到 logfiles_back 的移动由于路径错误等原因而失败...等 catch 块被执行，我希望 selenium 测试用例退出报告错误

现在发生的是它报告错误并完成测试用例的执行

如何使这成为可能？

4

1 回答 1

6

如果要引发实际触发的异常，可以raise自行调用，这将引发最后一个活动异常：

try:
    for contents in os.listdir(self.file_source_location):
        src_file = os.path.join(self.file_source_location, contents)
        dst_file = os.path.join(self.file_move_location, contents)
        shutil.move(src_file, dst_file)
    print'files_moved_success'
    driver.find_element_by_link_text("check out - definition of check out by the Free Online Dictionary ...").click()
except Exception as e:
    print e
    raise # Raise the exception that brought you here

如果您不想要回溯而只想退出，您也可以在sys.exit(1)之后调用（或您想要使用的任何错误代码）print e：

import sys
# ...
try:
    for contents in os.listdir(self.file_source_location):
        src_file = os.path.join(self.file_source_location, contents)
        dst_file = os.path.join(self.file_move_location, contents)
        shutil.move(src_file, dst_file)
    print'files_moved_success'
    driver.find_element_by_link_text("check out - definition of check out by the Free Online Dictionary ...").click()
except Exception as e:
    print e
    sys.exit(1)
于 2012-12-11T06:23:00.197 回答