Python 模拟Selenium响应以测试UI控制器

我使用Selenium接口来驱动UI，以便为当前仅为web UI的东西提供编程API。我想为这个基于selenium的应用程序编写单元测试。如何模拟对原始Selenium API的调用，以验证应用程序是否正确

以下是Selenium应用程序的一个最低示例：

class MyClass(object):
    def run_app(self):
        self.driver = webdriver.Firefox()
        self.driver.get("http://www.google.com")
        assert "Google" in driver.title
        self.driver.close()
        return driver.title

def main():
    test_class = MyClass()
    results = test_class.run_app()

if __name__ == '__main__':
    main()