tests/web_driver.py

   1 import unittest
   2
   3 from selenium.common.exceptions import TimeoutException
   4
   5 from manga_py.base_classes.web_driver import make_driver, get_display
   6
   7
   8 class TestWebDriver(unittest.TestCase):
   9     def test_driver(self):
  10         driver = make_driver()
  11         self.assertIsNotNone(driver)
  12         driver.get('https://ya.ru')
  13         result = True
  14         try:
  15             driver.find_element('#text,.logo-wrapper')
  16         except TimeoutException:
  17             result = False
  18         driver.close()
  19         get_display() is None or get_display().stop()
  20         self.assertTrue(result)