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)