mirror of
https://github.com/Skyvern-AI/skyvern.git
synced 2026-04-28 11:40:32 +00:00
31 lines
1 KiB
Python
31 lines
1 KiB
Python
import pytest
|
|
|
|
from skyvern.forge.sdk.schemas.credentials import (
|
|
CreateCredentialRequest,
|
|
CredentialType,
|
|
SecretCredential,
|
|
)
|
|
|
|
|
|
class TestSecretCredentialModels:
|
|
def test_secret_credential_creation(self) -> None:
|
|
cred = SecretCredential(secret_value="sk-abc123", secret_label="API Key")
|
|
assert cred.secret_value == "sk-abc123"
|
|
assert cred.secret_label == "API Key"
|
|
|
|
def test_secret_credential_optional_type(self) -> None:
|
|
cred = SecretCredential(secret_value="token123")
|
|
assert cred.secret_label is None
|
|
|
|
def test_non_empty_validation(self) -> None:
|
|
with pytest.raises(ValueError):
|
|
SecretCredential(secret_value="")
|
|
|
|
def test_create_request_with_secret(self) -> None:
|
|
req = CreateCredentialRequest(
|
|
name="My API Key",
|
|
credential_type=CredentialType.SECRET,
|
|
credential=SecretCredential(secret_value="sk-12345"),
|
|
)
|
|
assert req.credential_type == CredentialType.SECRET
|
|
assert req.credential.secret_value == "sk-12345"
|