32 lines
1.0 KiB
Python
32 lines
1.0 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"
|