blob42/trezor-agent
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

client: fixup UT

Browse Source
This commit is contained in:
Roman Zeyde 2016-03-12 20:57:16 +02:00
parent 2cdbc89d28
commit 0e2a19f7ce
1 changed files with 9 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
trezor_agent/tests/test_client.py
View File

@ -1,6 +1,7 @@
import io import io
import mock import mock
import pytest
from .. import client, factory, formats, util from .. import client, factory, formats, util
@ -112,6 +113,14 @@ def test_ssh_agent():
assert s[:1] == b'\x00' assert s[:1] == b'\x00'
assert r[1:] + s[1:] == SIG[1:] assert r[1:] + s[1:] == SIG[1:]
c.client.call_exception = ValueError
def cancel_sign_identity(identity, challenge_hidden,
challenge_visual, ecdsa_curve_name):
raise c.client.call_exception(42, 'ERROR')
c.client.sign_identity = cancel_sign_identity
with pytest.raises(IOError):
c.sign_ssh_challenge(label=label, blob=BLOB, visual='VISUAL')
def test_utils(): def test_utils():
identity = mock.Mock(spec=[]) identity = mock.Mock(spec=[])