trezor: fix UT

2015-08-18 17:05:27 +03:00
parent dd93dea0a9
commit 8d45e42d10
1 changed files with 5 additions and 2 deletions
--- 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=[])