From 8d45e42d107afbb347d0f2aab6a73bd882bb7966 Mon Sep 17 00:00:00 2001
From: Roman Zeyde <roman.zeyde@gmail.com>
Date: Tue, 18 Aug 2015 17:05:27 +0300
Subject: [PATCH] trezor: fix UT

---
 sshagent/tests/test_trezor.py | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/sshagent/tests/test_trezor.py b/sshagent/tests/test_trezor.py
index 8d5d252..202a7b5 100644
--- a/sshagent/tests/test_trezor.py
+++ b/sshagent/tests/test_trezor.py
@@ -29,9 +29,12 @@ class ConnectionMock(object):
     def close(self):
         self.closed = True
 
-    def get_public_node(self, addr, ecdsa_curve_name):
+    def clear_session(self):
+        self.closed = True
+
+    def get_public_node(self, n, ecdsa_curve_name):
         assert not self.closed
-        assert addr == ADDR
+        assert n == ADDR
         assert ecdsa_curve_name == CURVE
         result = mock.Mock(spec=[])
         result.node = mock.Mock(spec=[])