From e2d994b354dedeaee199a21badcdb291264d2637 Mon Sep 17 00:00:00 2001
From: Josh Matthews <josh@joshmatthews.net>
Date: Thu, 2 May 2019 10:45:50 -0400
Subject: [PATCH] Support features in generic-worker tasks.

---
 etc/taskcluster/decisionlib.py | 15 +++++++++++++--
 1 file changed, 13 insertions(+), 2 deletions(-)

diff --git a/etc/taskcluster/decisionlib.py b/etc/taskcluster/decisionlib.py
index 6ba349e8d0d..82d6dcacced 100644
--- a/etc/taskcluster/decisionlib.py
+++ b/etc/taskcluster/decisionlib.py
@@ -286,6 +286,7 @@ class GenericWorkerTask(Task):
         super().__init__(*args, **kwargs)
         self.max_run_time_minutes = 30
         self.env = {}
+        self.features = {}
         self.mounts = []
         self.artifacts = []
 
@@ -314,6 +315,7 @@ class GenericWorkerTask(Task):
             worker_payload,
             env=self.env,
             mounts=self.mounts,
+            features=self.features,
             artifacts=[
                 {
                     "type": type_,
@@ -337,6 +339,15 @@ class GenericWorkerTask(Task):
         self.artifacts.extend((type, path) for path in paths)
         return self
 
+    def with_features(self, *names):
+        """
+        Enable the given `generic-worker` features.
+
+        <https://github.com/taskcluster/generic-worker/blob/master/native_windows.yml>
+        """
+        self.features.update({name: True for name in names})
+        return self
+
     def _mount_content(self, url_or_artifact_name, task_id, sha256):
         if task_id:
             content = {"taskId": task_id, "artifact": url_or_artifact_name}
@@ -693,9 +704,9 @@ class DockerWorkerTask(UnixTaskMixin, Task):
 
     def with_features(self, *names):
         """
-        Enable the give `docker-worker` features.
+        Enable the given `docker-worker` features.
 
-        <https://docs.taskcluster.net/docs/reference/workers/docker-worker/docs/features>
+        <https://github.com/taskcluster/docker-worker/blob/master/docs/features.md>
         """
         self.features.update({name: True for name in names})
         return self