def test_give_me_the_money_should_return_available_funds_from_all_accounts( self): # account_1 = Mock() # type(account_1).balance = PropertyMock(return_value=50) # account_2 = Mock() # type(account_2).balance = PropertyMock(return_value=100) account_1 = Mock() account_1.get_balance.return_value = 50 account_2 = Mock() account_2.get_balance.return_value = 100 user = User("Janusz", 40, [account_1, account_2]) self.assertEqual(user.give_me_the_money(), 150)
def test_give_me_the_money(self, accounts, expected_amount): user = User("Janusz", 40, accounts) self.assertEqual(user.give_me_the_money(), expected_amount)