trezor-firmware/trezorlib/tests/device_tests/test_msg_clearsession.py

# This file is part of the Trezor project.
#
# Copyright (C) 2012-2018 SatoshiLabs and contributors
#
# This library is free software: you can redistribute it and/or modify
# it under the terms of the GNU Lesser General Public License version 3
# as published by the Free Software Foundation.
#
# This library is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU Lesser General Public License for more details.
#
# You should have received a copy of the License along with this library.
# If not, see <https://www.gnu.org/licenses/lgpl-3.0.html>.

import pytest

from .common import TrezorTest

from trezorlib import messages as proto


@pytest.mark.skip_t2
class TestMsgClearsession(TrezorTest):

    def test_clearsession(self):
        self.setup_mnemonic_pin_passphrase()

        with self.client:
            self.client.set_expected_responses([proto.ButtonRequest(code=proto.ButtonRequestType.ProtectCall), proto.PinMatrixRequest(), proto.PassphraseRequest(), proto.Success()])
            res = self.client.ping('random data', button_protection=True, pin_protection=True, passphrase_protection=True)
            assert res == 'random data'

        with self.client:
            # pin and passphrase are cached
            self.client.set_expected_responses([proto.ButtonRequest(code=proto.ButtonRequestType.ProtectCall), proto.Success()])
            res = self.client.ping('random data', button_protection=True, pin_protection=True, passphrase_protection=True)
            assert res == 'random data'

        self.client.clear_session()

        # session cache is cleared
        with self.client:
            self.client.set_expected_responses([proto.ButtonRequest(code=proto.ButtonRequestType.ProtectCall), proto.PinMatrixRequest(), proto.PassphraseRequest(), proto.Success()])
            res = self.client.ping('random data', button_protection=True, pin_protection=True, passphrase_protection=True)
            assert res == 'random data'

        with self.client:
            # pin and passphrase are cached
            self.client.set_expected_responses([proto.ButtonRequest(code=proto.ButtonRequestType.ProtectCall), proto.Success()])
            res = self.client.ping('random data', button_protection=True, pin_protection=True, passphrase_protection=True)
            assert res == 'random data'
regenerate license headers This clarifies the intent: the project is licenced under terms of LGPL version 3 only, but the standard headers cover only "3 or later", so we had to rewrite them. In the same step, we removed author information from individual files in favor of "SatoshiLabs and contributors", and include an AUTHORS file that lists the contributors. Apologies to those whose names are missing; please contact us if you wish to add your info to the AUTHORS file. 2018-06-21 14:28:34 +00:00			`# This file is part of the Trezor project.`
add license headers to unit tests 2017-01-03 18:40:05 +00:00			`#`
regenerate license headers This clarifies the intent: the project is licenced under terms of LGPL version 3 only, but the standard headers cover only "3 or later", so we had to rewrite them. In the same step, we removed author information from individual files in favor of "SatoshiLabs and contributors", and include an AUTHORS file that lists the contributors. Apologies to those whose names are missing; please contact us if you wish to add your info to the AUTHORS file. 2018-06-21 14:28:34 +00:00			`# Copyright (C) 2012-2018 SatoshiLabs and contributors`
add license headers to unit tests 2017-01-03 18:40:05 +00:00			`#`
			`# This library is free software: you can redistribute it and/or modify`
regenerate license headers This clarifies the intent: the project is licenced under terms of LGPL version 3 only, but the standard headers cover only "3 or later", so we had to rewrite them. In the same step, we removed author information from individual files in favor of "SatoshiLabs and contributors", and include an AUTHORS file that lists the contributors. Apologies to those whose names are missing; please contact us if you wish to add your info to the AUTHORS file. 2018-06-21 14:28:34 +00:00			`# it under the terms of the GNU Lesser General Public License version 3`
			`# as published by the Free Software Foundation.`
add license headers to unit tests 2017-01-03 18:40:05 +00:00			`#`
			`# This library is distributed in the hope that it will be useful,`
			`# but WITHOUT ANY WARRANTY; without even the implied warranty of`
			`# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the`
			`# GNU Lesser General Public License for more details.`
			`#`
regenerate license headers This clarifies the intent: the project is licenced under terms of LGPL version 3 only, but the standard headers cover only "3 or later", so we had to rewrite them. In the same step, we removed author information from individual files in favor of "SatoshiLabs and contributors", and include an AUTHORS file that lists the contributors. Apologies to those whose names are missing; please contact us if you wish to add your info to the AUTHORS file. 2018-06-21 14:28:34 +00:00			`# You should have received a copy of the License along with this library.`
			`# If not, see <https://www.gnu.org/licenses/lgpl-3.0.html>.`

tests/device_tests: remove all star imports 2018-05-11 12:53:51 +00:00			`import pytest`
add license headers to unit tests 2017-01-03 18:40:05 +00:00
tests/device_tests: remove all star imports 2018-05-11 12:53:51 +00:00			`from .common import TrezorTest`
add ClearSession message with unit test 2014-06-17 13:31:10 +00:00
Rework from Google's protobuf to pure-python protobuf implementation 2017-12-12 15:40:11 +00:00			`from trezorlib import messages as proto`
add ClearSession message with unit test 2014-06-17 13:31:10 +00:00
style: use flake8 2017-06-23 19:31:42 +00:00
device_tests: use skip_t1 and skip_t2 markers 2017-12-19 18:24:18 +00:00			`@pytest.mark.skip_t2`
device tests: simplify, drop unittest dependency 2017-12-23 20:20:49 +00:00			`class TestMsgClearsession(TrezorTest):`
add ClearSession message with unit test 2014-06-17 13:31:10 +00:00
			`def test_clearsession(self):`
			`self.setup_mnemonic_pin_passphrase()`

			`with self.client:`
Rework from Google's protobuf to pure-python protobuf implementation 2017-12-12 15:40:11 +00:00			`self.client.set_expected_responses([proto.ButtonRequest(code=proto.ButtonRequestType.ProtectCall), proto.PinMatrixRequest(), proto.PassphraseRequest(), proto.Success()])`
add ClearSession message with unit test 2014-06-17 13:31:10 +00:00			`res = self.client.ping('random data', button_protection=True, pin_protection=True, passphrase_protection=True)`
device tests: simplify, drop unittest dependency 2017-12-23 20:20:49 +00:00			`assert res == 'random data'`
add ClearSession message with unit test 2014-06-17 13:31:10 +00:00
			`with self.client:`
			`# pin and passphrase are cached`
Rework from Google's protobuf to pure-python protobuf implementation 2017-12-12 15:40:11 +00:00			`self.client.set_expected_responses([proto.ButtonRequest(code=proto.ButtonRequestType.ProtectCall), proto.Success()])`
add ClearSession message with unit test 2014-06-17 13:31:10 +00:00			`res = self.client.ping('random data', button_protection=True, pin_protection=True, passphrase_protection=True)`
device tests: simplify, drop unittest dependency 2017-12-23 20:20:49 +00:00			`assert res == 'random data'`
add ClearSession message with unit test 2014-06-17 13:31:10 +00:00
			`self.client.clear_session()`

			`# session cache is cleared`
			`with self.client:`
Rework from Google's protobuf to pure-python protobuf implementation 2017-12-12 15:40:11 +00:00			`self.client.set_expected_responses([proto.ButtonRequest(code=proto.ButtonRequestType.ProtectCall), proto.PinMatrixRequest(), proto.PassphraseRequest(), proto.Success()])`
add ClearSession message with unit test 2014-06-17 13:31:10 +00:00			`res = self.client.ping('random data', button_protection=True, pin_protection=True, passphrase_protection=True)`
device tests: simplify, drop unittest dependency 2017-12-23 20:20:49 +00:00			`assert res == 'random data'`
add ClearSession message with unit test 2014-06-17 13:31:10 +00:00
			`with self.client:`
			`# pin and passphrase are cached`
Rework from Google's protobuf to pure-python protobuf implementation 2017-12-12 15:40:11 +00:00			`self.client.set_expected_responses([proto.ButtonRequest(code=proto.ButtonRequestType.ProtectCall), proto.Success()])`
add ClearSession message with unit test 2014-06-17 13:31:10 +00:00			`res = self.client.ping('random data', button_protection=True, pin_protection=True, passphrase_protection=True)`
device tests: simplify, drop unittest dependency 2017-12-23 20:20:49 +00:00			`assert res == 'random data'`