From cdbd565f579dae617a6b3cdcb44c8adba369133f Mon Sep 17 00:00:00 2001
From: Joshua Nitschke <joshuanits@gmail.com>
Date: Wed, 19 Feb 2025 23:04:32 +1030
Subject: [PATCH] test: add test case

---
 tests/unit/graph/test_nodes.py | 23 +++++++++++++++++++++++
 1 file changed, 23 insertions(+)

diff --git a/tests/unit/graph/test_nodes.py b/tests/unit/graph/test_nodes.py
index a0e0a8d7e56..2769cf33684 100644
--- a/tests/unit/graph/test_nodes.py
+++ b/tests/unit/graph/test_nodes.py
@@ -410,6 +410,22 @@ def parsed_node(self) -> ParsedNode:
             database=None,
         )
 
+    @pytest.fixture(scope="class")
+    def parsed_yml_node(self) -> ParsedNode:
+        return ParsedNode(
+            resource_type=NodeType.Model,
+            unique_id="model.test_package.test_name",
+            name="test_name",
+            package_name="test_package",
+            schema="test_schema",
+            alias="test_alias",
+            fqn=["models", "test_name"],
+            original_file_path="folder/test_original_file_path.yml",
+            checksum=FileHash.from_contents("checksum"),
+            path="test_original_file_path.yml/test_path.sql",
+            database=None,
+        )
+
     def test_get_target_write_path(self, parsed_node):
         write_path = parsed_node.get_target_write_path("target_path", "subdirectory")
         assert (
@@ -423,3 +439,10 @@ def test_get_target_write_path_split(self, parsed_node):
             write_path
             == "target_path/subdirectory/test_package/test_original_file_path/test_path/test_path_split.sql"
         )
+
+    def test_get_target_write_path_for_yml_node(self, parsed_yml_node):
+        write_path = parsed_yml_node.get_target_write_path("target_path", "subdirectory")
+        assert (
+            write_path
+            == "target_path/subdirectory/test_package/folder/test_original_file_path_yml/test_path.sql"
+        )