88 lines
3.2 KiB
Python
88 lines
3.2 KiB
Python
import json
|
|
import pytest
|
|
from fastapi.testclient import TestClient
|
|
from decnet.web.api import app
|
|
import decnet.config
|
|
from pathlib import Path
|
|
from decnet.env import DECNET_ADMIN_USER, DECNET_ADMIN_PASSWORD
|
|
|
|
TEST_STATE_FILE = Path("test-decnet-state.json")
|
|
|
|
@pytest.fixture(autouse=True)
|
|
def patch_state_file(monkeypatch):
|
|
# Patch the global STATE_FILE variable in the config module
|
|
monkeypatch.setattr(decnet.config, "STATE_FILE", TEST_STATE_FILE)
|
|
|
|
@pytest.fixture
|
|
def mock_state_file():
|
|
# Create a dummy state file for testing
|
|
_test_state = {
|
|
"config": {
|
|
"mode": "unihost",
|
|
"interface": "eth0",
|
|
"subnet": "192.168.1.0/24",
|
|
"gateway": "192.168.1.1",
|
|
"deckies": [
|
|
{
|
|
"name": "test-decky-1",
|
|
"ip": "192.168.1.10",
|
|
"services": ["ssh"],
|
|
"distro": "debian",
|
|
"base_image": "debian",
|
|
"hostname": "test-host-1",
|
|
"service_config": {"ssh": {"banner": "SSH-2.0-OpenSSH_8.9"}},
|
|
"archetype": "deaddeck",
|
|
"nmap_os": "linux",
|
|
"build_base": "debian:bookworm-slim"
|
|
},
|
|
{
|
|
"name": "test-decky-2",
|
|
"ip": "192.168.1.11",
|
|
"services": ["http"],
|
|
"distro": "ubuntu",
|
|
"base_image": "ubuntu",
|
|
"hostname": "test-host-2",
|
|
"service_config": {},
|
|
"archetype": None,
|
|
"nmap_os": "linux",
|
|
"build_base": "debian:bookworm-slim"
|
|
}
|
|
],
|
|
"log_target": None,
|
|
"log_file": "test.log",
|
|
"ipvlan": False
|
|
},
|
|
"compose_path": "test-compose.yml"
|
|
}
|
|
TEST_STATE_FILE.write_text(json.dumps(_test_state))
|
|
|
|
yield _test_state
|
|
|
|
# Cleanup
|
|
if TEST_STATE_FILE.exists():
|
|
TEST_STATE_FILE.unlink()
|
|
|
|
def test_get_deckies_endpoint(mock_state_file):
|
|
with TestClient(app) as _client:
|
|
# Login to get token
|
|
_login_resp = _client.post("/api/v1/auth/login", json={"username": DECNET_ADMIN_USER, "password": DECNET_ADMIN_PASSWORD})
|
|
_token = _login_resp.json()["access_token"]
|
|
|
|
_response = _client.get("/api/v1/deckies", headers={"Authorization": f"Bearer {_token}"})
|
|
assert _response.status_code == 200
|
|
_data = _response.json()
|
|
assert len(_data) == 2
|
|
assert _data[0]["name"] == "test-decky-1"
|
|
assert _data[0]["service_config"]["ssh"]["banner"] == "SSH-2.0-OpenSSH_8.9"
|
|
|
|
def test_stats_includes_deployed_count(mock_state_file):
|
|
with TestClient(app) as _client:
|
|
_login_resp = _client.post("/api/v1/auth/login", json={"username": DECNET_ADMIN_USER, "password": DECNET_ADMIN_PASSWORD})
|
|
_token = _login_resp.json()["access_token"]
|
|
|
|
_response = _client.get("/api/v1/stats", headers={"Authorization": f"Bearer {_token}"})
|
|
assert _response.status_code == 200
|
|
_data = _response.json()
|
|
assert "deployed_deckies" in _data
|
|
assert _data["deployed_deckies"] == 2
|