mirror of
https://github.com/trezor/trezor-firmware.git
synced 2024-12-15 19:08:07 +00:00
49 lines
1.3 KiB
Python
49 lines
1.3 KiB
Python
import unittest
|
|
import common
|
|
|
|
from trezorlib import messages_pb2 as messages
|
|
|
|
'''
|
|
TODO:
|
|
* Features reflects all variations of LoadDevice
|
|
* Maxfee settings
|
|
* Client requires OTP
|
|
* Client requires PIN
|
|
|
|
'''
|
|
|
|
class TestBasic(common.TrezorTest):
|
|
def test_features(self):
|
|
features = self.client.call(messages.Initialize())
|
|
|
|
# Result is the same as reported by BitkeyClient class
|
|
self.assertEqual(features, self.client.features)
|
|
|
|
def test_ping(self):
|
|
ping = self.client.call(messages.Ping(message='ahoj!'))
|
|
|
|
# Ping results in Success(message='Ahoj!')
|
|
self.assertEqual(ping, messages.Success(message='ahoj!'))
|
|
|
|
def test_device_id_same(self):
|
|
id1 = self.client.get_device_id()
|
|
self.client.init_device()
|
|
id2 = self.client.get_device_id()
|
|
|
|
# ID must be at least 12 characters
|
|
self.assertTrue(len(id1) >= 12)
|
|
|
|
# Every resulf of UUID must be the same
|
|
self.assertEqual(id1, id2)
|
|
|
|
def test_device_id_different(self):
|
|
id1 = self.client.get_device_id()
|
|
self.client.wipe_device()
|
|
id2 = self.client.get_device_id()
|
|
|
|
# Device ID must be fresh after every reset
|
|
self.assertNotEqual(id1, id2)
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|