test: async function default value

tests/test_callable_default.py

@@ -0,0 +1,21 @@
+from tests import testmodels
+from tortoise.contrib import test
+
+
+class TestCallableDefault(test.TestCase):
+    async def test_default_create(self):
+        model = await testmodels.CallableDefault.create()
+        self.assertEqual(model.callable_default, "callable_default")
+        self.assertEqual(model.async_default, "async_callable_default")
+
+    async def test_default_by_save(self):
+        saved_model = testmodels.CallableDefault()
+        await saved_model.save()
+        self.assertEqual(saved_model.callable_default, "callable_default")
+        self.assertEqual(saved_model.async_default, "async_callable_default")
+
+    async def test_async_default_change(self):
+        default_change = testmodels.CallableDefault()
+        default_change.async_default = "changed"
+        await default_change.save()
+        self.assertEqual(default_change.async_default, "changed")

tests/testmodels.py

@@ -891,3 +891,17 @@ class PydanticMeta:
             alias_generator=camelize_var,
             populate_by_name=True,
         )
+
+
+def callable_default() -> str:
+    return "callable_default"
+
+
+async def async_callable_default() -> str:
+    return "async_callable_default"
+
+
+class CallableDefault(Model):
+    id = fields.IntField(pk=True)
+    callable_default = fields.CharField(max_length=32, default=callable_default)
+    async_default = fields.CharField(max_length=32, default=async_callable_default)

tortoise/models.py

@@ -859,7 +859,7 @@ def register_listener(cls, signal: Signals, listener: Callable):
     async def _set_async_default_field(self) -> None:
         """retrieve value from field's async default value"""
         if hasattr(self, "_await_when_save"):
-            for k, v in self._await_when_save.items():
+            for k, v in self._await_when_save.copy().items():
                 setattr(self, k, await v())
             self._await_when_save = {}