Added NestedQuery handling to PostCreateSetComputedFields (#13692) (#22387)

[upstream:53765df79fccbb2714c4593a4470af4e88480153]

Signed-off-by: Modular Magician <[email protected]>

Author: modular-magician

google/services/bigqueryreservation/resource_bigquery_reservation_assignment.go

@@ -412,6 +412,16 @@ func resourceBigqueryReservationReservationAssignmentFindNestedObjectInList(d *s
 }
 func resourceBigqueryReservationReservationAssignmentPostCreateSetComputedFields(d *schema.ResourceData, meta interface{}, res map[string]interface{}) error {
 	config := meta.(*transport_tpg.Config)
+	if _, ok := res["assignments"]; ok {
+		res, err := flattenNestedBigqueryReservationReservationAssignment(d, meta, res)
+		if err != nil {
+			return fmt.Errorf("Error getting nested object from operation response: %s", err)
+		}
+		if res == nil {
+			// Object isn't there any more - remove it from the state.
+			return fmt.Errorf("Error decoding response from operation, could not find nested object")
+		}
+	}
 	if err := d.Set("name", flattenNestedBigqueryReservationReservationAssignmentName(res["name"], d, config)); err != nil {
 		return fmt.Errorf(`Error setting computed identity field "name": %s`, err)
 	}

google/services/datacatalog/resource_data_catalog_tag.go

@@ -695,6 +695,16 @@ func resourceDataCatalogTagFindNestedObjectInList(d *schema.ResourceData, meta i
 }
 func resourceDataCatalogTagPostCreateSetComputedFields(d *schema.ResourceData, meta interface{}, res map[string]interface{}) error {
 	config := meta.(*transport_tpg.Config)
+	if _, ok := res["tags"]; ok {
+		res, err := flattenNestedDataCatalogTag(d, meta, res)
+		if err != nil {
+			return fmt.Errorf("Error getting nested object from operation response: %s", err)
+		}
+		if res == nil {
+			// Object isn't there any more - remove it from the state.
+			return fmt.Errorf("Error decoding response from operation, could not find nested object")
+		}
+	}
 	if err := d.Set("name", flattenNestedDataCatalogTagName(res["name"], d, config)); err != nil {
 		return fmt.Errorf(`Error setting computed identity field "name": %s`, err)
 	}

google/services/resourcemanager/resource_resource_manager_lien.go

@@ -514,6 +514,16 @@ func resourceResourceManagerLienPostCreateSetComputedFields(d *schema.ResourceDa
 	if res == nil {
 		return fmt.Errorf("decoding response, could not find object")
 	}
+	if _, ok := res["liens"]; ok {
+		res, err := flattenNestedResourceManagerLien(d, meta, res)
+		if err != nil {
+			return fmt.Errorf("Error getting nested object from operation response: %s", err)
+		}
+		if res == nil {
+			// Object isn't there any more - remove it from the state.
+			return fmt.Errorf("Error decoding response from operation, could not find nested object")
+		}
+	}
 	if err := d.Set("name", flattenNestedResourceManagerLienName(res["name"], d, config)); err != nil {
 		return fmt.Errorf(`Error setting computed identity field "name": %s`, err)
 	}