def test_get_balance_error(self):
self.requests = requests.get
requests.get = MagicMock(side_effect=requests.ConnectionError)
self.assertEquals(walletcontroller.get_balance('BTC'),
"No %s wallet found" % 'BTC')
requests.get = self.requests
def test_get_balance(self):
responses.add(
responses.GET,
'http://localhost:8085/wallets',
json={'wallets': {
'BTC': {
'balance': {
'available': 5000
}
}
}})
self.assertEqual(walletcontroller.get_balance('BTC'), 5000)