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
74 lines
2.6 KiB
Python
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
|