mirror of
https://github.com/trezor/trezor-firmware.git
synced 2024-11-26 17:38:39 +00:00
41 lines
1.1 KiB
Python
41 lines
1.1 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_uuid(self):
|
|
uuid1 = self.client.get_device_id()
|
|
self.client.init_device()
|
|
uuid2 = self.client.get_device_id()
|
|
|
|
# UUID must be longer than 10 characters
|
|
self.assertEqual(len(uuid1), 12)
|
|
|
|
# Every resulf of UUID must be the same
|
|
self.assertEqual(uuid1, uuid2)
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|