trezor-firmware/core/tests/test_storage.py

from common import *  # isort:skip

from storage import device
from trezor import config


class TestConfig(unittest.TestCase):
    def test_counter(self):
        config.init()
        config.wipe()
        for i in range(150):
            self.assertEqual(device.next_u2f_counter(), i)
        device.set_u2f_counter(350)
        for i in range(351, 500):
            self.assertEqual(device.next_u2f_counter(), i)
        device.set_u2f_counter(0)
        self.assertEqual(device.next_u2f_counter(), 1)


if __name__ == "__main__":
    unittest.main()