WIP: Integration testsorm

1 files changed, 22 insertions, 0 deletions

diff --git a/tests/integration/test_db.py b/tests/integration/test_db.py
new file mode 100644
index 0000000..89ec31f
--- /dev/null
+++ b/tests/integration/test_db.py
@@ -0,0 +1,22 @@
+from sqlalchemy import create_engine
+
+from requests_cache import ALL_METHODS, CachedSession
+from requests_cache.backends import DbCache, DbDict
+from tests.integration.base_cache_test import BaseCacheTest
+from tests.integration.base_storage_test import BaseStorageTest
+
+# class TestDbDict(BaseStorageTest):
+#     storage_class = DbDict
+
+
+class TestDbCache(BaseCacheTest):
+    backend_class = DbCache
+
+    def init_session(self, clear=True, **kwargs) -> CachedSession:
+        kwargs.setdefault('allowable_methods', ALL_METHODS)
+        engine = create_engine('postgresql://postgres:hunter2@localhost:5432/postgres')
+        backend = DbCache(engine, **kwargs)
+        if clear:
+            backend.clear()
+
+        return CachedSession(backend=backend, **kwargs)