25 lines
698 B
Python
25 lines
698 B
Python
|
"""Standalone RDP host model helpers."""
|
||
|
|
|
||
|
|
from server.domain.models import RdpHost
|
||
|
|
from server.infrastructure.database.crypto import decrypt_value, encrypt_value
|
||
|
|
from server.utils.rdp_config import normalize_rdp_port
|
||
|
|
|
||
|
|
|
||
|
|
def test_rdp_host_model_fields():
|
||
|
|
row = RdpHost(
|
||
|
|
name="win-bj",
|
||
|
|
host="10.0.0.5",
|
||
|
|
port=3389,
|
||
|
|
username="Administrator",
|
||
|
|
password=encrypt_value("Secret"),
|
||
|
|
)
|
||
|
|
assert row.name == "win-bj"
|
||
|
|
assert row.port == 3389
|
||
|
|
assert decrypt_value(row.password) == "Secret"
|
||
|
|
|
||
|
|
|
||
|
|
def test_normalize_rdp_port():
|
||
|
|
assert normalize_rdp_port(None) == 3389
|
||
|
|
assert normalize_rdp_port(3390) == 3390
|
||
|
|
assert normalize_rdp_port(99999) == 3389
|