Модульные тесты Python запускают функцию после всех тестов ⇐ Python

[Anonymous]

Мне нужно что-то протестировать на Python через ssh. Я не хочу делать ssh-соединение для каждого теста, потому что это слишком долго, я написал это:

Код: Выделить всё

class TestCase(unittest.TestCase):
client = None
def setUp(self):
if not hasattr(self.__class__, 'client') or self.__class__.client is None:
self.__class__.client = paramiko.SSHClient()
self.__class__.client.set_missing_host_key_policy(paramiko.AutoAddPolicy())
self.__class__.client.connect(hostname=consts.get_host(), port=consts.get_port(), username=consts.get_user(),
password=consts.get_password())

def test_a(self):
pass

def test_b(self):
pass

def test_c(self):
pass

def disconnect(self):
self.__class__.client.close()

и мой бегун

Код: Выделить всё

if __name__ == '__main__':
suite = unittest.TestSuite((
unittest.makeSuite(TestCase),
))
result = unittest.TextTestRunner().run(suite)
TestCase.disconnect()
sys.exit(not result.wasSuccessful())

В этой версии я получаю сообщение об ошибке TypeError: необходимо вызвать несвязанный метод Disconnect() с экземпляром TestCase в качестве первого аргумента (вместо этого ничего не получено). Итак, как я могу вызвать отключение после прохождения всех тестов?

Подробнее здесь: https://stackoverflow.com/questions/268 ... r-all-test