ipulse-shared-core-ftredge 27.2.0__tar.gz → 27.4.0__tar.gz

{ipulse_shared_core_ftredge-27.2.0/src/ipulse_shared_core_ftredge.egg-info → ipulse_shared_core_ftredge-27.4.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ipulse_shared_core_ftredge
-Version: 27.2.0
+Version: 27.4.0
 Summary: Shared Core models and Logger util for the Pulse platform project. Using AI for financial advisory and investment management.
 Home-page: https://github.com/TheFutureEdge/ipulse_shared_core
 Author: Russlan Ramdowar

{ipulse_shared_core_ftredge-27.2.0 → ipulse_shared_core_ftredge-27.4.0}/setup.py

@@ -3,7 +3,7 @@ from setuptools import setup, find_packages
 
 setup(
     name='ipulse_shared_core_ftredge',
-    version='27.2.0',
+    version='27.4.0',
     package_dir={'': 'src'},  # Specify the source directory
     packages=find_packages(where='src'),  # Look for packages in 'src'
     install_requires=[

{ipulse_shared_core_ftredge-27.2.0 → ipulse_shared_core_ftredge-27.4.0}/src/ipulse_shared_core_ftredge/models/base_api_response.py

@@ -16,6 +16,10 @@ class BaseAPIResponse(BaseModel, Generic[T]):
     message: Optional[str] = None
     error: Optional[str] = None
 
+    # Optional fields for specific use cases
+    cache_hit: Optional[bool] = None  # Whether data came from cache
+    charged: Optional[bool] = None    # Whether credits were charged for this request
+
     metadata: Dict[str, Any] = {
         "timestamp": dt.datetime.now(dt.timezone.utc).isoformat()
     }

{ipulse_shared_core_ftredge-27.2.0 → ipulse_shared_core_ftredge-27.4.0}/src/ipulse_shared_core_ftredge/models/time_series_packaged_dataset_model.py

@@ -3,7 +3,6 @@ from typing import List, Optional, TypeVar, Generic
 from datetime import datetime
 from pydantic import Field, BaseModel
 from ipulse_shared_core_ftredge.models.base_data_model import BaseDataModel
-from ipulse_shared_base_ftredge.enums import DatasetLineage, DatasetScope
 
 # Generic type for the records within the dataset
 RecordsSamplingType = TypeVar('RecordsSamplingType', bound=BaseModel)
@@ -13,11 +12,14 @@ class TimeSeriesPackagedDatasetModel(BaseDataModel, Generic[RecordsSamplingType]
     An intermediary model for time series datasets that holds aggregated records.
     It provides a generic way to handle different types of time series records.
     """
-    dataset_id: str = Field(..., description="The unique identifier for this dataset, often matching the asset ID.")
+    # Subject identification fields
+    subject_id: str = Field(default="", description="The unique identifier for the subject.")
+    pulse_sector_category: str = Field(default="", description="The sector category for the subject.")
 
-    dataset_modality: str = Field(..., description="The modality of the dataset, e.g., 'time_series', 'cross_sectional'.")
-    dataset_lineage: DatasetLineage = Field(..., description="The lineage of the data, indicating its origin and transformations.")
-    dataset_partition: DatasetScope = Field(..., description="The partition type of the dataset, e.g., full, subsampled.")
+    # Schema identification fields
+    schema_id: str = Field(default="", description="The unique identifier for the schema.")
+    schema_name: str = Field(default="", description="The name of the schema.")
+    schema_version: int = Field(default=1, description="The version of the schema.")
 
     # Generic lists for different temporal buckets of records
     max_bulk_records: List[RecordsSamplingType] = Field(default_factory=list)
@@ -39,5 +41,5 @@ class TimeSeriesPackagedDatasetModel(BaseDataModel, Generic[RecordsSamplingType]
 
     @property
     def id(self) -> str:
-        """Return dataset_id for backward compatibility and consistency."""
-        return self.dataset_id
+        """Return subject_id for backward compatibility and consistency."""
+        return self.subject_id

{ipulse_shared_core_ftredge-27.2.0 → ipulse_shared_core_ftredge-27.4.0}/src/ipulse_shared_core_ftredge/services/__init__.py

@@ -2,7 +2,7 @@
 
 
 # Import from base services
-from .base import BaseFirestoreService, CacheAwareFirestoreService
+from .base import BaseFirestoreService, CacheAwareFirestoreService, MultiCollectionCacheAwareFirestoreService
 
 from .charging_processors import ChargingProcessor
 from .user_charging_service import UserChargingService

{ipulse_shared_core_ftredge-27.2.0 → ipulse_shared_core_ftredge-27.4.0}/src/ipulse_shared_core_ftredge/services/base/__init__.py

@@ -7,6 +7,7 @@ preventing circular import dependencies.
 
 from .base_firestore_service import BaseFirestoreService
 from .cache_aware_firestore_service import CacheAwareFirestoreService
+from .multi_collection_cache_aware_firestore_service import MultiCollectionCacheAwareFirestoreService
 
 __all__ = [
     'BaseFirestoreService',

{ipulse_shared_core_ftredge-27.2.0 → ipulse_shared_core_ftredge-27.4.0}/src/ipulse_shared_core_ftredge/services/base/cache_aware_firestore_service.py

@@ -1,13 +1,14 @@
 """Cache-aware Firestore service base class."""
 import time
-from typing import TypeVar, Generic, Dict, Any, List, Optional, Union, Type
+from typing import TypeVar, Generic, Dict, Any, List, Optional, Union, Type, Tuple
 from google.cloud import firestore
 from . import BaseFirestoreService
 from ...exceptions import ResourceNotFoundError, ServiceError
 from ...cache.shared_cache import SharedCache
 from ...models import BaseDataModel
 
-T = TypeVar('T', bound=BaseDataModel)
+T = TypeVar('T', bound=BaseDataModel)0
+0
 
 class CacheAwareFirestoreService(BaseFirestoreService[T], Generic[T]):
     """
@@ -58,7 +59,9 @@ class CacheAwareFirestoreService(BaseFirestoreService[T], Generic[T]):
             cache_check_time = (time.time() - start_time) * 1000
 
             if cached_doc is not None:
-                self.logger.debug(f"Cache HIT for document {doc_id} in {cache_check_time:.2f}ms")
+                # SharedCache.get() already logs cache hit, only log timing if significant
+                if cache_check_time > 5.0:  # Only log if cache check took >5ms
+                    self.logger.debug(f"Cache HIT for document {doc_id} in {cache_check_time:.2f}ms")
                 if convert_to_model and self.model_class:
                     return self._convert_to_model(cached_doc, doc_id)
                 else:
@@ -68,7 +71,66 @@ class CacheAwareFirestoreService(BaseFirestoreService[T], Generic[T]):
                 self.logger.debug(f"Cache MISS for document {doc_id} - checking Firestore")
 
         # Fetch from Firestore using parent method
-        return await super().get_document(doc_id, convert_to_model)
+        result = await super().get_document(doc_id, convert_to_model)
+
+        # Cache the result if we have a cache and got valid data
+        if self.document_cache and result is not None:
+            if convert_to_model and isinstance(result, BaseDataModel):
+                # Cache the model's dict representation
+                self._cache_document_data(doc_id, result.model_dump())
+            elif isinstance(result, dict):
+                # Cache the dict directly
+                self._cache_document_data(doc_id, result)
+
+        return result
+
+    async def get_document_with_cache_info(self, doc_id: str, convert_to_model: bool = True) -> Tuple[Union[T, Dict[str, Any], None], bool]:
+        """
+        Get a document with cache hit information.
+
+        Args:
+            doc_id: Document ID to fetch
+            convert_to_model: Whether to convert to Pydantic model
+
+        Returns:
+            Tuple of (document, cache_hit) where cache_hit indicates if from cache
+
+        Raises:
+            ResourceNotFoundError: If document doesn't exist
+        """
+        cache_hit = False
+
+        # Check cache first
+        if self.document_cache:
+            cached_doc = self.document_cache.get(doc_id)
+            if cached_doc is not None:
+                cache_hit = True
+                # Note: SharedCache.get() already logs cache hit at DEBUG level
+                if convert_to_model and self.model_class:
+                    return self._convert_to_model(cached_doc, doc_id), cache_hit
+                else:
+                    cached_doc['id'] = doc_id
+                    return cached_doc, cache_hit
+
+        # Cache miss - fetch from Firestore
+        self.logger.debug(f"Cache MISS for document {doc_id} - checking Firestore")
+
+        try:
+            result = await super().get_document(doc_id, convert_to_model)
+
+            # Cache the result if we have a cache and got valid data
+            if self.document_cache and result is not None:
+                if convert_to_model and isinstance(result, BaseDataModel):
+                    # Cache the model's dict representation
+                    self._cache_document_data(doc_id, result.model_dump())
+                elif isinstance(result, dict):
+                    # Cache the dict directly
+                    self._cache_document_data(doc_id, result)
+
+            return result, cache_hit
+
+        except ResourceNotFoundError:
+            return None, cache_hit
 
     async def get_all_documents(self, cache_key: Optional[str] = None, as_models: bool = True) -> Union[List[T], List[Dict[str, Any]]]:
         """
@@ -151,7 +213,7 @@ class CacheAwareFirestoreService(BaseFirestoreService[T], Generic[T]):
         """Helper to cache document data if document_cache is available."""
         if self.document_cache:
             self.document_cache.set(doc_id, data)
-            self.logger.debug(f"Cached item {doc_id} in {self.document_cache.name}")
+            # Note: SharedCache.set() already logs at DEBUG level
 
     async def create_document(self, doc_id: str, data: Union[T, Dict[str, Any]], creator_uid: str, merge: bool = False) -> Dict[str, Any]:
         """Create document and invalidate cache."""

ipulse_shared_core_ftredge-27.4.0/src/ipulse_shared_core_ftredge/services/base/multi_collection_cache_aware_firestore_service.py

@@ -0,0 +1,244 @@
+"""
+Generic multi-collection cache-aware Firestore service.
+
+This service extends CacheAwareFirestoreService to support dynamic collection operations
+while maintaining all proven infrastructure patterns. It's designed to be generic and
+reusable across different model types.
+"""
+from typing import Dict, Any, List, Optional, Union, Type, TypeVar, Generic
+from google.cloud import firestore
+from .cache_aware_firestore_service import CacheAwareFirestoreService
+from ...exceptions import ServiceError, ValidationError, ResourceNotFoundError
+from ...cache.shared_cache import SharedCache
+from ...models import BaseDataModel
+import logging
+
+# Generic type for BaseDataModel subclasses
+T = TypeVar('T', bound=BaseDataModel)
+
+
+class MultiCollectionCacheAwareFirestoreService(CacheAwareFirestoreService[T], Generic[T]):
+    """
+    Generic multi-collection extension of CacheAwareFirestoreService.
+
+    This service extends the proven CacheAwareFirestoreService infrastructure to support
+    dynamic collection operations based on storage_location_path while maintaining
+    all caching, error handling, and CRUD capabilities.
+
+    This is a generic base class that can be extended for specific model types.
+    """
+
+    def __init__(self,
+                 db: firestore.Client,
+                 logger: logging.Logger,
+                 model_class: Type[T],
+                 resource_type: str,
+                 base_collection_name: str,
+                 timeout: float = 30.0):
+
+        # Initialize the parent CacheAwareFirestoreService with a base collection
+        # We'll override the collection_name dynamically per operation
+        super().__init__(
+            db=db,
+            collection_name=base_collection_name,  # Base collection name
+            resource_type=resource_type,
+            model_class=model_class,
+            logger=logger,
+            document_cache=None,  # We'll manage caches per collection
+            collection_cache=None,  # We'll manage caches per collection
+            timeout=timeout
+        )
+
+        # Cache for per-collection cache instances
+        self._collection_caches: Dict[str, Dict[str, SharedCache]] = {}
+
+        self.logger.info(f"MultiCollectionCacheAwareFirestoreService initialized for {resource_type}")
+
+    def _get_collection_caches(self, storage_location_path: str) -> Dict[str, SharedCache]:
+        """Get or create cache instances for a specific storage location."""
+        if storage_location_path not in self._collection_caches:
+            # Create collection-specific cache instances
+            # No need for safe_name transformation - dots are fine in strings
+
+            document_cache = SharedCache(
+                name=f"MultiColDoc_{storage_location_path}",
+                ttl=600.0,  # 10 minutes
+                enabled=True,
+                logger=self.logger
+            )
+
+            collection_cache = SharedCache(
+                name=f"MultiColCollection_{storage_location_path}",
+                ttl=600.0,  # 10 minutes
+                enabled=True,
+                logger=self.logger
+            )
+
+            self._collection_caches[storage_location_path] = {
+                'document': document_cache,
+                'collection': collection_cache
+            }
+
+            self.logger.info(f"Created cache instances for collection: {storage_location_path}")
+
+        return self._collection_caches[storage_location_path]
+
+    def _set_collection_context(self, storage_location_path: str):
+        """Set the collection context for the current operation."""
+        # Update the collection name for this operation
+        self.collection_name = storage_location_path
+
+        # Update the cache references for this collection
+        caches = self._get_collection_caches(storage_location_path)
+        self.document_cache = caches['document']
+        self.collection_cache = caches['collection']
+
+    async def get_document_from_collection(self,
+                                         storage_location_path: str,
+                                         doc_id: str,
+                                         convert_to_model: bool = True) -> Union[T, Dict[str, Any], None]:
+        """
+        Get a document from a specific collection using the cache-aware infrastructure.
+        """
+        try:
+            # Set collection context
+            self._set_collection_context(storage_location_path)
+
+            # Use the parent's cache-aware get_document method
+            return await super().get_document(doc_id, convert_to_model)
+
+        except ResourceNotFoundError:
+            self.logger.info(f"Document {doc_id} not found in {storage_location_path}")
+            return None
+        except Exception as e:
+            self.logger.error(f"Error getting document {doc_id} from {storage_location_path}: {str(e)}", exc_info=True)
+            raise ServiceError(
+                operation=f"getting document from {storage_location_path}",
+                error=e,
+                resource_type=self.resource_type,
+                resource_id=doc_id
+            ) from e
+
+    async def get_all_documents_from_collection(self,
+                                              storage_location_path: str,
+                                              cache_key: Optional[str] = None) -> List[T]:
+        """
+        Get all documents from a specific collection using cache-aware infrastructure.
+        """
+        try:
+            # Set collection context
+            self._set_collection_context(storage_location_path)
+
+            # Use cache key if not provided
+            if not cache_key:
+                cache_key = f"all_documents_{storage_location_path}"
+
+            # Use the parent's cache-aware get_all_documents method
+            results = await super().get_all_documents(cache_key=cache_key, as_models=True)
+
+            # Ensure we return model instances
+            model_results: List[T] = []
+            for item in results:
+                if isinstance(item, BaseDataModel) and self.model_class and isinstance(item, self.model_class):
+                    model_results.append(item)  # type: ignore
+                elif isinstance(item, dict) and self.model_class:
+                    try:
+                        model_results.append(self.model_class.model_validate(item))
+                    except Exception as e:
+                        self.logger.warning(f"Failed to convert dict to model: {e}")
+
+            return model_results
+
+        except Exception as e:
+            self.logger.error(f"Error getting all documents from {storage_location_path}: {str(e)}", exc_info=True)
+            raise ServiceError(
+                operation=f"getting all documents from {storage_location_path}",
+                error=e,
+                resource_type=self.resource_type
+            ) from e
+
+    async def create_document_in_collection(self,
+                                          storage_location_path: str,
+                                          doc_id: str,
+                                          data: Union[T, Dict[str, Any]],
+                                          creator_uid: str,
+                                          merge: bool = False) -> Dict[str, Any]:
+        """
+        Create a document in a specific collection using cache-aware infrastructure.
+        Automatically handles cache invalidation.
+        """
+        try:
+            # Set collection context
+            self._set_collection_context(storage_location_path)
+
+            # Use the parent's cache-aware create_document method
+            return await super().create_document(doc_id, data, creator_uid, merge)
+
+        except Exception as e:
+            self.logger.error(f"Error creating document {doc_id} in {storage_location_path}: {str(e)}", exc_info=True)
+            raise ServiceError(
+                operation=f"creating document in {storage_location_path}",
+                error=e,
+                resource_type=self.resource_type,
+                resource_id=doc_id
+            ) from e
+
+    async def update_document_in_collection(self,
+                                          storage_location_path: str,
+                                          doc_id: str,
+                                          update_data: Dict[str, Any],
+                                          updater_uid: str,
+                                          require_exists: bool = True) -> Dict[str, Any]:
+        """
+        Update a document in a specific collection using cache-aware infrastructure.
+        Automatically handles cache invalidation.
+        """
+        try:
+            # Set collection context
+            self._set_collection_context(storage_location_path)
+
+            # Use the parent's cache-aware update_document method
+            return await super().update_document(doc_id, update_data, updater_uid, require_exists)
+
+        except Exception as e:
+            self.logger.error(f"Error updating document {doc_id} in {storage_location_path}: {str(e)}", exc_info=True)
+            raise ServiceError(
+                operation=f"updating document in {storage_location_path}",
+                error=e,
+                resource_type=self.resource_type,
+                resource_id=doc_id
+            ) from e
+
+    async def delete_document_from_collection(self,
+                                            storage_location_path: str,
+                                            doc_id: str,
+                                            require_exists: bool = True) -> bool:
+        """
+        Delete a document from a specific collection using cache-aware infrastructure.
+        Automatically handles cache invalidation.
+        """
+        try:
+            # Set collection context
+            self._set_collection_context(storage_location_path)
+
+            # Use the parent's cache-aware delete_document method
+            return await super().delete_document(doc_id, require_exists)
+
+        except Exception as e:
+            self.logger.error(f"Error deleting document {doc_id} from {storage_location_path}: {str(e)}", exc_info=True)
+            raise ServiceError(
+                operation=f"deleting document from {storage_location_path}",
+                error=e,
+                resource_type=self.resource_type,
+                resource_id=doc_id
+            ) from e
+
+    def get_cache_stats(self) -> Dict[str, Any]:
+        """Get cache statistics for all collections managed by this service."""
+        stats = {}
+        for storage_path, caches in self._collection_caches.items():
+            stats[storage_path] = {
+                'document_cache': caches['document'].get_stats(),
+                'collection_cache': caches['collection'].get_stats()
+            }
+        return stats

{ipulse_shared_core_ftredge-27.2.0 → ipulse_shared_core_ftredge-27.4.0/src/ipulse_shared_core_ftredge.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ipulse_shared_core_ftredge
-Version: 27.2.0
+Version: 27.4.0
 Summary: Shared Core models and Logger util for the Pulse platform project. Using AI for financial advisory and investment management.
 Home-page: https://github.com/TheFutureEdge/ipulse_shared_core
 Author: Russlan Ramdowar

{ipulse_shared_core_ftredge-27.2.0 → ipulse_shared_core_ftredge-27.4.0}/src/ipulse_shared_core_ftredge.egg-info/SOURCES.txt

@@ -42,6 +42,7 @@ src/ipulse_shared_core_ftredge/services/user_charging_service.py
 src/ipulse_shared_core_ftredge/services/base/__init__.py
 src/ipulse_shared_core_ftredge/services/base/base_firestore_service.py
 src/ipulse_shared_core_ftredge/services/base/cache_aware_firestore_service.py
+src/ipulse_shared_core_ftredge/services/base/multi_collection_cache_aware_firestore_service.py
 src/ipulse_shared_core_ftredge/services/catalog/__init__.py
 src/ipulse_shared_core_ftredge/services/catalog/catalog_subscriptionplan_service.py
 src/ipulse_shared_core_ftredge/services/catalog/catalog_usertype_service.py
@@ -55,7 +56,5 @@ src/ipulse_shared_core_ftredge/services/user/userauth_operations.py
 src/ipulse_shared_core_ftredge/services/user/userprofile_operations.py
 src/ipulse_shared_core_ftredge/services/user/userstatus_operations.py
 src/ipulse_shared_core_ftredge/utils/__init__.py
-src/ipulse_shared_core_ftredge/utils/authz_credit_extraction.py
 src/ipulse_shared_core_ftredge/utils/custom_json_encoder.py
-src/ipulse_shared_core_ftredge/utils/json_encoder.py
-tests/test_shared_cache.py
+src/ipulse_shared_core_ftredge/utils/json_encoder.py

ipulse_shared_core_ftredge-27.2.0/tests/test_shared_cache.py

@@ -1,147 +0,0 @@
-"""Tests for the SharedCache implementation."""
-
-import time
-import unittest
-import logging
-from ipulse_shared_core_ftredge.cache.shared_cache import SharedCache
-
-# Configure logging for tests
-logging.basicConfig(level=logging.INFO)
-logger = logging.getLogger(__name__)
-
-
-class TestSharedCache(unittest.TestCase):
-    """Test cases for SharedCache."""
-
-    def setUp(self):
-        """Set up test fixtures."""
-        self.cache = SharedCache[str](
-            name="TestCache",
-            ttl=0.5,  # Short TTL for faster testing
-            enabled=True,
-            logger=logger
-        )
-
-    def test_cache_set_get(self):
-        """Test basic cache set and get operations."""
-        # Set a value
-        self.cache.set("test_key", "test_value")
-
-        # Get the value
-        cached_value = self.cache.get("test_key")
-
-        # Verify value was cached
-        self.assertEqual(cached_value, "test_value")
-
-    def test_cache_ttl_expiration(self):
-        """Test cache TTL expiration."""
-        # Set a value
-        self.cache.set("expiring_key", "expiring_value")
-
-        # Verify it's initially cached
-        self.assertEqual(self.cache.get("expiring_key"), "expiring_value")
-
-        # Wait for TTL to expire
-        time.sleep(0.6)  # Slightly longer than TTL (0.5s)
-
-        # Verify value is no longer cached
-        self.assertIsNone(self.cache.get("expiring_key"))
-
-    def test_cache_invalidate(self):
-        """Test cache invalidation."""
-        # Set multiple values
-        self.cache.set("key1", "value1")
-        self.cache.set("key2", "value2")
-
-        # Invalidate specific key
-        self.cache.invalidate("key1")
-
-        # Verify key1 is gone but key2 remains
-        self.assertIsNone(self.cache.get("key1"))
-        self.assertEqual(self.cache.get("key2"), "value2")
-
-    def test_cache_invalidate_all(self):
-        """Test invalidating all cache entries."""
-        # Set multiple values
-        self.cache.set("key1", "value1")
-        self.cache.set("key2", "value2")
-
-        # Invalidate all
-        self.cache.invalidate_all()
-
-        # Verify both keys are gone
-        self.assertIsNone(self.cache.get("key1"))
-        self.assertIsNone(self.cache.get("key2"))
-
-    def test_cache_get_or_set(self):
-        """Test get_or_set functionality."""
-        # Define a counter to verify how many times the loader is called
-        counter = [0]
-
-        def data_loader():
-            counter[0] += 1
-            return f"loaded_value_{counter[0]}"
-
-        # First call should use data_loader
-        value1, was_cached1 = self.cache.get_or_set("loader_key", data_loader)
-
-        # Second call should use cached value
-        value2, was_cached2 = self.cache.get_or_set("loader_key", data_loader)
-
-        # Verify results
-        self.assertEqual(value1, "loaded_value_1")
-        self.assertEqual(value2, "loaded_value_1")  # Same value from cache
-        self.assertFalse(was_cached1)  # First call was not cached
-        self.assertTrue(was_cached2)   # Second call was cached
-        self.assertEqual(counter[0], 1)  # Loader called exactly once
-
-    def test_cache_disabled(self):
-        """Test cache behavior when disabled."""
-        # Create disabled cache
-        disabled_cache = SharedCache[str](
-            name="DisabledCache",
-            ttl=1.0,
-            enabled=False,
-            logger=logger
-        )
-
-        # Set a value
-        disabled_cache.set("disabled_key", "disabled_value")
-
-        # Attempt to get - should return None since cache is disabled
-        cached_value = disabled_cache.get("disabled_key")
-        self.assertIsNone(cached_value)
-
-    def test_cache_generic_typing(self):
-        """Test cache with different data types."""
-        # Integer cache
-        int_cache = SharedCache[int](name="IntCache", ttl=1.0, enabled=True)
-        int_cache.set("int_key", 123)
-        self.assertEqual(int_cache.get("int_key"), 123)
-
-        # Dictionary cache
-        dict_cache = SharedCache[dict](name="DictCache", ttl=1.0, enabled=True)
-        dict_cache.set("dict_key", {"a": 1, "b": 2})
-        self.assertEqual(dict_cache.get("dict_key"), {"a": 1, "b": 2})
-
-    def test_cache_stats(self):
-        """Test cache statistics."""
-        # Add some data
-        self.cache.set("stats_key1", "stats_value1")
-        self.cache.set("stats_key2", "stats_value2")
-
-        # Get stats
-        stats = self.cache.get_stats()
-
-        # Verify stats
-        self.assertEqual(stats["name"], "TestCache")
-        self.assertEqual(stats["enabled"], True)
-        self.assertEqual(stats["ttl_seconds"], 0.5)
-        self.assertEqual(stats["item_count"], 2)
-        self.assertIn("stats_key1", stats["first_20_keys"])
-        self.assertIn("stats_key2", stats["first_20_keys"])
-        self.assertEqual(stats["total_keys"], 2)
-
-
-if __name__ == "__main__":
-    unittest.main()