dbos 0.26.0a6__tar.gz → 0.26.0a7__tar.gz

{dbos-0.26.0a6 → dbos-0.26.0a7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: dbos
-Version: 0.26.0a6
+Version: 0.26.0a7
 Summary: Ultra-lightweight durable execution in Python
 Author-Email: "DBOS, Inc." <contact@dbos.dev>
 License: MIT

{dbos-0.26.0a6 → dbos-0.26.0a7}/dbos/_dbos.py

@@ -172,6 +172,12 @@ class DBOSRegistry:
         if name in self.function_type_map:
             if self.function_type_map[name] != functype:
                 raise DBOSConflictingRegistrationError(name)
+            if name != TEMP_SEND_WF_NAME:
+                # Remove the `<temp>` prefix from the function name to avoid confusion
+                truncated_name = name.replace("<temp>.", "")
+                dbos_logger.warning(
+                    f"Duplicate registration of function '{truncated_name}'. A function named '{truncated_name}' has already been registered with DBOS. All functions registered with DBOS must have unique names."
+                )
         self.function_type_map[name] = functype
         self.workflow_info_map[name] = wrapped_func
 

{dbos-0.26.0a6 → dbos-0.26.0a7}/pyproject.toml

@@ -28,7 +28,7 @@ dependencies = [
 ]
 requires-python = ">=3.9"
 readme = "README.md"
-version = "0.26.0a6"
+version = "0.26.0a7"
 
 [project.license]
 text = "MIT"

{dbos-0.26.0a6 → dbos-0.26.0a7}/tests/test_dbos.py

@@ -1257,6 +1257,63 @@ def test_double_decoration(dbos: DBOS) -> None:
         my_function()
 
 
+def test_duplicate_registration(
+    dbos: DBOS, caplog: pytest.LogCaptureFixture, config: ConfigFile
+) -> None:
+    original_propagate = logging.getLogger("dbos").propagate
+    caplog.set_level(logging.WARNING, "dbos")
+    logging.getLogger("dbos").propagate = True
+
+    @DBOS.transaction()
+    def my_transaction() -> None:
+        pass
+
+    @DBOS.transaction()
+    def my_transaction() -> None:
+        pass
+
+    assert (
+        "Duplicate registration of function 'test_duplicate_registration.<locals>.my_transaction'"
+        in caplog.text
+    )
+
+    @DBOS.step()
+    def my_step() -> None:
+        pass
+
+    @DBOS.step()
+    def my_step() -> None:
+        pass
+
+    assert (
+        "Duplicate registration of function 'test_duplicate_registration.<locals>.my_step'"
+        in caplog.text
+    )
+
+    @DBOS.workflow()
+    def my_workflow() -> None:
+        my_step()
+        my_transaction()
+
+    @DBOS.workflow()
+    def my_workflow() -> None:
+        my_step()
+        my_transaction()
+
+    assert (
+        "Duplicate registration of function 'test_duplicate_registration.<locals>.my_workflow'"
+        in caplog.text
+    )
+
+    DBOS.destroy()
+    DBOS(config=config)
+    DBOS.launch()
+    assert "Duplicate registration of function 'temp_send_workflow'" not in caplog.text
+
+    # Reset logging
+    logging.getLogger("dbos").propagate = original_propagate
+
+
 def test_app_version(config: ConfigFile) -> None:
     def is_hex(s: str) -> bool:
         return all(c in "0123456789abcdefABCDEF" for c in s)