provider/aws: Use helper methods for checking partition

hashicorp · May 9, 2017 · 0285407 · 0285407
1 parent f862d73
commit 0285407
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/resource_aws_cloudwatch_log_group.go b/resource_aws_cloudwatch_log_group.go
@@ -109,7 +109,7 @@ func resourceAwsCloudWatchLogGroupRead(d *schema.ResourceData, meta interface{})
 		d.Set("retention_in_days", lg.RetentionInDays)
 	}
 
-	if meta.(*AWSClient).partition != "aws-us-gov" {
+	if meta.(*AWSClient).IsChinaCloud() || meta.(*AWSClient).IsGovCloud() {
 		tags, err := flattenCloudWatchTags(d, conn)
 		if err != nil {
 			return err
@@ -172,7 +172,9 @@ func resourceAwsCloudWatchLogGroupUpdate(d *schema.ResourceData, meta interface{
 		}
 	}
 
-	if meta.(*AWSClient).partition != "aws-us-gov" && d.HasChange("tags") {
+	restricted := meta.(*AWSClient).IsChinaCloud() || meta.(*AWSClient).IsGovCloud()
+
+	if !restricted && d.HasChange("tags") {
 		oraw, nraw := d.GetChange("tags")
 		o := oraw.(map[string]interface{})
 		n := nraw.(map[string]interface{})