From 15fa5b8d2a7750b2ead67580343bf96ff93f6840 Mon Sep 17 00:00:00 2001 From: Lans Zhang Date: Wed, 26 Jul 2017 09:54:41 +0800 Subject: test_hostkeys: test SubDict.__delitem__() SubDict.__delitem__() would trigger the KeyError exception if a key type to be deleted is not present. Signed-off-by: Lans Zhang --- tests/test_hostkeys.py | 17 +++++++++++++++++ 1 file changed, 17 insertions(+) (limited to 'tests') diff --git a/tests/test_hostkeys.py b/tests/test_hostkeys.py index da47362c..e375687c 100644 --- a/tests/test_hostkeys.py +++ b/tests/test_hostkeys.py @@ -130,3 +130,20 @@ class HostKeysTest(unittest.TestCase): pass # Good else: assert False, "Entry was not deleted from HostKeys on delitem!" + + def test_entry_delitem(self): + hostdict = paramiko.HostKeys('hostfile.temp') + target = 'happy.example.com' + entry = hostdict[target] + key_type_list = [ key_type for key_type in entry ] + for key_type in key_type_list: + del entry[key_type] + + # will KeyError if not present + for key_type in key_type_list: + try: + del entry[key_type] + except KeyError: + pass # Good + else: + assert False, "Key was not deleted from Entry on delitem!" -- cgit v1.2.3