mirror of
https://github.com/crewAIInc/crewAI.git
synced 2026-01-18 20:38:13 +00:00
feat: restructure project as UV workspace with crewai in lib/
This commit is contained in:
@@ -1,118 +0,0 @@
|
||||
"""Test for the SecurityConfig class."""
|
||||
|
||||
import json
|
||||
from datetime import datetime
|
||||
|
||||
from crewai.security import Fingerprint, SecurityConfig
|
||||
|
||||
|
||||
def test_security_config_creation_with_defaults():
|
||||
"""Test creating a SecurityConfig with default values."""
|
||||
config = SecurityConfig()
|
||||
|
||||
# Check default values
|
||||
assert config.fingerprint is not None # Fingerprint is auto-generated
|
||||
assert isinstance(config.fingerprint, Fingerprint)
|
||||
assert config.fingerprint.uuid_str is not None # UUID is auto-generated
|
||||
|
||||
|
||||
def test_security_config_fingerprint_generation():
|
||||
"""Test that SecurityConfig automatically generates fingerprints."""
|
||||
config = SecurityConfig()
|
||||
|
||||
# Check that fingerprint was auto-generated
|
||||
assert config.fingerprint is not None
|
||||
assert isinstance(config.fingerprint, Fingerprint)
|
||||
assert isinstance(config.fingerprint.uuid_str, str)
|
||||
assert len(config.fingerprint.uuid_str) > 0
|
||||
|
||||
|
||||
def test_security_config_init_params():
|
||||
"""Test that SecurityConfig can be initialized and modified."""
|
||||
# Create a config
|
||||
config = SecurityConfig()
|
||||
|
||||
# Create a custom fingerprint
|
||||
fingerprint = Fingerprint(metadata={"version": "1.0"})
|
||||
|
||||
# Set the fingerprint
|
||||
config.fingerprint = fingerprint
|
||||
|
||||
# Check fingerprint was set correctly
|
||||
assert config.fingerprint is fingerprint
|
||||
assert config.fingerprint.metadata == {"version": "1.0"}
|
||||
|
||||
|
||||
def test_security_config_to_dict():
|
||||
"""Test converting SecurityConfig to dictionary."""
|
||||
# Create a config with a fingerprint that has metadata
|
||||
config = SecurityConfig()
|
||||
config.fingerprint.metadata = {"version": "1.0"}
|
||||
|
||||
config_dict = config.to_dict()
|
||||
|
||||
# Check the fingerprint is in the dict
|
||||
assert "fingerprint" in config_dict
|
||||
assert isinstance(config_dict["fingerprint"], dict)
|
||||
assert config_dict["fingerprint"]["metadata"] == {"version": "1.0"}
|
||||
|
||||
|
||||
def test_security_config_from_dict():
|
||||
"""Test creating SecurityConfig from dictionary."""
|
||||
# Create a fingerprint dict
|
||||
fingerprint_dict = {
|
||||
"uuid_str": "b723c6ff-95de-5e87-860b-467b72282bd8",
|
||||
"created_at": datetime.now().isoformat(),
|
||||
"metadata": {"version": "1.0"}
|
||||
}
|
||||
|
||||
# Create a config dict with just the fingerprint
|
||||
config_dict = {
|
||||
"fingerprint": fingerprint_dict
|
||||
}
|
||||
|
||||
# Create config manually since from_dict has a specific implementation
|
||||
config = SecurityConfig()
|
||||
|
||||
# Set the fingerprint manually from the dict
|
||||
fingerprint = Fingerprint.from_dict(fingerprint_dict)
|
||||
config.fingerprint = fingerprint
|
||||
|
||||
# Check fingerprint was properly set
|
||||
assert config.fingerprint is not None
|
||||
assert isinstance(config.fingerprint, Fingerprint)
|
||||
assert config.fingerprint.uuid_str == fingerprint_dict["uuid_str"]
|
||||
assert config.fingerprint.metadata == fingerprint_dict["metadata"]
|
||||
|
||||
|
||||
def test_security_config_json_serialization():
|
||||
"""Test that SecurityConfig can be JSON serialized and deserialized."""
|
||||
# Create a config with fingerprint metadata
|
||||
config = SecurityConfig()
|
||||
config.fingerprint.metadata = {"version": "1.0"}
|
||||
|
||||
# Convert to dict and then JSON
|
||||
config_dict = config.to_dict()
|
||||
|
||||
# Make sure fingerprint is properly converted to dict
|
||||
assert isinstance(config_dict["fingerprint"], dict)
|
||||
|
||||
# Now it should be JSON serializable
|
||||
json_str = json.dumps(config_dict)
|
||||
|
||||
# Should be able to parse back to dict
|
||||
parsed_dict = json.loads(json_str)
|
||||
|
||||
# Check fingerprint values match
|
||||
assert parsed_dict["fingerprint"]["metadata"] == {"version": "1.0"}
|
||||
|
||||
# Create a new config manually
|
||||
new_config = SecurityConfig()
|
||||
|
||||
# Set the fingerprint from the parsed data
|
||||
fingerprint_data = parsed_dict["fingerprint"]
|
||||
new_fingerprint = Fingerprint.from_dict(fingerprint_data)
|
||||
new_config.fingerprint = new_fingerprint
|
||||
|
||||
# Check the new config has the same fingerprint metadata
|
||||
assert new_config.fingerprint.metadata == {"version": "1.0"}
|
||||
Reference in New Issue
Block a user