selenium pythonでカスタム関数を定義する

Question

カスタム関数を定義してそれをテスト関数で使用する方法は、単一のテストケースを実行するときに機能しますが、複数のテストケースを実行すると機能しません。

class AlphaTest(unittest.TestCase): 
     def setUp(self): 
      self.driver = webdriver.Firefox() 
      self.driver.implicitly_wait(30) 
      self.driver.get("http://google.com/") 

     def asserTrueId(self, value): 
      self.assertTrue(self.driver.find_element_by_id(value)) 
      time.sleep(1) 


     def test_flush_cache(self): 
      self.asserTrueId("block-menu-menu-menu-for-directories") 

Answer 1

あなたはクラスレベルdriverインスタンスをインスタンス化するためにunittest.setUpClass()を使用することができます。同様に、必要に応じてtearDownClass()を使用してクラスレベルの変数をクリーンアップすることもできます。

setUpClass()とtearDownClass()は、テストクラス内で1回だけ実行されるため、これを使用して1つのドライバしかないことを保証できます。それ以外の場合は、毎回ブラウザインスタンスを作成する各テストメソッドに対して、setUp()を使用して実行します。これは、速度が遅く、メモリが非効率である可能性があります。

class AlphaTest(unittest.TestCase): 

    @classmethod 
    def setUpClass(cls): 
     cls.driver = webdriver.Firefox() 
     cls.driver.implicitly_wait(30) 
     cls.driver.get("http://google.com/") 

    def asserTrueId(self, value): 
     self.assertTrue(self.driver.find_element_by_id(value)) 
     time.sleep(1) 

    def test_flush_cache(self): 
     self.asserTrueId("block-menu-menu-menu-for-directories")