HOUSEKEEPING

mabel-dev · Sep 14, 2024 · 7942fcb · 7942fcb
1 parent 8c7f4ae
commit 7942fcb
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 7 deletions.
diff --git a/opteryx/managers/cache/memcached.py b/opteryx/managers/cache/memcached.py
@@ -98,7 +98,6 @@ def get(self, key: bytes) -> Union[bytes, None]:
             self.skips += 1
             return None
         try:
-            print(key)
             response = self._server.get(bytes(key))
             self._consecutive_failures = 0
             if response:

diff --git a/tests/storage/test_cache_memcached.py b/tests/storage/test_cache_memcached.py
@@ -37,15 +37,18 @@ def test_memcached_cache():
     for i in range(10):
         cur = conn.cursor()
         time.sleep(0.01)
-        cur.execute("SELECT * FROM testdata.flat.ten_files;")
+        cur.execute("SELECT count(*) FROM testdata.flat.ten_files;")
 
     print(f"hits: {cache.hits}, misses: {cache.misses}, skips: {cache.skips}, errors: {cache.errors}, sets: {cache.sets}")
 
-    # read the data again time, this should hit the cache
-    buffer = BufferPool()
-    buffer.reset()
-    cur = conn.cursor()
-    cur.execute("SELECT * FROM testdata.flat.ten_files;")
+    for i in range(10):
+        # read the data again time, this should hit the cache
+        buffer = BufferPool()
+        buffer.reset()
+
+        cur = conn.cursor()
+        cur.execute("SELECT count(*) FROM testdata.flat.ten_files;")
+
     stats = cur.stats
 
     print(f"hits: {cache.hits}, misses: {cache.misses}, skips: {cache.skips}, errors: {cache.errors}, sets: {cache.sets}")