import paramiko

from gpu_rent.tunnel import _patch_paramiko_for_sshtunnel, _ssh_tunnel_forwarder


def test_dsskey_shim_allows_sshtunnel_import():
    # Simulate paramiko>=4 (no DSSKey) then ensure sshtunnel can load.
    had = getattr(paramiko, "DSSKey", None)
    if had is not None:
        delattr(paramiko, "DSSKey")
    try:
        _patch_paramiko_for_sshtunnel()
        assert hasattr(paramiko, "DSSKey")
        cls = _ssh_tunnel_forwarder()
        assert cls is not None
    finally:
        if had is not None:
            paramiko.DSSKey = had