From 711972f8c203bb039ccf4a6f6eb765907eddc4ce Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Jos=C3=A9=20Antonio=20Reyes?= <imjaroiswebdev@gmail.com>
Date: Fri, 5 Apr 2024 14:18:05 -0300
Subject: [PATCH] remove options for querying eps for 403 errors

---
 .../resource_pagerduty_escalation_policy.go   | 41 ++++++++++++++-----
 1 file changed, 31 insertions(+), 10 deletions(-)

diff --git a/pagerduty/resource_pagerduty_escalation_policy.go b/pagerduty/resource_pagerduty_escalation_policy.go
index ccdd8648d..a75a9a910 100644
--- a/pagerduty/resource_pagerduty_escalation_policy.go
+++ b/pagerduty/resource_pagerduty_escalation_policy.go
@@ -172,10 +172,23 @@ func fetchEscalationPolicy(d *schema.ResourceData, meta interface{}, errCallback
 
 	o := &pagerduty.GetEscalationPolicyOptions{Includes: []string{"escalation_rule_assignment_strategies"}}
 
+	var escalationPolicyFirstAttempt *pagerduty.EscalationPolicy
+
+	escalationPolicyFirstAttempt, _, err = client.EscalationPolicies.Get(d.Id(), o)
+	if err != nil && isErrCode(err, http.StatusForbidden) {
+		// Removing the inclusion of escalation_rule_assignment_strategies for
+		// accounts wihtout the required entitlements.
+		o = nil
+	}
+
+	if err == nil && escalationPolicyFirstAttempt == nil {
+		return setResourceEPProps(d, escalationPolicyFirstAttempt)
+	}
+
 	return retry.Retry(5*time.Minute, func() *retry.RetryError {
 		escalationPolicy, _, err := client.EscalationPolicies.Get(d.Id(), o)
 		if err != nil {
-			if isErrCode(err, http.StatusBadRequest) {
+			if isErrCode(err, http.StatusBadRequest) || isErrCode(err, http.StatusForbidden) {
 				return retry.NonRetryableError(err)
 			}
 
@@ -189,15 +202,8 @@ func fetchEscalationPolicy(d *schema.ResourceData, meta interface{}, errCallback
 			return nil
 		}
 
-		d.Set("name", escalationPolicy.Name)
-		d.Set("description", escalationPolicy.Description)
-		d.Set("num_loops", escalationPolicy.NumLoops)
-
-		if err := d.Set("teams", flattenTeams(escalationPolicy.Teams)); err != nil {
-			return retry.NonRetryableError(fmt.Errorf("error setting teams: %s", err))
-		}
-
-		if err := d.Set("rule", flattenEscalationRules(escalationPolicy.EscalationRules)); err != nil {
+		err = setResourceEPProps(d, escalationPolicy)
+		if err != nil {
 			return retry.NonRetryableError(err)
 		}
 
@@ -205,6 +211,21 @@ func fetchEscalationPolicy(d *schema.ResourceData, meta interface{}, errCallback
 	})
 }
 
+func setResourceEPProps(d *schema.ResourceData, escalationPolicy *pagerduty.EscalationPolicy) error {
+	d.Set("name", escalationPolicy.Name)
+	d.Set("description", escalationPolicy.Description)
+	d.Set("num_loops", escalationPolicy.NumLoops)
+
+	if err := d.Set("teams", flattenTeams(escalationPolicy.Teams)); err != nil {
+		return fmt.Errorf("error setting teams: %s", err)
+	}
+
+	if err := d.Set("rule", flattenEscalationRules(escalationPolicy.EscalationRules)); err != nil {
+		return err
+	}
+	return nil
+}
+
 func resourcePagerDutyEscalationPolicyUpdate(d *schema.ResourceData, meta interface{}) error {
 	client, err := meta.(*Config).Client()
 	if err != nil {