pyssh-client/tests/test_session_dialog_bastion.py

"""Editeur de session : section Bastion / Jump Host (construction, pre-remplissage
a l'edition, et resultat renvoye pour chacun des 4 modes)."""
import pathlib
import sys

sys.path.insert(0, str(pathlib.Path(__file__).resolve().parent.parent))

import tempfile
from pathlib import Path
from unittest.mock import patch

from PySide6.QtWidgets import QApplication

from app.dialogs.session_dialog import SessionDialog
from app.key_manager import KeyManager
from app.session_manager import (
    AUTH_KEY,
    BASTION_MANUAL,
    BASTION_NONE,
    BASTION_OPENSSH,
    BASTION_SESSION,
    BastionConfig,
    Session,
    SessionStore,
)

app = QApplication.instance() or QApplication([])

tmp = Path(tempfile.mkdtemp())
key_manager = KeyManager(registry_path=tmp / "keys.json")
store = SessionStore(path=tmp / "sessions.json")
bastion_session = Session(name="bastion-srv", host="10.0.0.1", username="root")
store.add(bastion_session)
target_session = Session(name="cible", host="10.20.20.15", username="root")
store.add(target_session)

config_path = tmp / "sshconfig"
config_path.write_text("Host bastion-openssh\n    HostName 1.2.3.4\n    User jump\n", encoding="utf-8")

print("=== 1. Par defaut : mode Aucun ===")
dlg = SessionDialog(key_manager, session=Session(), session_store=store)
assert dlg.bastion_mode_combo.currentData() == BASTION_NONE
assert dlg._result_bastion().mode == BASTION_NONE
dlg.close()
print("  OK")

print()
print("=== 2. La session en cours d'edition n'apparait pas dans sa propre liste de bastions ===")
dlg = SessionDialog(key_manager, session=target_session, session_store=store)
ids_proposes = [dlg.bastion_session_combo.itemData(i) for i in range(dlg.bastion_session_combo.count())]
print("  sessions proposees comme bastion:", ids_proposes)
assert target_session.id not in ids_proposes
assert bastion_session.id in ids_proposes
dlg.close()
print("  OK")

print()
print("=== 3. Mode Session : selection puis resultat ===")
dlg = SessionDialog(key_manager, session=Session(), session_store=store)
idx = dlg.bastion_mode_combo.findData(BASTION_SESSION)
dlg.bastion_mode_combo.setCurrentIndex(idx)
idx_session = dlg.bastion_session_combo.findData(bastion_session.id)
dlg.bastion_session_combo.setCurrentIndex(idx_session)
result = dlg._result_bastion()
print("  resultat:", result)
assert result.mode == BASTION_SESSION
assert result.session_id == bastion_session.id
dlg.close()
print("  OK")

print()
print("=== 4. Mode OpenSSH : liste chargee depuis ~/.ssh/config, puis resultat ===")
with patch("app.openssh.config.default_config_path", return_value=config_path):
    dlg = SessionDialog(key_manager, session=Session(), session_store=store)
    idx = dlg.bastion_mode_combo.findData(BASTION_OPENSSH)
    dlg.bastion_mode_combo.setCurrentIndex(idx)
    idx_host = dlg.bastion_openssh_combo.findData("bastion-openssh")
    assert idx_host >= 0, "l'alias detecte doit apparaitre dans la liste"
    dlg.bastion_openssh_combo.setCurrentIndex(idx_host)
    result = dlg._result_bastion()
print("  resultat:", result)
assert result.mode == BASTION_OPENSSH
assert result.openssh_alias == "bastion-openssh"
dlg.close()
print("  OK")

print()
print("=== 5. Mode manuel : champs -> resultat ===")
dlg = SessionDialog(key_manager, session=Session(), session_store=store)
idx = dlg.bastion_mode_combo.findData(BASTION_MANUAL)
dlg.bastion_mode_combo.setCurrentIndex(idx)
dlg.bastion_host_edit.setText("bastion.example.com")
dlg.bastion_port_spin.setValue(2222)
dlg.bastion_user_edit.setText("jump")
result = dlg._result_bastion()
print("  resultat:", result)
assert result.mode == BASTION_MANUAL
assert result.host == "bastion.example.com"
assert result.port == 2222
assert result.username == "jump"
dlg.close()
print("  OK")

print()
print("=== 6. Edition d'une session avec bastion deja configure : pre-remplissage correct ===")
pre_filled = Session(
    name="avec-bastion", host="10.20.20.15", username="root",
    bastion=BastionConfig(mode=BASTION_MANUAL, host="old-bastion", port=2200, username="olduser"),
)
dlg = SessionDialog(key_manager, session=pre_filled, session_store=store)
print("  mode pre-selectionne:", dlg.bastion_mode_combo.currentData())
print("  page affichee (stack index):", dlg.bastion_stack.currentIndex())
assert dlg.bastion_mode_combo.currentData() == BASTION_MANUAL
assert dlg.bastion_stack.currentIndex() == dlg.bastion_mode_combo.currentIndex()
assert dlg.bastion_host_edit.text() == "old-bastion"
assert dlg.bastion_port_spin.value() == 2200
assert dlg.bastion_user_edit.text() == "olduser"
dlg.close()
print("  OK")

print()
print("=== 7. Cle bastion visible seulement en authentification par cle ===")
dlg = SessionDialog(key_manager, session=Session(), session_store=store)
idx = dlg.bastion_mode_combo.findData(BASTION_MANUAL)
dlg.bastion_mode_combo.setCurrentIndex(idx)
assert not dlg.bastion_key_combo.isEnabled()
idx_key = dlg.bastion_auth_combo.findData(AUTH_KEY)
dlg.bastion_auth_combo.setCurrentIndex(idx_key)
assert dlg.bastion_key_combo.isEnabled()
dlg.close()
print("  OK")

print()
print("SESSION DIALOG - BASTION : TOUS LES TESTS OK")