def digitalbitbox_test_suite(simulator, rpc, userpass, interface):
# Start the Digital bitbox simulator
simulator_proc = subprocess.Popen(
['./' + os.path.basename(simulator), '../../tests/sd_files/'],
cwd=os.path.dirname(simulator),
stderr=subprocess.DEVNULL)
# Wait for simulator to be up
while True:
try:
dev = BitboxSimulator('127.0.0.1', 35345)
reply = send_plain(b'{"password":"******"}', dev)
if 'error' not in reply:
break
except:
pass
time.sleep(0.5)
# Cleanup
def cleanup_simulator():
simulator_proc.kill()
simulator_proc.wait()
atexit.register(cleanup_simulator)
# Set password and load from backup
send_encrypt(
json.dumps({
"seed": {
"source": "backup",
"filename": "test_backup.pdf",
"key": "key"
}
}), '0000', dev)
# params
type = 'digitalbitbox'
full_type = 'digitalbitbox'
path = 'udp:127.0.0.1:35345'
fingerprint = 'a31b978a'
master_xpub = 'xpub6BsWJiRvbzQJg3J6tgUKmHWYbHJSj41EjAAje6LuDwnYLqLiNSWK4N7rCXwiUmNJTBrKL8AEH3LBzhJdgdxoy4T9aMPLCWAa6eWKGCFjQhq'
# DigitalBitbox specific management command tests
class TestDBBManCommands(DeviceTestCase):
def test_restore(self):
result = self.do_command(self.dev_args + ['-i', 'restore'])
self.assertIn('error', result)
self.assertIn('code', result)
self.assertEqual(
result['error'],
'The Digital Bitbox does not support restoring via software')
self.assertEqual(result['code'], -9)
def test_pin(self):
result = self.do_command(self.dev_args + ['promptpin'])
self.assertIn('error', result)
self.assertIn('code', result)
self.assertEqual(
result['error'],
'The Digital Bitbox does not need a PIN sent from the host')
self.assertEqual(result['code'], -9)
result = self.do_command(self.dev_args + ['sendpin', '1234'])
self.assertIn('error', result)
self.assertIn('code', result)
self.assertEqual(
result['error'],
'The Digital Bitbox does not need a PIN sent from the host')
self.assertEqual(result['code'], -9)
def test_display(self):
result = self.do_command(self.dev_args +
['displayaddress', '--path', 'm/0h'])
self.assertIn('error', result)
self.assertIn('code', result)
self.assertEqual(
result['error'],
'The Digital Bitbox does not have a screen to display addresses on'
)
self.assertEqual(result['code'], -9)
def test_setup_wipe(self):
# Device is init, setup should fail
result = self.do_command(self.dev_args + [
'-i', 'setup', '--label', 'setup_test', '--backup_passphrase',
'testpass'
])
self.assertEquals(result['code'], -10)
self.assertEquals(
result['error'],
'Device is already initialized. Use wipe first and try again')
# Wipe
result = self.do_command(self.dev_args + ['wipe'])
self.assertTrue(result['success'])
# Check arguments
result = self.do_command(self.dev_args +
['-i', 'setup', '--label', 'setup_test'])
self.assertEquals(result['code'], -7)
self.assertEquals(
result['error'],
'The label and backup passphrase for a new Digital Bitbox wallet must be specified and cannot be empty'
)
result = self.do_command(
self.dev_args +
['-i', 'setup', '--backup_passphrase', 'testpass'])
self.assertEquals(result['code'], -7)
self.assertEquals(
result['error'],
'The label and backup passphrase for a new Digital Bitbox wallet must be specified and cannot be empty'
)
# Setup
result = self.do_command(self.dev_args + [
'-i', 'setup', '--label', 'setup_test', '--backup_passphrase',
'testpass'
])
self.assertTrue(result['success'])
# Reset back to original
result = self.do_command(self.dev_args + ['wipe'])
self.assertTrue(result['success'])
send_plain(b'{"password":"******"}', dev)
send_encrypt(
json.dumps({
"seed": {
"source": "backup",
"filename": "test_backup.pdf",
"key": "key"
}
}), '0000', dev)
# Make sure device is init, setup should fail
result = self.do_command(self.dev_args + [
'-i', 'setup', '--label', 'setup_test', '--backup_passphrase',
'testpass'
])
self.assertEquals(result['code'], -10)
self.assertEquals(
result['error'],
'Device is already initialized. Use wipe first and try again')
def test_backup(self):
# Check arguments
result = self.do_command(self.dev_args +
['backup', '--label', 'backup_test'])
self.assertEquals(result['code'], -7)
self.assertEquals(
result['error'],
'The label and backup passphrase for a Digital Bitbox backup must be specified and cannot be empty'
)
result = self.do_command(self.dev_args +
['backup', '--backup_passphrase', 'key'])
self.assertEquals(result['code'], -7)
self.assertEquals(
result['error'],
'The label and backup passphrase for a Digital Bitbox backup must be specified and cannot be empty'
)
# Wipe
result = self.do_command(self.dev_args + ['wipe'])
self.assertTrue(result['success'])
# Setup
result = self.do_command(self.dev_args + [
'-i', 'setup', '--label', 'backup_test', '--backup_passphrase',
'testpass'
])
self.assertTrue(result['success'])
# make the backup
result = self.do_command(self.dev_args + [
'backup', '--label', 'backup_test_backup',
'--backup_passphrase', 'testpass'
])
self.assertTrue(result['success'])
# Generic Device tests
suite = unittest.TestSuite()
suite.addTest(
DeviceTestCase.parameterize(TestDBBManCommands,
rpc,
userpass,
type,
full_type,
path,
fingerprint,
master_xpub,
'0000',
interface=interface))
suite.addTest(
DeviceTestCase.parameterize(TestDeviceConnect,
rpc,
userpass,
type,
full_type,
path,
fingerprint,
master_xpub,
'0000',
interface=interface))
suite.addTest(
DeviceTestCase.parameterize(TestDeviceConnect,
rpc,
userpass,
'digitalbitbox_01_simulator',
full_type,
path,
fingerprint,
master_xpub,
'0000',
interface=interface))
suite.addTest(
DeviceTestCase.parameterize(TestGetDescriptors,
rpc,
userpass,
type,
full_type,
path,
fingerprint,
master_xpub,
'0000',
interface=interface))
suite.addTest(
DeviceTestCase.parameterize(TestGetKeypool,
rpc,
userpass,
type,
full_type,
path,
fingerprint,
master_xpub,
'0000',
interface=interface))
suite.addTest(
DeviceTestCase.parameterize(TestSignTx,
rpc,
userpass,
type,
full_type,
path,
fingerprint,
master_xpub,
'0000',
interface=interface))
suite.addTest(
DeviceTestCase.parameterize(TestSignMessage,
rpc,
userpass,
type,
full_type,
path,
fingerprint,
master_xpub,
'0000',
interface=interface))
return suite
def test_setup_wipe(self):
# Device is init, setup should fail
result = self.do_command(self.dev_args + [
'-i', 'setup', '--label', 'setup_test', '--backup_passphrase',
'testpass'
])
self.assertEquals(result['code'], -10)
self.assertEquals(
result['error'],
'Device is already initialized. Use wipe first and try again')
# Wipe
result = self.do_command(self.dev_args + ['wipe'])
self.assertTrue(result['success'])
# Check arguments
result = self.do_command(self.dev_args +
['-i', 'setup', '--label', 'setup_test'])
self.assertEquals(result['code'], -7)
self.assertEquals(
result['error'],
'The label and backup passphrase for a new Digital Bitbox wallet must be specified and cannot be empty'
)
result = self.do_command(
self.dev_args +
['-i', 'setup', '--backup_passphrase', 'testpass'])
self.assertEquals(result['code'], -7)
self.assertEquals(
result['error'],
'The label and backup passphrase for a new Digital Bitbox wallet must be specified and cannot be empty'
)
# Setup
result = self.do_command(self.dev_args + [
'-i', 'setup', '--label', 'setup_test', '--backup_passphrase',
'testpass'
])
self.assertTrue(result['success'])
# Reset back to original
result = self.do_command(self.dev_args + ['wipe'])
self.assertTrue(result['success'])
send_plain(b'{"password":"******"}', dev)
send_encrypt(
json.dumps({
"seed": {
"source": "backup",
"filename": "test_backup.pdf",
"key": "key"
}
}), '0000', dev)
# Make sure device is init, setup should fail
result = self.do_command(self.dev_args + [
'-i', 'setup', '--label', 'setup_test', '--backup_passphrase',
'testpass'
])
self.assertEquals(result['code'], -10)
self.assertEquals(
result['error'],
'Device is already initialized. Use wipe first and try again')
def digitalbitbox_test_suite(simulator, rpc, userpass, interface):
try:
os.unlink('bitbox-emulator.stderr')
except FileNotFoundError:
pass
bitbox_log = open('bitbox-emulator.stderr', 'a')
# Start the Digital bitbox simulator
simulator_proc = subprocess.Popen(
['./' + os.path.basename(simulator), '../../tests/sd_files/'],
cwd=os.path.dirname(simulator),
stderr=bitbox_log)
# Wait for simulator to be up
while True:
try:
dev = BitboxSimulator('127.0.0.1', 35345)
reply = send_plain(b'{"password":"******"}', dev)
if 'error' not in reply:
break
except Exception:
pass
time.sleep(0.5)
# Cleanup
def cleanup_simulator():
simulator_proc.terminate()
simulator_proc.wait()
bitbox_log.close()
atexit.register(cleanup_simulator)
# Set password and load from backup
send_encrypt(
json.dumps({
"seed": {
"source": "backup",
"filename": "test_backup.pdf",
"key": "key"
}
}), '0000', dev)
# params
type = 'digitalbitbox'
full_type = 'digitalbitbox'
path = 'udp:127.0.0.1:35345'
fingerprint = 'a31b978a'
master_xpub = 'xpub6BsWJiRvbzQJg3J6tgUKmHWYbHJSj41EjAAje6LuDwnYLqLiNSWK4N7rCXwiUmNJTBrKL8AEH3LBzhJdgdxoy4T9aMPLCWAa6eWKGCFjQhq'
# DigitalBitbox specific management command tests
class TestDBBManCommands(DeviceTestCase):
def test_restore(self):
result = self.do_command(self.dev_args + ['-i', 'restore'])
self.assertIn('error', result)
self.assertIn('code', result)
self.assertEqual(
result['error'],
'The Digital Bitbox does not support restoring via software')
self.assertEqual(result['code'], -9)
def test_pin(self):
result = self.do_command(self.dev_args + ['promptpin'])
self.assertIn('error', result)
self.assertIn('code', result)
self.assertEqual(
result['error'],
'The Digital Bitbox does not need a PIN sent from the host')
self.assertEqual(result['code'], -9)
result = self.do_command(self.dev_args + ['sendpin', '1234'])
self.assertIn('error', result)
self.assertIn('code', result)
self.assertEqual(
result['error'],
'The Digital Bitbox does not need a PIN sent from the host')
self.assertEqual(result['code'], -9)
def test_display(self):
result = self.do_command(self.dev_args +
['displayaddress', '--path', 'm/0h'])
self.assertIn('error', result)
self.assertIn('code', result)
self.assertEqual(
result['error'],
'The Digital Bitbox does not have a screen to display addresses on'
)
self.assertEqual(result['code'], -9)
def test_setup_wipe(self):
# Device is init, setup should fail
result = self.do_command(self.dev_args + [
'-i', 'setup', '--label', 'setup_test', '--backup_passphrase',
'testpass'
])
self.assertEquals(result['code'], -10)
self.assertEquals(
result['error'],
'Device is already initialized. Use wipe first and try again')
# Wipe
result = self.do_command(self.dev_args + ['wipe'])
self.assertTrue(result['success'])
# Check arguments
result = self.do_command(self.dev_args +
['-i', 'setup', '--label', 'setup_test'])
self.assertEquals(result['code'], -7)
self.assertEquals(
result['error'],
'The label and backup passphrase for a new Digital Bitbox wallet must be specified and cannot be empty'
)
result = self.do_command(
self.dev_args +
['-i', 'setup', '--backup_passphrase', 'testpass'])
self.assertEquals(result['code'], -7)
self.assertEquals(
result['error'],
'The label and backup passphrase for a new Digital Bitbox wallet must be specified and cannot be empty'
)
# Setup
result = self.do_command(self.dev_args + [
'-i', 'setup', '--label', 'setup_test', '--backup_passphrase',
'testpass'
])
self.assertTrue(result['success'])
# Reset back to original
result = self.do_command(self.dev_args + ['wipe'])
self.assertTrue(result['success'])
send_plain(b'{"password":"******"}', dev)
send_encrypt(
json.dumps({
"seed": {
"source": "backup",
"filename": "test_backup.pdf",
"key": "key"
}
}), '0000', dev)
# Make sure device is init, setup should fail
result = self.do_command(self.dev_args + [
'-i', 'setup', '--label', 'setup_test', '--backup_passphrase',
'testpass'
])
self.assertEquals(result['code'], -10)
self.assertEquals(
result['error'],
'Device is already initialized. Use wipe first and try again')
def test_backup(self):
# Check arguments
result = self.do_command(self.dev_args +
['backup', '--label', 'backup_test'])
self.assertEquals(result['code'], -7)
self.assertEquals(
result['error'],
'The label and backup passphrase for a Digital Bitbox backup must be specified and cannot be empty'
)
result = self.do_command(self.dev_args +
['backup', '--backup_passphrase', 'key'])
self.assertEquals(result['code'], -7)
self.assertEquals(
result['error'],
'The label and backup passphrase for a Digital Bitbox backup must be specified and cannot be empty'
)
# Wipe
result = self.do_command(self.dev_args + ['wipe'])
self.assertTrue(result['success'])
# Setup
result = self.do_command(self.dev_args + [
'-i', 'setup', '--label', 'backup_test', '--backup_passphrase',
'testpass'
])
self.assertTrue(result['success'])
# make the backup
result = self.do_command(self.dev_args + [
'backup', '--label', 'backup_test_backup',
'--backup_passphrase', 'testpass'
])
self.assertTrue(result['success'])
class TestBitboxGetXpub(DeviceTestCase):
def setUp(self):
self.dev_args.remove('--chain')
self.dev_args.remove('test')
def test_getxpub(self):
result = self.do_command(self.dev_args +
['--expert', 'getxpub', 'm/44h/0h/0h/3'])
self.assertEqual(
result['xpub'],
'xpub6Du9e5Cz1NZWz3dvsvM21tsj4xEdbAb7AcbysFL42Y3yr8PLMnsaxhetHxurTpX5Rp5RbnFFwP1wct8K3gErCUSwcxFhxThsMBSxdmkhTNf'
)
self.assertFalse(result['testnet'])
self.assertFalse(result['private'])
self.assertEqual(result['depth'], 4)
self.assertEqual(result['parent_fingerprint'], '31d5e5ea')
self.assertEqual(result['child_num'], 3)
self.assertEqual(
result['chaincode'],
'7062818c752f878bf96ca668f77630452c3fa033b7415eed3ff568e04ada8104'
)
self.assertEqual(
result['pubkey'],
'029078c9ad8421afd958d7bc054a0952874923e2586fc9375604f0479a354ea193'
)
signtx_cases = [
(["legacy"], True, True, True),
(["segwit"], True, True, True),
(["legacy", "segwit"], True, True, True),
]
# Generic Device tests
suite = unittest.TestSuite()
suite.addTest(
DeviceTestCase.parameterize(TestDBBManCommands,
rpc,
userpass,
type,
full_type,
path,
fingerprint,
master_xpub,
'0000',
interface=interface))
suite.addTest(
DeviceTestCase.parameterize(TestBitboxGetXpub,
rpc,
userpass,
type,
full_type,
path,
fingerprint,
master_xpub,
'0000',
interface=interface))
suite.addTest(
DeviceTestCase.parameterize(TestDeviceConnect,
rpc,
userpass,
type,
full_type,
path,
fingerprint,
master_xpub,
'0000',
interface=interface))
suite.addTest(
DeviceTestCase.parameterize(TestDeviceConnect,
rpc,
userpass,
'digitalbitbox_01_simulator',
full_type,
path,
fingerprint,
master_xpub,
'0000',
interface=interface))
suite.addTest(
DeviceTestCase.parameterize(TestGetDescriptors,
rpc,
userpass,
type,
full_type,
path,
fingerprint,
master_xpub,
'0000',
interface=interface))
suite.addTest(
DeviceTestCase.parameterize(TestGetKeypool,
rpc,
userpass,
type,
full_type,
path,
fingerprint,
master_xpub,
'0000',
interface=interface))
suite.addTest(
DeviceTestCase.parameterize(TestSignTx,
rpc,
userpass,
type,
full_type,
path,
fingerprint,
master_xpub,
'0000',
interface=interface,
signtx_cases=signtx_cases))
result = unittest.TextTestRunner(stream=sys.stdout, verbosity=2).run(suite)
cleanup_simulator()
atexit.unregister(cleanup_simulator)
return result.wasSuccessful()