extended logs
This commit is contained in:
125
backend/tests/test_routeros_logging_and_files_filters.py
Normal file
125
backend/tests/test_routeros_logging_and_files_filters.py
Normal file
@@ -0,0 +1,125 @@
|
||||
from datetime import datetime
|
||||
|
||||
from app.db.session import SessionLocal
|
||||
from app.models.backup import Backup
|
||||
|
||||
from fastapi.testclient import TestClient
|
||||
|
||||
from app.main import app
|
||||
|
||||
|
||||
def _login(client: TestClient) -> dict[str, str]:
|
||||
response = client.post('/api/auth/login', data={'username': 'admin', 'password': 'admin'})
|
||||
token = response.json()['access_token']
|
||||
return {'Authorization': f'Bearer {token}'}
|
||||
|
||||
|
||||
def test_routeros_connection_test_creates_verbose_operation_log(monkeypatch):
|
||||
from app.services import router_service as router_service_module
|
||||
|
||||
monkeypatch.setattr(
|
||||
router_service_module.router_service,
|
||||
'probe_connection',
|
||||
lambda router, global_ssh_key=None, global_settings=None: {
|
||||
'success': True,
|
||||
'tested_at': datetime(2026, 4, 13, 10, 30, 0),
|
||||
'model': 'RB5009UG+S+',
|
||||
'uptime': '1d2h',
|
||||
'hostname': 'rb5009-core',
|
||||
'version': '7.18.2',
|
||||
'error': None,
|
||||
'transport': 'ssh',
|
||||
'server': None,
|
||||
'auth_mode': 'ssh',
|
||||
'http_status': None,
|
||||
'backup_available': None,
|
||||
},
|
||||
)
|
||||
|
||||
with TestClient(app) as client:
|
||||
headers = _login(client)
|
||||
create_response = client.post(
|
||||
'/api/routers',
|
||||
json={
|
||||
'name': 'core01',
|
||||
'device_type': 'routeros',
|
||||
'host': '10.10.10.1',
|
||||
'port': 2222,
|
||||
'ssh_user': 'backup',
|
||||
'ssh_password': 'secret',
|
||||
'ssh_key': None,
|
||||
},
|
||||
headers=headers,
|
||||
)
|
||||
assert create_response.status_code == 200
|
||||
device_id = create_response.json()['id']
|
||||
|
||||
response = client.get(f'/api/routers/{device_id}/test-connection', headers=headers)
|
||||
assert response.status_code == 200
|
||||
|
||||
logs_response = client.get('/api/logs', headers=headers)
|
||||
assert logs_response.status_code == 200
|
||||
assert any(
|
||||
'Connection test OK for RouterOS device core01' in item['message']
|
||||
and 'via ssh' in item['message']
|
||||
and 'target=10.10.10.1:2222' in item['message']
|
||||
and 'user=backup' in item['message']
|
||||
and 'hostname=rb5009-core' in item['message']
|
||||
and 'model=RB5009UG+S+' in item['message']
|
||||
and 'version=7.18.2' in item['message']
|
||||
and 'uptime=1d2h' in item['message']
|
||||
for item in logs_response.json()
|
||||
)
|
||||
|
||||
|
||||
def test_files_endpoint_filters_backups_by_created_on_date(monkeypatch):
|
||||
from app.services import router_service as router_service_module
|
||||
|
||||
monkeypatch.setattr(
|
||||
router_service_module.router_service,
|
||||
'export',
|
||||
lambda router, global_ssh_key=None: f'/system identity set name={router.name}',
|
||||
)
|
||||
|
||||
with TestClient(app) as client:
|
||||
headers = _login(client)
|
||||
create_response = client.post(
|
||||
'/api/routers',
|
||||
json={
|
||||
'name': 'archive01',
|
||||
'device_type': 'routeros',
|
||||
'host': '10.10.10.2',
|
||||
'port': 22,
|
||||
'ssh_user': 'admin',
|
||||
'ssh_password': 'secret',
|
||||
'ssh_key': None,
|
||||
},
|
||||
headers=headers,
|
||||
)
|
||||
assert create_response.status_code == 200
|
||||
device_id = create_response.json()['id']
|
||||
|
||||
first = client.post(f'/api/backups/router/{device_id}/export', headers=headers)
|
||||
second = client.post(f'/api/backups/router/{device_id}/export', headers=headers)
|
||||
assert first.status_code == 200
|
||||
assert second.status_code == 200
|
||||
|
||||
with SessionLocal() as db:
|
||||
first_backup = db.query(Backup).filter(Backup.id == first.json()['id']).first()
|
||||
second_backup = db.query(Backup).filter(Backup.id == second.json()['id']).first()
|
||||
first_backup.created_at = datetime(2026, 4, 12, 9, 15, 0)
|
||||
second_backup.created_at = datetime(2026, 4, 13, 11, 45, 0)
|
||||
db.add(first_backup)
|
||||
db.add(second_backup)
|
||||
db.commit()
|
||||
|
||||
filtered = client.get(f'/api/backups?router_id={device_id}&created_on=2026-04-13', headers=headers)
|
||||
assert filtered.status_code == 200
|
||||
payload = filtered.json()
|
||||
assert len(payload) == 1
|
||||
assert payload[0]['created_at'].startswith('2026-04-13T11:45:00')
|
||||
|
||||
previous_day = client.get(f'/api/backups?router_id={device_id}&created_on=2026-04-12', headers=headers)
|
||||
assert previous_day.status_code == 200
|
||||
assert len(previous_day.json()) == 1
|
||||
assert previous_day.json()[0]['created_at'].startswith('2026-04-12T09:15:00')
|
||||
Reference in New Issue
Block a user