Files
tracer-cloud--opensre/tests/tools/test_azure_sql_slow_queries_tool.py
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

74 lines
2.6 KiB
Python

"""Tests for AzureSQLSlowQueriesTool (function-based, @tool decorated)."""
from __future__ import annotations
from unittest.mock import patch
from integrations.azure_sql.tools.azure_sql_slow_queries_tool import get_azure_sql_slow_queries
from tests.tools.conftest import BaseToolContract
class TestAzureSQLSlowQueriesToolContract(BaseToolContract):
def get_tool_under_test(self):
return get_azure_sql_slow_queries.__opensre_registered_tool__
def test_metadata() -> None:
rt = get_azure_sql_slow_queries.__opensre_registered_tool__
assert rt.name == "get_azure_sql_slow_queries"
assert rt.source == "azure_sql"
def test_run_happy_path() -> None:
fake_result = {
"source": "azure_sql",
"available": True,
"threshold_ms": 1000,
"total_queries": 1,
"queries": [
{
"query_hash": "0xABCD",
"query_text": "SELECT * FROM large_table",
"execution_count": 100,
"avg_time_ms": 2500.0,
}
],
}
with patch(
"integrations.azure_sql.tools.azure_sql_slow_queries_tool.get_slow_queries",
return_value=fake_result,
):
result = get_azure_sql_slow_queries(
server="myserver.database.windows.net", database="testdb"
)
assert result["total_queries"] == 1
assert result["queries"][0]["avg_time_ms"] == 2500.0
def test_run_error_propagated() -> None:
with patch(
"integrations.azure_sql.tools.azure_sql_slow_queries_tool.get_slow_queries",
return_value={"source": "azure_sql", "available": False, "error": "timeout"},
):
result = get_azure_sql_slow_queries(server="invalid", database="testdb")
assert result["available"] is False
def test_default_db_warning_present_when_database_omitted() -> None:
with patch(
"integrations.azure_sql.tools.azure_sql_slow_queries_tool.get_slow_queries",
return_value={"source": "azure_sql", "available": True, "queries": []},
):
result = get_azure_sql_slow_queries(server="myserver.database.windows.net")
assert "default_db_warning" in result
assert "master" in result["default_db_warning"]
def test_no_default_db_warning_when_database_provided() -> None:
with patch(
"integrations.azure_sql.tools.azure_sql_slow_queries_tool.get_slow_queries",
return_value={"source": "azure_sql", "available": True, "queries": []},
):
result = get_azure_sql_slow_queries(server="myserver.database.windows.net", database="mydb")
assert "default_db_warning" not in result