Attempt to read team mapping from AuthN response

DataDog · Nov 20, 2024 · 3d637e3 · 3d637e3
1 parent 1b6c3dd
commit 3d637e3
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/datadog/resource_datadog_authn_mapping.go b/datadog/resource_datadog_authn_mapping.go
@@ -145,7 +145,9 @@ func updateAuthNMappingState(d *schema.ResourceData, authNMapping *datadogV2.Aut
 	authNMappingAttributes := authNMapping.GetAttributes()
 	authNMappingRelations := authNMapping.GetRelationships()
 	authNMappingRoleRelation := authNMappingRelations.GetRole()
+	authNMappingTeamRelation := authNMappingRelations.GetTeam()
 	authNRole := authNMappingRoleRelation.GetData()
+	authNTeam := authNMappingTeamRelation.GetData()
 
 	if err := d.Set("key", authNMappingAttributes.GetAttributeKey()); err != nil {
 		return diag.FromErr(err)
@@ -158,6 +160,11 @@ func updateAuthNMappingState(d *schema.ResourceData, authNMapping *datadogV2.Aut
 	if err := d.Set("role", authNRole.GetId()); err != nil {
 		return diag.FromErr(err)
 	}
+
+	if err := d.Set("team", authNTeam.GetId()); err != nil {
+		return diag.FromErr(err)
+	}
+
 	return nil
 }