class TestCategoryCase(object):
def SetUp(self):
"""
Запускается в начале каждого теста,
создает пользователя для тестирования
"""
self.name, self.total_money = 'test Category', 1000000
self.category = Category(self.name, self.total_money)
print self.category.name
def test_add_money(self):
"""
Проверяет функцию пополнения баланса
"""
self.category.add_money(666)
print self.category.total_money
def test_move_money(self):
"""
Проверяет функцию перемещения денег
"""
#<----->
def run_all_test(self):
"""
Запускает все тесты
"""
self.SetUp()
self.test_add_money()
class TestCategoryCase(object):
def SetUp(self):
"""
Запускается в начале каждого теста,
создает пользователя для тестирования
"""
self.name, self.total_money = 'test Category', 1000000
self.category = Category(self.name, self.total_money)
def test_add_money(self):
"""
Проверяет функцию пополнения баланса
"""
self.SetUp()
before_balance = self.category.balance
self.category.add_money(666)
assert self.category.balance == before_balance + 666
def test_move_money_from_cat_to_cat(self):
self.SetUp()
test_category = Category("test Catogory 2", 500)
self.category.move_money(300000, test_category)
assert self.category.balance == 700000
assert test_category.balance == 300500
def test_move_money_from_cat_to_wallet(self):
self.SetUp()
test_wallet = Wallet("test Wallet", 500)
self.category.move_money(300000, test_wallet)
assert self.category.balance == 700000
assert test_wallet.balance == 300500
def run_all_test(self):
"""
Запускает все тесты
"""
self.test_add_money()
self.test_move_money_from_cat_to_cat()
self.test_move_money_from_cat_to_wallet()
