Add identity creation in the create as well

remyleone · remyleone · commit bb9db15cc7df · 2025-08-25T18:09:45.000+02:00
diff --git a/internal/services/block/snapshot.go b/internal/services/block/snapshot.go
@@ -151,6 +151,10 @@ func ResourceBlockSnapshotCreate(ctx context.Context, d *schema.ResourceData, m
 	}
 
 	d.SetId(zonal.NewIDString(zone, snapshot.ID))
+	diags := applySnapshotIdentity(d, snapshot.ID, zone)
+	if diags != nil {
+		return diags
+	}
 
 	_, err = waitForBlockSnapshot(ctx, api, zone, snapshot.ID, d.Timeout(schema.TimeoutCreate))
 	if err != nil {
diff --git a/internal/services/block/volume.go b/internal/services/block/volume.go
@@ -156,6 +156,10 @@ func ResourceBlockVolumeCreate(ctx context.Context, d *schema.ResourceData, m an
 	}
 
 	d.SetId(zonal.NewIDString(zone, volume.ID))
+	diags := applyVolumeIdentity(d, volume.ID, zone)
+	if diags != nil {
+		return diags
+	}
 
 	_, err = waitForBlockVolume(ctx, api.BlockAPI, zone, volume.ID, d.Timeout(schema.TimeoutCreate))
 	if err != nil {

Original file line number	Diff line number	Diff line change
`@@ -151,6 +151,10 @@ func ResourceBlockSnapshotCreate(ctx context.Context, d *schema.ResourceData, m`
`151`	`151`	`}`
`152`	`152`
`153`	`153`	`d.SetId(zonal.NewIDString(zone, snapshot.ID))`
	`154`	`+ diags := applySnapshotIdentity(d, snapshot.ID, zone)`
	`155`	`+ if diags != nil {`
	`156`	`+ return diags`
	`157`	`+ }`
`154`	`158`
`155`	`159`	`_, err = waitForBlockSnapshot(ctx, api, zone, snapshot.ID, d.Timeout(schema.TimeoutCreate))`
`156`	`160`	`if err != nil {`
Original file line number	Diff line number	Diff line change
`@@ -156,6 +156,10 @@ func ResourceBlockVolumeCreate(ctx context.Context, d *schema.ResourceData, m an`
`156`	`156`	`}`
`157`	`157`
`158`	`158`	`d.SetId(zonal.NewIDString(zone, volume.ID))`
	`159`	`+ diags := applyVolumeIdentity(d, volume.ID, zone)`
	`160`	`+ if diags != nil {`
	`161`	`+ return diags`
	`162`	`+ }`
`159`	`163`
`160`	`164`	`_, err = waitForBlockVolume(ctx, api.BlockAPI, zone, volume.ID, d.Timeout(schema.TimeoutCreate))`
`161`	`165`	`if err != nil {`