Add CacheDatabaseAdapter keys method

nautilus_trader/cache/database.pyx

@@ -188,6 +188,36 @@ cdef class CacheDatabaseAdapter(CacheDatabaseFacade):
         self._backing.flushdb()
         self._log.info("Flushed database.", LogColor.BLUE)
 
+    cpdef list[str] keys(self, str pattern = "*"):
+        """
+        Return all keys in the database matching the given `pattern`.
+
+        Parameters
+        ----------
+        pattern : str, default '*'
+            The glob-style pattern to match against the keys in the database.
+
+        Returns
+        -------
+        list[str]
+
+        Raises
+        ------
+        ValueError
+            If `pattern` is not a valid string.
+
+        Warnings
+        --------
+        Using the default '*' pattern string can have serious performance implications and
+        can take a long time to execute if many keys exist in the database. This operation
+        can lead to high memory and CPU usage, and should be used with caution, especially
+        in production environments.
+
+        """
+        Condition.valid_string(pattern, "pattern")
+
+        return self._backing.keys(pattern)
+
     cpdef dict load(self):
         """
         Load all general objects from the database.

nautilus_trader/cache/facade.pxd

@@ -57,6 +57,7 @@ cdef class CacheDatabaseFacade:
     cdef LoggerAdapter _log
 
     cpdef void flush(self)
+    cpdef list[str] keys(self, str pattern=*)
     cpdef dict load(self)
     cpdef dict load_currencies(self)
     cpdef dict load_instruments(self)

nautilus_trader/cache/facade.pyx

@@ -81,6 +81,10 @@ cdef class CacheDatabaseFacade:
         """Abstract method (implement in subclass)."""
         raise NotImplementedError("method must be implemented in the subclass")  # pragma: no cover
 
+    cpdef list[str] keys(self, str pattern = "*"):
+        """Abstract method (implement in subclass)."""
+        raise NotImplementedError("method must be implemented in the subclass")  # pragma: no cover
+
     cpdef dict load(self):
         """Abstract method (implement in subclass)."""
         raise NotImplementedError("method must be implemented in the subclass")  # pragma: no cover

tests/integration_tests/infrastructure/test_cache_database.py

@@ -147,7 +147,11 @@ def setup(self):
     def teardown(self):
         # Tests will fail if Redis is not flushed on tear down
         self.database.flush()  # Comment this line out to preserve data between tests for debugging
-        time.sleep(0.1)  # Ensure clean slate
+
+        # Ensure clean slate
+        while self.database.keys("*"):
+            time.sleep(0.1)
+            continue
 
     @pytest.mark.asyncio
     async def test_load_general_objects_when_nothing_in_cache_returns_empty_dict(self):
@@ -156,6 +160,7 @@ async def test_load_general_objects_when_nothing_in_cache_returns_empty_dict(sel
 
         # Assert
         assert result == {}
+        assert self.database.keys() == []
 
     @pytest.mark.asyncio
     async def test_add_general_object_adds_to_cache(self):
@@ -171,6 +176,7 @@ async def test_add_general_object_adds_to_cache(self):
 
         # Assert
         assert self.database.load() == {key: str(bar).encode()}
+        assert self.database.keys() == ["trader-TESTER-000:general:" + key]
 
     @pytest.mark.asyncio
     async def test_add_currency(self):
@@ -1191,6 +1197,8 @@ async def test_rerunning_backtest_with_redis_db_builds_correct_index(self):
         # Reset engine
         self.engine.reset()
 
+        await asyncio.sleep(0.5)
+
         # Act
         self.engine.run()