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)