From 9259111374667fedc8f9b96c0abf4ff55f588714 Mon Sep 17 00:00:00 2001
From: atulya-astronomer <atulya.tibrewal@astronomer.io>
Date: Thu, 7 Nov 2024 18:47:46 +0530
Subject: [PATCH] Updated test values for task instances

---
 .../compat/starship_compatability.py          | 30 ++++++++++++++-----
 1 file changed, 23 insertions(+), 7 deletions(-)

diff --git a/astronomer_starship/compat/starship_compatability.py b/astronomer_starship/compat/starship_compatability.py
index f4a1516..413e595 100644
--- a/astronomer_starship/compat/starship_compatability.py
+++ b/astronomer_starship/compat/starship_compatability.py
@@ -1041,6 +1041,11 @@ def task_instance_attrs(self):
         }
         return attrs
 
+    def task_instances_attrs(self):
+        attrs = super().task_instances_attrs()
+        attrs["task_instances"]["test_value"][0]["custom_operator_name"] = None
+        return attrs
+
 
 class StarshipAirflow28(StarshipAirflow27):
     """
@@ -1072,16 +1077,26 @@ def task_instance_attrs(self):
         attrs = super().task_instance_attrs()
         attrs["rendered_map_index"] = {
             "attr": "rendered_map_index",
-            "methods": [("POST", False)],
+            "methods": [("POST", True)],
             "test_value": "rendered_map_index",
         }
         attrs["task_display_name"] = {
             "attr": "task_display_name",
-            "methods": [("POST", False)],
+            "methods": [("POST", True)],
             "test_value": "task_display_name",
         }
         return attrs
 
+    def task_instances_attrs(self):
+        attrs = super().task_instances_attrs()
+        attrs["task_instances"]["test_value"][0][
+            "rendered_map_index"
+        ] = "rendered_map_index"
+        attrs["task_instances"]["test_value"][0][
+            "task_display_name"
+        ] = "task_display_name"
+        return attrs
+
 
 class StarshipAirflow210(StarshipAirflow28):
     """
@@ -1093,11 +1108,7 @@ class StarshipAirflow210(StarshipAirflow28):
 
     def task_instance_attrs(self):
         attrs = super().task_instance_attrs()
-        attrs["try_number"] = {
-            "attr": "try_number",
-            "methods": [("POST", True)],
-            "test_value": 0,
-        }
+        attrs["try_number"]["attr"] = "try_number"
         attrs["executor"] = {
             "attr": "executor",
             "methods": [("POST", True)],
@@ -1105,6 +1116,11 @@ def task_instance_attrs(self):
         }
         return attrs
 
+    def task_instances_attrs(self):
+        attrs = super().task_instances_attrs()
+        attrs["task_instances"]["test_value"][0]["executor"] = "executor"
+        return attrs
+
 
 class StarshipCompatabilityLayer:
     """StarshipCompatabilityLayer is a factory class that returns the correct StarshipAirflow class for a version