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