mirror of
https://github.com/stripe/ai.git
synced 2026-09-14 18:39:59 +08:00
sgr-stripe
37 lines
1.1 KiB
Python
37 lines
1.1 KiB
Python
"""Tests for Configuration types."""
|
|
|
|
import unittest
|
|
from stripe_agent_toolkit.configuration import Configuration, Context
|
|
|
|
|
|
class TestConfiguration(unittest.TestCase):
|
|
"""Tests for Configuration type."""
|
|
|
|
def test_empty_configuration(self):
|
|
"""Empty configuration should be valid."""
|
|
config: Configuration = {}
|
|
self.assertEqual(config, {})
|
|
|
|
def test_configuration_with_context(self):
|
|
"""Configuration with context should be valid."""
|
|
config: Configuration = {
|
|
"context": {
|
|
"account": "acct_123",
|
|
"customer": "cus_456",
|
|
}
|
|
}
|
|
self.assertEqual(config["context"]["account"], "acct_123")
|
|
self.assertEqual(config["context"]["customer"], "cus_456")
|
|
|
|
def test_context_with_mode(self):
|
|
"""Context with mode should be valid."""
|
|
context: Context = {
|
|
"account": "acct_123",
|
|
"mode": "modelcontextprotocol",
|
|
}
|
|
self.assertEqual(context["mode"], "modelcontextprotocol")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|