python - unitest setUp tearDown в разных контекстах

Question

Поддержка Deo python setUp() и tearDown() действует по-разному в зависимости от контекста? Все средства, я спрашиваю о том, могу ли я сделать что-то вроде этого:

setUp() { 
    if(context1){ 
     do A; 
    } 
    else{ 
     do B; 
    } 
} 


tearDown() { 
    if(context1){ 
     do A; 
    } 
    else{ 
     do B; 
    } 
} 

Answer 1

Да, и так же, как вы его показываете: используйте, если блоки, и делайте только определенные части установки, если условие истинно.

Я думаю, что у вас есть разные версии setUp и tearDown для различных тестов. Я бы на самом деле предположить, что вы либо:

• разделить тесты на различные TestCase подклассов с соответствующими setUp/tearDown методов

• или не используют setUp и tearDown на всех - сделать что-то вроде этого

  class MyTestCase: 
      def _setup_for_foo_tests(): 
       # blah blah blah 
      def _setup_for_bar_tests(): 
       # blah blah blah 
      def test_foo_1(): 
       self._setup_for_foo_tests() 
       # test code 
      def test_foo_2(): 
       self._setup_for_foo_tests() 
       # test code 
      def test_bar_1(): 
       self._setup_for_bar_tests() 
       # test code 
      # etc etc etc 
  

Answer 2

Вы должны думать о делать 2 разных классы (возможно, с некоторым общим предком) тест для каждого контекста теста вам нужно, это было бы проще.

Что-то вроде этого:

class BaseTest(): 
    def test_01a(self): 
     pass 

class Context1TestCase(BaseTest, unittest.TestCase): 
    def setUp(self): 
     # do what you need for context1 

    def tearDown(self): 
     # do what you need for context1 

class Context2TestCase(BaseTest, unittest.TestCase): 
    def setUp(self): 
     # do what you need for context2 

    def tearDown(self): 
     # do what you need for context2 

таким образом, test_01a будет выполняться один раз в context1, один раз в context2.