mirror of
https://github.com/microsoft/agent-framework.git
synced 2026-06-16 21:04:09 +08:00
Rishabh Chawla
64826b8f56
* [PythonPurview] Add Caching and background processing * [PythonPurview] Updates based on comments
46 lines
1.7 KiB
Python
46 lines
1.7 KiB
Python
# Copyright (c) Microsoft. All rights reserved.
|
|
|
|
"""Tests for Purview exceptions."""
|
|
|
|
from agent_framework_purview import (
|
|
PurviewAuthenticationError,
|
|
PurviewPaymentRequiredError,
|
|
PurviewRateLimitError,
|
|
PurviewRequestError,
|
|
PurviewServiceError,
|
|
)
|
|
|
|
|
|
class TestPurviewExceptions:
|
|
"""Test custom Purview exception classes."""
|
|
|
|
def test_purview_service_error(self) -> None:
|
|
"""Test PurviewServiceError base exception."""
|
|
error = PurviewServiceError("Service error occurred")
|
|
assert str(error) == "Service error occurred"
|
|
assert isinstance(error, Exception)
|
|
|
|
def test_purview_authentication_error(self) -> None:
|
|
"""Test PurviewAuthenticationError exception."""
|
|
error = PurviewAuthenticationError("Authentication failed")
|
|
assert str(error) == "Authentication failed"
|
|
assert isinstance(error, PurviewServiceError)
|
|
|
|
def test_purview_payment_required_error(self) -> None:
|
|
"""Test PurviewPaymentRequiredError exception."""
|
|
error = PurviewPaymentRequiredError("Payment required")
|
|
assert str(error) == "Payment required"
|
|
assert isinstance(error, PurviewServiceError)
|
|
|
|
def test_purview_rate_limit_error(self) -> None:
|
|
"""Test PurviewRateLimitError exception."""
|
|
error = PurviewRateLimitError("Rate limit exceeded")
|
|
assert str(error) == "Rate limit exceeded"
|
|
assert isinstance(error, PurviewServiceError)
|
|
|
|
def test_purview_request_error(self) -> None:
|
|
"""Test PurviewRequestError exception."""
|
|
error = PurviewRequestError("Request failed")
|
|
assert str(error) == "Request failed"
|
|
assert isinstance(error, PurviewServiceError)
|