Files
tracer-cloud--opensre/tests/tools/test_postgresql_server_status_tool.py
T
wehub-resource-sync 4b6817381b
CI (OpenClaw E2E) / openclaw test (push) Has been cancelled
CI / coverage-report (push) Has been cancelled
CI / test-kubernetes (push) Has been cancelled
CI / should-run-thorough (push) Has been cancelled
CI / test-thorough (cloudwatch-demo) (push) Has been cancelled
CI / test-thorough (flink-ecs) (push) Has been cancelled
CI / test-thorough (upstream-lambda) (push) Has been cancelled
CI / test-thorough (prefect-ecs-fargate) (push) Has been cancelled
Release / build-binaries (zip, opensre.exe, onefile, windows-latest, windows-x64) (push) Has been cancelled
Benchmark image — build + push to ECR (any adapter) / build + push (push) Has been cancelled
CI / quality (ubuntu-latest) (push) Has been cancelled
CI / test (tools-runtime) (push) Has been cancelled
CI / test (e2e-general) (push) Has been cancelled
CI / test (cli-runtime) (push) Has been cancelled
CI / test (e2e-provider-and-openclaw) (push) Has been cancelled
CI / test (integrations-and-misc) (push) Has been cancelled
Release / verify (push) Has been cancelled
Release / build-python-dist (push) Has been cancelled
Release / build-binaries (tar.gz, opensre, onedir, macos-15-intel, darwin-x64) (push) Has been cancelled
Release / build-binaries (tar.gz, opensre, onedir, macos-latest, darwin-arm64) (push) Has been cancelled
Release / build-binaries (tar.gz, opensre, onedir, ubuntu-22.04, linux-x64) (push) Has been cancelled
Release / publish-release (push) Has been cancelled
Release / publish-main-release (push) Has been cancelled
Interactive Shell Live (PR + post-merge) / turn-checks (no-LLM) (push) Has been cancelled
CodeQL / Analyze (python) (push) Has been cancelled
Interactive Shell Live (PR + post-merge) / turn-live shard ${{ matrix.shard_index }} (push) Has been cancelled
Release / prepare (push) Has been cancelled
Release / build-binaries (tar.gz, opensre, onedir, ubuntu-22.04-arm, linux-arm64) (push) Has been cancelled
Synthetic Deterministic Tests / Synthetic offline (deterministic) (push) Has been cancelled
chore: import upstream snapshot with attribution
2026-07-13 13:10:45 +08:00

60 lines
2.1 KiB
Python

"""Tests for PostgreSQLServerStatusTool (function-based, @tool decorated)."""
from __future__ import annotations
from unittest.mock import patch
from integrations.postgresql.tools.postgresql_server_status_tool import get_postgresql_server_status
from tests.tools.conftest import BaseToolContract
class TestPostgreSQLServerStatusToolContract(BaseToolContract):
def get_tool_under_test(self):
return get_postgresql_server_status.__opensre_registered_tool__
def test_metadata() -> None:
rt = get_postgresql_server_status.__opensre_registered_tool__
assert rt.name == "get_postgresql_server_status"
assert rt.source == "postgresql"
def test_run_happy_path() -> None:
fake_result = {
"source": "postgresql",
"available": True,
"version": "16.1",
"uptime": "5 days",
"connections": {"total": 25, "active": 15, "idle": 10, "max_connections": 100},
"database_stats": {
"backends": 25,
"transactions": {"committed": 12345, "rolled_back": 123},
"cache_hit_ratio_percent": 95.5,
"tuples": {
"returned": 987654,
"fetched": 543210,
"inserted": 1234,
"updated": 567,
"deleted": 89,
},
},
}
with patch(
"integrations.postgresql.tools.postgresql_server_status_tool.get_server_status",
return_value=fake_result,
):
result = get_postgresql_server_status(host="localhost", database="testdb")
assert result["version"] == "16.1"
assert result["connections"]["total"] == 25
assert result["database_stats"]["cache_hit_ratio_percent"] == 95.5
def test_run_error_propagated() -> None:
with patch(
"integrations.postgresql.tools.postgresql_server_status_tool.get_server_status",
return_value={"source": "postgresql", "available": False, "error": "connection timeout"},
):
result = get_postgresql_server_status(host="invalid", database="testdb")
assert "error" in result
assert result["available"] is False