88 lines
4.0 KiB
Python
88 lines
4.0 KiB
Python
"""Integration: the Disk page's hosts side by side with their growth (#5304)
|
|
and the overview's disk line (#5306), from the totals ingest records as
|
|
metrics. Requires STEWARD_DATABASE_URL."""
|
|
from __future__ import annotations
|
|
|
|
import asyncio
|
|
import importlib
|
|
import os
|
|
import time
|
|
import uuid
|
|
from datetime import datetime, timedelta, timezone
|
|
|
|
import pytest
|
|
|
|
pytestmark = pytest.mark.integration
|
|
|
|
|
|
@pytest.fixture
|
|
def app():
|
|
if not os.environ.get("STEWARD_DATABASE_URL"):
|
|
pytest.skip("needs Postgres")
|
|
from steward.app import create_app
|
|
return create_app(testing=False)
|
|
|
|
|
|
def _docker(sub):
|
|
"""The app's loaded docker plugin module (see test_docker_recent_usage)."""
|
|
from steward.core.plugin_manager import get_loaded_plugin_modules
|
|
loaded = get_loaded_plugin_modules().get("docker")
|
|
base = loaded.__name__ if loaded is not None else "plugins.docker"
|
|
return importlib.import_module(f"{base}.{sub}")
|
|
|
|
|
|
def test_the_disk_page_says_how_much_each_host_grew(app):
|
|
from steward.models.hosts import Host
|
|
from steward.models.metrics import PluginMetric
|
|
from steward.models.users import User, UserRole
|
|
m = _docker("models")
|
|
disk_hosts = _docker("service").disk_hosts
|
|
|
|
tag = uuid.uuid4().hex[:8]
|
|
hid, uid = str(uuid.uuid4()), str(uuid.uuid4())
|
|
name = f"disk-{tag}"
|
|
now = datetime.now(timezone.utc)
|
|
gib = 1 << 30
|
|
|
|
async def go():
|
|
async with app.db_sessionmaker() as s:
|
|
async with s.begin():
|
|
s.add(User(id=uid, username=f"dp-{tag}", email=f"dp-{tag}@test",
|
|
password_hash="x", role=UserRole.viewer, is_active=True))
|
|
s.add(Host(id=hid, name=name, address="10.9.6.1"))
|
|
async with s.begin():
|
|
s.add(m.DockerDiskUsage(host_id=hid, images_size=30 * gib, containers_size=2 * gib,
|
|
volumes_size=10 * gib, build_cache_size=0, images_reclaimable=5 * gib,
|
|
images_total=12, images_active=8, updated_at=now))
|
|
s.add(m.DockerImage(host_id=hid, image_id="a" * 12, repo_tag=f"ghcr.io/x/app{tag}:1",
|
|
size=gib, shared_size=0, containers=1, updated_at=now))
|
|
s.add(m.DockerImage(host_id=hid, image_id="b" * 12, repo_tag=f"ghcr.io/x/app{tag}:0",
|
|
size=gib, shared_size=0, containers=0, updated_at=now))
|
|
s.add_all([PluginMetric(source_module="docker", resource_name=name, metric_name="disk_total_bytes",
|
|
value=float(v * gib), recorded_at=now - timedelta(days=d))
|
|
for d, v in ((6, 40), (3, 41), (0, 42))])
|
|
async with s.begin():
|
|
rows = await disk_hosts(s, now - timedelta(days=7))
|
|
|
|
client = app.test_client()
|
|
async with client.session_transaction() as sess:
|
|
sess.update(user_id=uid, user_role="viewer", username=f"dp-{tag}", login_at=time.time())
|
|
frag = await (await client.get("/plugins/docker/disk/hosts?range=7d")).get_data(as_text=True)
|
|
page = await (await client.get("/plugins/docker/disk")).get_data(as_text=True)
|
|
index = await (await client.get("/plugins/docker/")).get_data(as_text=True)
|
|
async with app.db_sessionmaker() as s:
|
|
async with s.begin():
|
|
await s.delete(await s.get(Host, hid))
|
|
await s.delete(await s.get(User, uid))
|
|
return rows, frag, page, index
|
|
|
|
rows, frag, page, index = asyncio.run(go())
|
|
(row,) = [r for r in rows if r["host_id"] == hid]
|
|
assert row["total"] == 42 * gib and row["reclaimable"] == 5 * gib
|
|
assert row["growth"]["change"] == 2 * gib and row["growth"]["from"] == 40 * gib
|
|
|
|
assert f'<a href="#host-{hid}">{name}</a>' in frag and "+2.0 GiB" in frag and "<polyline" in frag
|
|
assert f'id="host-{hid}"' in page and f"ghcr.io/x/app{tag}" in page and "(1 unused)" in page
|
|
assert 'hx-get="/plugins/docker/disk/hosts"' in page
|
|
assert f'href="/plugins/docker/disk#host-{hid}">{name}</a>' in index and "+2.0 GiB in 7d" in index
|