format code as per rules

This commit is contained in:
Pankaj Kaushal 2025-10-31 22:59:31 +05:30
parent b848224c78
commit ef1392dc57
2 changed files with 6 additions and 19 deletions

View File

@ -5,7 +5,6 @@ import pytest
from configs import dify_config
from core.app.entities.app_invoke_entities import InvokeFrom
from core.helper import encrypter
from core.variables import SecretVariable
from core.workflow.entities import GraphInitParams, GraphRuntimeState, VariablePool
from core.workflow.enums import WorkflowNodeExecutionStatus

View File

@ -7,10 +7,10 @@ import pytest
from core.helper.encrypter import (
batch_decrypt_token,
decrypt_token,
encrypt_secret_keys,
encrypt_token,
get_decrypt_decoding,
obfuscated_token,
encrypt_secret_keys,
)
from libs.rsa import PrivkeyNotFoundError
@ -38,10 +38,7 @@ class TestObfuscatedToken:
assert "*" * 12 in obfuscated
def test_encrypt_secret_keys_simple_dict(self):
data = {
"api_key": "fake-secret-key",
"username": "admin"
}
data = {"api_key": "fake-secret-key", "username": "admin"}
secret_vars = {"api_key"}
result = encrypt_secret_keys(data, secret_vars)
@ -52,13 +49,7 @@ class TestObfuscatedToken:
assert result["username"] == "admin"
def test_encrypt_secret_keys_nested_dict(self):
data = {
"outer": {
"inner_secret": "super-secret",
"inner_public": "visible"
},
"non_secret": "plain"
}
data = {"outer": {"inner_secret": "super-secret", "inner_public": "visible"}, "non_secret": "plain"}
secret_vars = {"inner_secret"}
result = encrypt_secret_keys(data, secret_vars)
@ -68,11 +59,8 @@ class TestObfuscatedToken:
assert result["non_secret"] == "plain"
def test_encrypt_secret_keys_list_of_dicts(self):
data = [
{"token1": "abc123", "id": 1},
{"token2": "xyz789", "id": 2}
]
secret_vars = {"token1","token2"}
data = [{"token1": "abc123", "id": 1}, {"token2": "xyz789", "id": 2}]
secret_vars = {"token1", "token2"}
result = encrypt_secret_keys(data, secret_vars)
@ -80,7 +68,6 @@ class TestObfuscatedToken:
assert result[1]["token2"] == obfuscated_token("xyz789")
assert result[0]["id"] == 1
def test_encrypt_secret_keys_non_secret_scalar(self):
# When the object is just a string, it should remain unchanged
result = encrypt_secret_keys("hello-world", secret_variables={"api_key"})
@ -91,6 +78,7 @@ class TestObfuscatedToken:
assert encrypt_secret_keys([], {"secret"}) == []
assert encrypt_secret_keys(None, {"secret"}) is None
class TestEncryptToken:
@patch("models.engine.db.session.query")
@patch("libs.rsa.encrypt")