mirror of
https://github.com/langgenius/dify.git
synced 2026-05-28 21:03:22 +08:00
43 lines
1.2 KiB
Python
43 lines
1.2 KiB
Python
import inspect
|
|
|
|
from controllers.openapi.auth.conditions import Cond
|
|
from controllers.openapi.auth.data import AuthData, RequestContext
|
|
from controllers.openapi.auth.flow import When
|
|
from libs.oauth_bearer import TokenType
|
|
|
|
|
|
def _ctx():
|
|
return RequestContext(token_type=TokenType.OAUTH_ACCOUNT, path_params={})
|
|
|
|
|
|
def _data():
|
|
return AuthData(token_type=TokenType.OAUTH_ACCOUNT, token_hash="x", scopes=frozenset())
|
|
|
|
|
|
def test_applies_returns_true_when_condition_true():
|
|
w = When(Cond(lambda ctx, _: True), then=lambda b: None)
|
|
assert w.applies(_ctx()) is True
|
|
|
|
|
|
def test_applies_returns_false_when_condition_false():
|
|
w = When(Cond(lambda ctx, _: False), then=lambda b: None)
|
|
assert w.applies(_ctx()) is False
|
|
|
|
|
|
def test_applies_with_data():
|
|
w = When(Cond(lambda ctx, data: data is not None), then=lambda b: None)
|
|
assert w.applies(_ctx(), _data()) is True
|
|
assert w.applies(_ctx(), None) is False
|
|
|
|
|
|
def test_call_invokes_step():
|
|
calls = []
|
|
w = When(Cond(lambda ctx, _: True), then=lambda arg: calls.append(arg))
|
|
w("payload")
|
|
assert calls == ["payload"]
|
|
|
|
|
|
def test_then_is_keyword_only():
|
|
sig = inspect.signature(When.__init__)
|
|
assert sig.parameters["then"].kind.name == "KEYWORD_ONLY"
|