fix: Fixed URL Category resource READ function (#498)

Author: willguibr

CHANGELOG.md

@@ -1,5 +1,16 @@
 # Changelog
 
+## 4.6.1 (November, 20 2025)
+
+### Notes
+
+- Release date: **(November, 20  2025)**
+- Supported Terraform version: **v1.x**
+
+### Bug Fixes
+
+- [PR #498](https://github.com/zscaler/terraform-provider-zia/pull/498) - Fixed `zia_url_categories` resource READ function for better state refresh and rate limiting conservation. Also fixed issue with category_type `TLD_CATEGORY` not being set correctly in the state during creation.
+
 ## 4.6.0 (November, 19 2025)
 
 ### Notes
@@ -9,14 +20,14 @@
 
 ### Enhancements
 
-- [PR #495](https://github.com/zscaler/terraform-provider-zia/pull/495) - Added new data source `zia_file_type_categories`. This datasource can be referenced within the `zia_dlp_web_rules` in the attribute `file_type_categories`
-- [PR #495](https://github.com/zscaler/terraform-provider-zia/pull/495) - Added new data source and resource `zia_custom_file_types`
-- [PR #495](https://github.com/zscaler/terraform-provider-zia/pull/495) - Improved `zia_url_categories` resource READ function for better state refresh and rate limiting conservation.
+- [PR #496](https://github.com/zscaler/terraform-provider-zia/pull/496) - Added new data source `zia_file_type_categories`. This datasource can be referenced within the `zia_dlp_web_rules` in the attribute `file_type_categories`
+- [PR #496](https://github.com/zscaler/terraform-provider-zia/pull/496) - Added new data source and resource `zia_custom_file_types`
+- [PR #496](https://github.com/zscaler/terraform-provider-zia/pull/496) - Improved `zia_url_categories` resource READ function for better state refresh and rate limiting conservation.
 
 ### Bug Fixes
 
-- [PR #495](https://github.com/zscaler/terraform-provider-zia/pull/495) - Upgraded to [Zscaler SDK GO v3.8.6](https://github.com/zscaler/zscaler-sdk-go/releases/tag/v3.8.6) to fix pagination in some resources.
-- [PR #495](https://github.com/zscaler/terraform-provider-zia/pull/495) - Fixed drift with attribute `static_location_groups` in the resource `zia_location_management`
+- [PR #496](https://github.com/zscaler/terraform-provider-zia/pull/496) - Upgraded to [Zscaler SDK GO v3.8.6](https://github.com/zscaler/zscaler-sdk-go/releases/tag/v3.8.6) to fix pagination in some resources.
+- [PR #496](https://github.com/zscaler/terraform-provider-zia/pull/496) - Fixed drift with attribute `static_location_groups` in the resource `zia_location_management`
 
 ## 4.5.3 (October, 17 2025)
 

GNUmakefile

@@ -196,14 +196,14 @@ test\:integration\:zscalertwo:
 build13: GOOS=$(shell go env GOOS)
 build13: GOARCH=$(shell go env GOARCH)
 ifeq ($(OS),Windows_NT)  # is Windows_NT on XP, 2000, 7, Vista, 10...
-build13: DESTINATION=$(APPDATA)/terraform.d/plugins/$(ZIA_PROVIDER_NAMESPACE)/4.6.0/$(GOOS)_$(GOARCH)
+build13: DESTINATION=$(APPDATA)/terraform.d/plugins/$(ZIA_PROVIDER_NAMESPACE)/4.6.1/$(GOOS)_$(GOARCH)
 else
-build13: DESTINATION=$(HOME)/.terraform.d/plugins/$(ZIA_PROVIDER_NAMESPACE)/4.6.0/$(GOOS)_$(GOARCH)
+build13: DESTINATION=$(HOME)/.terraform.d/plugins/$(ZIA_PROVIDER_NAMESPACE)/4.6.1/$(GOOS)_$(GOARCH)
 endif
 build13: fmtcheck
 	@echo "==> Installing plugin to $(DESTINATION)"
 	@mkdir -p $(DESTINATION)
-	go build -o $(DESTINATION)/terraform-provider-zia_v4.6.0
+	go build -o $(DESTINATION)/terraform-provider-zia_v4.6.1
 
 coverage: test
 	@echo "✓ Opening coverage for unit tests ..."

docs/guides/release-notes.md

@@ -12,10 +12,21 @@ description: |-
 Track all ZIA Terraform provider's releases. New resources, features, and bug fixes will be tracked here.
 
 ---
-``Last updated: v4.6.0``
+``Last updated: v4.6.1``
 
 ---
 
+## 4.6.1 (November, 20 2025)
+
+### Notes
+
+- Release date: **(November, 20  2025)**
+- Supported Terraform version: **v1.x**
+
+### Bug Fixes
+
+- [PR #498](https://github.com/zscaler/terraform-provider-zia/pull/498) - Fixed `zia_url_categories` resource READ function for better state refresh and rate limiting conservation. Also fixed issue with category_type `TLD_CATEGORY` not being set correctly in the state during creation.
+
 ## 4.6.0 (November, 19 2025)
 
 ### Notes
@@ -25,14 +36,14 @@ Track all ZIA Terraform provider's releases. New resources, features, and bug fi
 
 ### Enhancements
 
-- [PR #495](https://github.com/zscaler/terraform-provider-zia/pull/495) - Added new data source `zia_file_type_categories`. This datasource can be referenced within the `zia_dlp_web_rules` in the attribute `file_type_categories`
-- [PR #495](https://github.com/zscaler/terraform-provider-zia/pull/495) - Added new data source and resource `zia_custom_file_types`
-- [PR #495](https://github.com/zscaler/terraform-provider-zia/pull/495) - Improved `zia_url_categories` resource READ function for better state refresh and rate limiting conservation.
+- [PR #496](https://github.com/zscaler/terraform-provider-zia/pull/496) - Added new data source `zia_file_type_categories`. This datasource can be referenced within the `zia_dlp_web_rules` in the attribute `file_type_categories`
+- [PR #496](https://github.com/zscaler/terraform-provider-zia/pull/496) - Added new data source and resource `zia_custom_file_types`
+- [PR #496](https://github.com/zscaler/terraform-provider-zia/pull/496) - Improved `zia_url_categories` resource READ function for better state refresh and rate limiting conservation.
 
 ### Bug Fixes
 
-- [PR #495](https://github.com/zscaler/terraform-provider-zia/pull/495) - Upgraded to [Zscaler SDK GO v3.8.6](https://github.com/zscaler/zscaler-sdk-go/releases/tag/v3.8.6) to fix pagination in some resources.
-- [PR #495](https://github.com/zscaler/terraform-provider-zia/pull/495) - Fixed drift with attribute `static_location_groups` in the resource `zia_location_management`
+- [PR #496](https://github.com/zscaler/terraform-provider-zia/pull/496) - Upgraded to [Zscaler SDK GO v3.8.6](https://github.com/zscaler/zscaler-sdk-go/releases/tag/v3.8.6) to fix pagination in some resources.
+- [PR #496](https://github.com/zscaler/terraform-provider-zia/pull/496) - Fixed drift with attribute `static_location_groups` in the resource `zia_location_management`
 
 ## 4.5.3 (October, 17 2025)
 

zia/common/version.go

@@ -1,6 +1,6 @@
 package common
 
-var version = "4.6.0"
+var version = "4.6.1"
 
 // Version returns version of provider
 func Version() string {

zia/resource_zia_url_categories.go

@@ -263,8 +263,9 @@ func resourceURLCategoriesRead(ctx context.Context, d *schema.ResourceData, meta
 	}
 
 	// Use GetAll() instead of Get() to reduce API calls during terraform refresh
-	// This fetches all custom URL categories in a single API call instead of one per category
-	allCategories, err := urlcategories.GetAll(ctx, service, true, false) // customOnly=true, includeOnlyUrlKeywordCounts=false
+	// customOnly=true to only retrieve custom categories (which are the ones managed by Terraform)
+	// includeOnlyUrlKeywordCounts=false to get full category details
+	allCategories, err := urlcategories.GetAll(ctx, service, true, false)
 	if err != nil {
 		return diag.FromErr(err)
 	}
@@ -278,11 +279,20 @@ func resourceURLCategoriesRead(ctx context.Context, d *schema.ResourceData, meta
 		}
 	}
 
-	// Category not found
+	// If not found in GetAll(), fall back to individual Get() call
+	// This handles newly created categories that may not be in the cached list yet
 	if resp == nil {
-		log.Printf("[WARN] Removing zia url category %s from state because it no longer exists in ZIA", d.Id())
-		d.SetId("")
-		return nil
+		log.Printf("[DEBUG] Category %s not found in GetAll() response, falling back to individual Get() call", id)
+		individualResp, err := urlcategories.Get(ctx, service, id)
+		if err != nil {
+			if respErr, ok := err.(*errorx.ErrorResponse); ok && respErr.IsObjectNotFound() {
+				log.Printf("[WARN] Removing url category %s from state because it no longer exists in ZIA", d.Id())
+				d.SetId("")
+				return nil
+			}
+			return diag.FromErr(err)
+		}
+		resp = individualResp
 	}
 
 	log.Printf("[INFO] Getting url category :\n%+v\n", resp)