mirror of
https://github.com/stripe/ai.git
synced 2026-09-14 18:39:59 +08:00
59 lines
1.5 KiB
Python
59 lines
1.5 KiB
Python
|
import unittest
|
||
|
|
from stripe_agent_toolkit.configuration import is_tool_allowed
|
||
|
|
|
||
|
|
|
||
|
|
class TestConfigurations(unittest.TestCase):
|
||
|
|
def test_allowed(self):
|
||
|
|
tool = {
|
||
|
|
"actions": {
|
||
|
|
"customers": {"create": True, "read": True},
|
||
|
|
"invoices": {"create": True, "read": True},
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
configuration = {
|
||
|
|
"actions": {
|
||
|
|
"customers": {"create": True, "read": True},
|
||
|
|
"invoices": {"create": True, "read": True},
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
self.assertTrue(is_tool_allowed(tool, configuration))
|
||
|
|
|
||
|
|
def test_partial_allowed(self):
|
||
|
|
tool = {
|
||
|
|
"actions": {
|
||
|
|
"customers": {"create": True, "read": True},
|
||
|
|
"invoices": {"create": True, "read": True},
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
configuration = {
|
||
|
|
"actions": {
|
||
|
|
"customers": {"create": True, "read": True},
|
||
|
|
"invoices": {"create": True, "read": False},
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
self.assertFalse(is_tool_allowed(tool, configuration))
|
||
|
|
|
||
|
|
def test_not_allowed(self):
|
||
|
|
tool = {
|
||
|
|
"actions": {
|
||
|
|
"payment_links": {"create": True},
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
configuration = {
|
||
|
|
"actions": {
|
||
|
|
"customers": {"create": True, "read": True},
|
||
|
|
"invoices": {"create": True, "read": True},
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
self.assertFalse(is_tool_allowed(tool, configuration))
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
unittest.main()
|