Get test coverage above 61% for python 3.10

Signed-off-by: Julie Langmann <12345+jlangm@users.noreply.github.com>
aws-deadline · Sep 15, 2023 · 2089356 · 2089356
1 parent 1a48167
commit 2089356
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/test/deadline_adaptor_for_nuke/unit/NukeClient/test_nuke_handler.py b/test/deadline_adaptor_for_nuke/unit/NukeClient/test_nuke_handler.py
@@ -276,6 +276,19 @@ def test_set_write_nodes(self, write_nodes: List[MockNode], nukehandler: NukeHan
         # THEN
         assert nukehandler.write_nodes == write_nodes
 
+    def test_set_write_nodes_all_write_nodes(
+        self, write_nodes: List[MockNode], nukehandler: NukeHandler
+    ):
+        # GIVEN
+        all_write_nodes = ["All Write Nodes"]
+        data = {"write_nodes": all_write_nodes}
+
+        # WHEN
+        nukehandler.set_write_nodes(data)
+
+        # THEN
+        assert nukehandler.write_nodes == write_nodes
+
     missing_nodes_params = [
         (
             SortedList(["these", "do", "not", "exist"]),