From 5d510c4a60558d3f6c06dce0f2af18a870e245a1 Mon Sep 17 00:00:00 2001
From: Roman Zeyde <roman.zeyde@gmail.com>
Date: Tue, 18 Aug 2015 21:30:58 +0300
Subject: [PATCH] trezor: add ping for self-test

---
 sshagent/tests/test_trezor.py | 4 ++++
 sshagent/trezor/client.py     | 2 ++
 2 files changed, 6 insertions(+)

diff --git a/sshagent/tests/test_trezor.py b/sshagent/tests/test_trezor.py
index 2464e65..ecf0323 100644
--- a/sshagent/tests/test_trezor.py
+++ b/sshagent/tests/test_trezor.py
@@ -42,6 +42,10 @@ class ConnectionMock(object):
         result.node.public_key = PUBKEY
         return result
 
+    def ping(self, msg):
+        assert not self.closed
+        return msg
+
 
 class FactoryMock(object):
 
diff --git a/sshagent/trezor/client.py b/sshagent/trezor/client.py
index 50dd501..589e6e6 100644
--- a/sshagent/trezor/client.py
+++ b/sshagent/trezor/client.py
@@ -25,6 +25,8 @@ class Client(object):
         log.debug('revision : %s', binascii.hexlify(f.revision))
 
     def __enter__(self):
+        msg = 'Hello World!'
+        assert self.client.ping(msg) == msg
         return self
 
     def __exit__(self, *args):