Extend new algorithm disabling to compressions

1 files changed, 5 insertions, 0 deletions

diff --git a/tests/test_transport.py b/tests/test_transport.py
index 28f30d63..e2174896 100644
--- a/tests/test_transport.py
+++ b/tests/test_transport.py
@@ -1139,8 +1139,12 @@ class AlgorithmDisablingTests(unittest.TestCase):
                 "macs": ["hmac-md5"],
                 "keys": ["ssh-dss"],
                 "kex": ["diffie-hellman-group14-sha256"],
+                "compression": ["zlib"],
             },
         )
+        # Enable compression cuz otherwise disabling one option for it makes no
+        # sense...
+        t.use_compression(True)
         # Effectively a random spot check, but kex init touches most/all of the
         # algorithm lists so it's a good spot.
         t._send_message = Mock()
@@ -1164,3 +1168,4 @@ class AlgorithmDisablingTests(unittest.TestCase):
         assert "hmac-md5" not in macs
         assert "ssh-dss" not in server_keys
         assert "diffie-hellman-group14-sha256" not in kexen
+        assert "zlib" not in compressions