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 <[email protected]>
mistio · Nov 29, 2021 · 15fa5b8 · 15fa5b8
1 parent b79892d
commit 15fa5b8
Showing 1 changed file with 17 additions and 0 deletions.
diff --git a/tests/test_hostkeys.py b/tests/test_hostkeys.py
@@ -130,3 +130,20 @@ def test_delitem(self):
             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!"