From 787aaad8b26692d2868f8342d09e5dd97a988bf9 Mon Sep 17 00:00:00 2001
From: Ben Hearsum <ben@mozilla.com>
Date: Thu, 5 Jun 2025 16:21:09 -0400
Subject: [PATCH] fix: allow taskref's in upload data destinations

I realized shortly after merging #102 that this is needed :(
---
 src/mozilla_taskgraph/worker_types.py | 2 +-
 test/test_worker_types.py             | 2 ++
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/src/mozilla_taskgraph/worker_types.py b/src/mozilla_taskgraph/worker_types.py
index ce37271..3bac674 100644
--- a/src/mozilla_taskgraph/worker_types.py
+++ b/src/mozilla_taskgraph/worker_types.py
@@ -478,7 +478,7 @@ def dash_to_underscore(obj):
             {
                 Required("data"): str,
                 Required("content-type"): str,
-                Required("destinations"): [str],
+                Required("destinations"): [taskref_or_string],
             },
         ],
         Optional("dry-run"): bool,
diff --git a/test/test_worker_types.py b/test/test_worker_types.py
index 96912fe..f375dd9 100644
--- a/test/test_worker_types.py
+++ b/test/test_worker_types.py
@@ -1129,6 +1129,7 @@ def test_beetmover_upload_data(build_payload):
                 "destinations": [
                     "foo.txt",
                     "bar.txt",
+                    {"task-reference": "<foo>"},
                 ],
             },
         ],
@@ -1146,6 +1147,7 @@ def test_beetmover_upload_data(build_payload):
                     "destinations": [
                         "foo.txt",
                         "bar.txt",
+                        {"task-reference": "<foo>"},
                     ],
                 },
             ],