mirror of
https://github.com/trezor/trezor-firmware.git
synced 2024-11-15 20:19:23 +00:00
40 lines
1.1 KiB
Python
40 lines
1.1 KiB
Python
import unittest
|
|
import common
|
|
|
|
from bitkeylib import proto
|
|
|
|
'''
|
|
TODO:
|
|
* Features reflects all variations of LoadDevice
|
|
* Maxfee settings
|
|
* Client requires OTP
|
|
* Client requires PIN
|
|
|
|
'''
|
|
|
|
class TestBasic(common.BitkeyTest):
|
|
def test_features(self):
|
|
features = self.bitkey.call(proto.Initialize(session_id=self.bitkey.session_id))
|
|
|
|
# Result is the same as reported by BitkeyClient class
|
|
self.assertEqual(features, self.bitkey.features)
|
|
|
|
def test_ping(self):
|
|
ping = self.bitkey.call(proto.Ping(message='ahoj!'))
|
|
|
|
# Ping results in Success(message='Ahoj!')
|
|
self.assertEqual(ping, proto.Success(message='ahoj!'))
|
|
|
|
def test_uuid(self):
|
|
uuid1 = self.bitkey.get_uuid()
|
|
self.bitkey.init_device()
|
|
uuid2 = self.bitkey.get_uuid()
|
|
|
|
# UUID must be longer than 10 characters
|
|
self.assertEqual(len(uuid1), 9)
|
|
|
|
# Every resulf of UUID must be the same
|
|
self.assertEqual(uuid1, uuid2)
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main() |