feat: RDG 302/303 display path labels and RDS session break (0.3.1)
Show RDS access via RDG-Comp or Haproxy-RDG-Comp, enable qwinsta/logoff on any RDG event with internal_ip. Co-authored-by: Cursor <cursoragent@cursor.com>
This commit is contained in:
@@ -0,0 +1,127 @@
|
||||
"""Tests for RDG event display (access path, title/summary)."""
|
||||
|
||||
from datetime import datetime, timezone
|
||||
|
||||
import pytest
|
||||
|
||||
from app.models import Event, Host
|
||||
from app.services.event_summary import event_to_summary
|
||||
from app.services.rdg_display import (
|
||||
ACCESS_PATH_DIRECT,
|
||||
ACCESS_PATH_HAPROXY,
|
||||
build_rdg_display,
|
||||
classify_rdg_access_path,
|
||||
event_supports_rdg_client_qwinsta,
|
||||
)
|
||||
|
||||
|
||||
def _rdg_event(
|
||||
db_session,
|
||||
*,
|
||||
gw: Host,
|
||||
event_type: str = "rdg.connection.success",
|
||||
external_ip: str = "10.0.0.5",
|
||||
internal_ip: str = "192.168.160.3",
|
||||
win_id: int = 302,
|
||||
) -> Event:
|
||||
event = Event(
|
||||
event_id=f"ev-rdg-{win_id}",
|
||||
host_id=gw.id,
|
||||
occurred_at=datetime.now(timezone.utc),
|
||||
received_at=datetime.now(timezone.utc),
|
||||
category="auth",
|
||||
type=event_type,
|
||||
severity="info",
|
||||
title="RD Gateway event 302",
|
||||
summary="",
|
||||
payload={},
|
||||
details={
|
||||
"user": r"B26\papatramp",
|
||||
"external_ip": external_ip,
|
||||
"internal_ip": internal_ip,
|
||||
"event_id_windows": win_id,
|
||||
},
|
||||
)
|
||||
db_session.add(event)
|
||||
db_session.commit()
|
||||
db_session.refresh(event)
|
||||
return event
|
||||
|
||||
|
||||
def test_classify_rdg_access_path_direct(monkeypatch):
|
||||
monkeypatch.setenv("SAC_RDG_HAPROXY_EXTERNAL_IPS", "192.168.160.50")
|
||||
from app.config import get_settings
|
||||
|
||||
get_settings.cache_clear()
|
||||
assert classify_rdg_access_path("10.0.0.5") == ACCESS_PATH_DIRECT
|
||||
|
||||
|
||||
def test_classify_rdg_access_path_haproxy(monkeypatch):
|
||||
monkeypatch.setenv("SAC_RDG_HAPROXY_EXTERNAL_IPS", "192.168.160.50,10.0.0.100")
|
||||
from app.config import get_settings
|
||||
|
||||
get_settings.cache_clear()
|
||||
assert classify_rdg_access_path("10.0.0.100") == ACCESS_PATH_HAPROXY
|
||||
|
||||
|
||||
def test_build_rdg_display_title_and_path(db_session, monkeypatch):
|
||||
monkeypatch.setenv("SAC_RDG_HAPROXY_EXTERNAL_IPS", "")
|
||||
from app.config import get_settings
|
||||
|
||||
get_settings.cache_clear()
|
||||
|
||||
gw = Host(hostname="K6A-DC3", os_family="windows", product="rdp-login-monitor", ipv4="192.168.160.40")
|
||||
ws = Host(hostname="WS-PC", os_family="windows", product="rdp-login-monitor", ipv4="192.168.160.3")
|
||||
db_session.add_all([gw, ws])
|
||||
db_session.commit()
|
||||
|
||||
event = _rdg_event(db_session, gw=gw)
|
||||
display = build_rdg_display(event, db_session)
|
||||
assert display is not None
|
||||
assert ACCESS_PATH_DIRECT in display.title
|
||||
assert "WS-PC" in display.title
|
||||
assert "192.168.160.3" in display.title
|
||||
assert display.access_path == ACCESS_PATH_DIRECT
|
||||
assert display.qwinsta_enabled is True
|
||||
assert "papatramp" in display.summary
|
||||
assert "шлюз K6A-DC3" in display.summary
|
||||
|
||||
|
||||
def test_event_to_summary_enriches_rdg(db_session, monkeypatch):
|
||||
monkeypatch.setenv("SAC_RDG_HAPROXY_EXTERNAL_IPS", "10.0.0.5")
|
||||
from app.config import get_settings
|
||||
|
||||
get_settings.cache_clear()
|
||||
|
||||
gw = Host(hostname="K6A-DC3", os_family="windows", product="rdp-login-monitor", ipv4="192.168.160.40")
|
||||
db_session.add(gw)
|
||||
db_session.commit()
|
||||
|
||||
event = _rdg_event(db_session, gw=gw, external_ip="10.0.0.5")
|
||||
summary = event_to_summary(event, db_session)
|
||||
assert summary.rdg_access_path == ACCESS_PATH_HAPROXY
|
||||
assert summary.rdg_qwinsta_enabled is True
|
||||
assert "RDS подключение" in summary.title
|
||||
assert ACCESS_PATH_HAPROXY in summary.title
|
||||
|
||||
|
||||
def test_event_supports_rdg_client_qwinsta_without_flap(db_session):
|
||||
gw = Host(hostname="K6A-DC3", os_family="windows", product="rdp-login-monitor", ipv4="192.168.160.40")
|
||||
db_session.add(gw)
|
||||
db_session.commit()
|
||||
event = _rdg_event(db_session, gw=gw)
|
||||
assert event_supports_rdg_client_qwinsta(event) is True
|
||||
|
||||
|
||||
@pytest.mark.parametrize(
|
||||
"internal_ip",
|
||||
["", None],
|
||||
)
|
||||
def test_event_supports_rdg_client_qwinsta_requires_internal_ip(db_session, internal_ip):
|
||||
gw = Host(hostname="K6A-DC3", os_family="windows", product="rdp-login-monitor", ipv4="192.168.160.40")
|
||||
db_session.add(gw)
|
||||
db_session.commit()
|
||||
event = _rdg_event(db_session, gw=gw, internal_ip=internal_ip or "")
|
||||
if internal_ip is None:
|
||||
event.details = {k: v for k, v in event.details.items() if k != "internal_ip"}
|
||||
assert event_supports_rdg_client_qwinsta(event) is False
|
||||
Reference in New Issue
Block a user