diff --git a/src/tests/custom/database/src/model/db_context.py b/src/tests/custom/database/src/model/db_context.py
new file mode 100644
index 00000000..160ed8b4
--- /dev/null
+++ b/src/tests/custom/database/src/model/db_context.py
@@ -0,0 +1,8 @@
+from cpl.database import DatabaseSettings
+from cpl.database.context import DatabaseContext
+
+
+class DBContext(DatabaseContext):
+
+    def __init__(self, db_settings: DatabaseSettings):
+        DatabaseContext.__init__(self, db_settings)