Support marshmallow 4

Author: lafrech

pyproject.toml

@@ -45,6 +45,7 @@ docs = [
   "alabaster==1.0.0",
 ]
 tests = [
+  "packaging>=24.2",
   "pytest==8.3.5",
   "pytest-cov==6.0.0",
   "coverage==7.6.12",

tests/conftest.py

@@ -1,3 +1,4 @@
+import importlib.metadata
 from collections import namedtuple
 
 import pytest
@@ -6,8 +7,12 @@
 
 import marshmallow as ma
 
+from packaging.version import Version
+
 from .mocks import DatabaseMock
 
+MA_VERSION = Version(importlib.metadata.version("marshmallow"))
+
 
 class AppConfig:
     """Base application configuration class
@@ -36,6 +41,12 @@ def app(request):
     return _app
 
 
+if MA_VERSION.major >= 4:
+    pass_collection_true_kwargs = {"pass_collection": True}
+else:
+    pass_collection_true_kwargs = {"pass_many": True}
+
+
 class CounterSchema(ma.Schema):
     """Base Schema with load/dump counters"""
 
@@ -50,12 +61,12 @@ def reset_load_count(cls):
     def reset_dump_count(cls):
         cls.dump_count = 0
 
-    @ma.post_load(pass_many=True)
+    @ma.post_load(**pass_collection_true_kwargs)
     def increment_load_count(self, data, **kwargs):
         self.__class__.load_count += 1
         return data
 
-    @ma.post_dump(pass_many=True)
+    @ma.post_dump(**pass_collection_true_kwargs)
     def increment_dump_count(self, data, **kwargs):
         self.__class__.dump_count += 1
         return data