test: add GetAPIResource for floatingip, primaryip and loadbalancer

Author: jooola

internal/floatingip/testing.go

@@ -6,9 +6,18 @@ import (
 	"testing"
 
 	"github.com/hetznercloud/hcloud-go/v2/hcloud"
+	"github.com/hetznercloud/terraform-provider-hcloud/internal/testsupport"
 	"github.com/hetznercloud/terraform-provider-hcloud/internal/testtemplate"
 )
 
+// GetAPIResource returns a [testsupport.GetAPIResourceFunc] for [hcloud.FloatingIP].
+func GetAPIResource() testsupport.GetAPIResourceFunc[hcloud.FloatingIP] {
+	return func(c *hcloud.Client, attrs map[string]string) (*hcloud.FloatingIP, error) {
+		result, _, err := c.FloatingIP.Get(context.Background(), attrs["id"])
+		return result, err
+	}
+}
+
 // ByID returns a function that obtains a Floating IP by its ID.
 func ByID(t *testing.T, fl *hcloud.FloatingIP) func(*hcloud.Client, int64) bool {
 	return func(c *hcloud.Client, id int64) bool {

internal/loadbalancer/testing.go

@@ -6,9 +6,18 @@ import (
 	"testing"
 
 	"github.com/hetznercloud/hcloud-go/v2/hcloud"
+	"github.com/hetznercloud/terraform-provider-hcloud/internal/testsupport"
 	"github.com/hetznercloud/terraform-provider-hcloud/internal/testtemplate"
 )
 
+// GetAPIResource returns a [testsupport.GetAPIResourceFunc] for [hcloud.LoadBalancer].
+func GetAPIResource() testsupport.GetAPIResourceFunc[hcloud.LoadBalancer] {
+	return func(c *hcloud.Client, attrs map[string]string) (*hcloud.LoadBalancer, error) {
+		result, _, err := c.LoadBalancer.Get(context.Background(), attrs["id"])
+		return result, err
+	}
+}
+
 // ByID returns a function that obtains a loadbalancer by its ID.
 func ByID(t *testing.T, lb *hcloud.LoadBalancer) func(*hcloud.Client, int64) bool {
 	return func(c *hcloud.Client, id int64) bool {

internal/primaryip/testing.go

@@ -6,9 +6,18 @@ import (
 	"testing"
 
 	"github.com/hetznercloud/hcloud-go/v2/hcloud"
+	"github.com/hetznercloud/terraform-provider-hcloud/internal/testsupport"
 	"github.com/hetznercloud/terraform-provider-hcloud/internal/testtemplate"
 )
 
+// GetAPIResource returns a [testsupport.GetAPIResourceFunc] for [hcloud.PrimaryIP].
+func GetAPIResource() testsupport.GetAPIResourceFunc[hcloud.PrimaryIP] {
+	return func(c *hcloud.Client, attrs map[string]string) (*hcloud.PrimaryIP, error) {
+		result, _, err := c.PrimaryIP.Get(context.Background(), attrs["id"])
+		return result, err
+	}
+}
+
 // ByID returns a function that obtains a primary IP by its ID.
 func ByID(t *testing.T, fl *hcloud.PrimaryIP) func(*hcloud.Client, int64) bool {
 	return func(c *hcloud.Client, id int64) bool {