def create_wallet():
if settings.wallets_testnet():
# attempt to create testnet wallet
logger.log("create Testnet wallet", "setup")
x = wallet_controller.create_wallet('TBTC')
if x:
settings.wallets_testnet_created("1")
settings.wallets_initiate_once("1")
settings.settings.write()
else:
# attempt to create bitcoin wallet
y = wallet_controller.create_wallet('BTC')
if y:
settings.wallets_initiate_once("1")
settings.settings.write()
def test_create_wallet_error(self):
self.popen = subprocess.Popen.communicate
subprocess.Popen.communicate = MagicMock(
side_effect=requests.ConnectionError)
self.assertFalse(walletcontroller.create_wallet('TBTC'))
subprocess.Popen.communicate = self.popen
def test_create_wallet_different_error(self):
self.popen = subprocess.Popen.communicate
self.json = json.loads
json.loads = MagicMock(return_value={'error': 'unspecified error'})
self.assertFalse(walletcontroller.create_wallet('TBTC'))
json.loads = self.json
subprocess.Popen.communicate = self.popen
def create_wallet():
"""
Checks if a Bitcoin (BTC) wallet or a Testnet Bitcoin (TBTC) wallet needs to be made.
"""
if settings.wallets_testnet():
# Attempt to create Testnet wallet
logger.log("create Testnet wallet")
x = wallet_controller.create_wallet('TBTC')
if x:
settings.wallets_testnet_created("1")
settings.wallets_initiate_once("1")
settings.settings.write()
os.environ['TESTNET'] = '1'
else:
# Attempt to create Bitcoin wallet
logger.log("create Bitcoin wallet")
y = wallet_controller.create_wallet('BTC')
if y:
settings.wallets_initiate_once("1")
settings.settings.write()
def test_create_wallet_unknown_error(self):
responses.add(responses.PUT,
'http://localhost:8085/wallets/BTC',
json={'error': 'unknown error'})
self.assertFalse(walletcontroller.create_wallet('BTC'))
def test_create_wallet_already_created(self):
responses.add(responses.PUT,
'http://localhost:8085/wallets/BTC',
json={'error': 'this wallet already exists'})
assert walletcontroller.create_wallet('BTC')
def test_create_wallet(self):
responses.add(responses.PUT,
'http://localhost:8085/wallets/BTC',
json={'created': 'true'})
assert walletcontroller.create_wallet('BTC')
def test_create_wallet_no_market(self):
marketcontroller.is_market_running = MagicMock(return_value=False)
self.assertFalse(walletcontroller.create_wallet('TBTC'))
def test_create_wallet_wrong_type(self):
self.assertFalse(walletcontroller.create_wallet('nonesense'))
def test_create_wallet_testnet_created(self):
plebnet_settings.Init.wallets_testnet_created = MagicMock(
return_value=True)
assert walletcontroller.create_wallet('TBTC')
