More tests moved db folder to tests folder (#7)

Author: janbjorge

.github/workflows/ci.yml

@@ -30,7 +30,7 @@ jobs:
         run: |
           docker build \
             --build-arg POSTGRES_VERSION=${{ matrix.postgres-version }} \
-            -t custom-postgres:latest db/
+            -t custom-postgres:latest tests/db/
         env:
           POSTGRES_VERSION: ${{ matrix.postgres-version }}
 

db/Dockerfile tests/db/Dockerfiledb/Dockerfile renamed to tests/db/Dockerfile

@@ -73,3 +73,56 @@ async def raise_runtime_error() -> NoReturn:
     )
     assert len(exceptions) == N
     assert all(isinstance(exc, RuntimeError) for exc in exceptions)
+
+
+@pytest.mark.parametrize("N", (1, 2, 4, 16, 64))
+async def test_gready_cache_identity(
+    N: int,
+    pgconn: asyncpg.Connection,
+) -> None:
+    statistics = collections.Counter[str]()
+    listener = listeners.PGEventQueue()
+    await listener.connect(
+        pgconn,
+        models.PGChannel("test_gready_cache_decorator_exceptions"),
+    )
+
+    @decorators.cache(
+        strategy=strategies.Gready(listener=listener),
+        statistics_callback=lambda x: statistics.update([x]),
+    )
+    async def identity(x: int) -> int:
+        return x
+
+    results = await asyncio.gather(*[identity(n) for n in range(N)])
+
+    assert sorted(results) == list(range(N))
+    assert statistics["miss"] == N
+    assert statistics["hit"] == 0
+
+
+@pytest.mark.parametrize("N", (1, 2, 4, 16, 64))
+async def test_gready_cache_sleepy(
+    N: int,
+    pgconn: asyncpg.Connection,
+) -> None:
+    statistics = collections.Counter[str]()
+    listener = listeners.PGEventQueue()
+    await listener.connect(
+        pgconn,
+        models.PGChannel("test_gready_cache_decorator_exceptions"),
+    )
+
+    @decorators.cache(
+        strategy=strategies.Gready(listener=listener),
+        statistics_callback=lambda x: statistics.update([x]),
+    )
+    async def now() -> datetime.datetime:
+        await asyncio.sleep(0.01)
+        return datetime.datetime.now()
+
+    results = await asyncio.gather(*[now() for _ in range(N)])
+
+    assert len(set(results)) == 1
+    assert statistics["miss"] == 1
+    assert statistics["hit"] == N - 1

db/init_db.sh tests/db/init_db.shdb/init_db.sh renamed to tests/db/init_db.sh

@@ -38,7 +38,7 @@ async def test_2_caching(
 ) -> None:
     statistics = collections.Counter[str]()
     listener = listeners.PGEventQueue()
-    await listener.connect(pgconn, models.PGChannel("test_2_caching"))
+    await listener.connect(pgconn, models.PGChannel("ch_pgcachewatch_table_change"))
 
     cnt = 0