Check for Task metadata inputs in mocking system.

lsst · Aug 1, 2024 · 36fc00e · 36fc00e
1 parent d909c49
commit 36fc00e
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/doc/changes/DM-45536.bugfix.md b/doc/changes/DM-45536.bugfix.md
@@ -0,0 +1 @@
+Fix support for task metadata as inputs in the `PipelineTask` mocking system.
diff --git a/python/lsst/pipe/base/tests/mocks/_pipeline_task.py b/python/lsst/pipe/base/tests/mocks/_pipeline_task.py
@@ -425,6 +425,11 @@ def __init__(self, *, config: MockPipelineTaskConfig):
                 raise ValueError(
                     f"Unmocked dataset type {connection.name!r} cannot be used as an init-output."
                 )
+            elif connection.name.endswith("_metadata") and connection.storageClass == "TaskMetadata":
+                # Task metadata does not use a mock storage class, because it's
+                # written by the system, but it does end up with the _mock_*
+                # prefix because the task label does.
+                connection = dataclasses.replace(connection, name=get_mock_name(connection.name))
             setattr(self, name, connection)
 
     def getSpatialBoundsConnections(self) -> Iterable[str]: