[project @ robey@lag.net-20060309080450-bad95b03d60d3d4f]

improve HostKeys so that it more correctly emulates a dict, and add a unit test to verify that

1 files changed, 11 insertions, 0 deletions

diff --git a/tests/test_hostkeys.py b/tests/test_hostkeys.py
index 13426387..6f8eb57e 100644
--- a/tests/test_hostkeys.py
+++ b/tests/test_hostkeys.py
@@ -71,3 +71,14 @@ class HostKeysTest (unittest.TestCase):
         fp = paramiko.util.hexify(x['ssh-rsa'].get_fingerprint())
         self.assertEquals('7EC91BB336CB6D810B124B1353C32396', fp)
         self.assertTrue(hostdict.check('foo.example.com', key))
+
+    def test_3_dict(self):
+        hostdict = paramiko.HostKeys('hostfile.temp')
+        self.assert_('secure.example.com' in hostdict)
+        self.assert_('not.example.com' not in hostdict)
+        self.assert_(hostdict.has_key('secure.example.com'))
+        self.assert_(not hostdict.has_key('not.example.com'))
+        x = hostdict.get('secure.example.com', None)
+        self.assertTrue(x is not None)
+        fp = paramiko.util.hexify(x['ssh-rsa'].get_fingerprint())
+        self.assertEquals('E6684DB30E109B67B70FF1DC5C7F1363', fp)