fixes #345: introduce data-source for hosts in cluster

ovirt/data_source_ovirt_cluster_hosts.go

@@ -0,0 +1,74 @@
+package ovirt
+
+import (
+	"context"
+
+	"github.com/hashicorp/terraform-plugin-sdk/v2/diag"
+	"github.com/hashicorp/terraform-plugin-sdk/v2/helper/schema"
+)
+
+func (p *provider) clusterHostsDataSource() *schema.Resource {
+	return &schema.Resource{
+		ReadContext: p.clusterHostsDataSourceRead,
+		Schema: map[string]*schema.Schema{
+			"cluster_id": {
+				Type:             schema.TypeString,
+				Required:         true,
+				Description:      "oVirt cluster ID in the Data Center.",
+				ValidateDiagFunc: validateUUID,
+			},
+			"hosts": {
+				Type:     schema.TypeSet,
+				Computed: true,
+				Elem: &schema.Resource{
+					Schema: map[string]*schema.Schema{
+						"id": {
+							Type:        schema.TypeString,
+							Computed:    true,
+							Description: "ID of the host.",
+						},
+						"status": {
+							Type:        schema.TypeString,
+							Computed:    true,
+							Description: "status of the host.",
+						},
+					},
+				},
+			},
+		},
+		Description: `A set of all hosts of a Cluster.`,
+	}
+}
+
+func (p *provider) clusterHostsDataSourceRead(
+	ctx context.Context,
+	data *schema.ResourceData,
+	_ interface{},
+) diag.Diagnostics {
+	client := p.client.WithContext(ctx)
+	clusterID := data.Get("cluster_id").(string)
+	allHosts, err := client.ListHosts()
+
+	if err != nil {
+		return errorToDiags("list all hosts", err)
+	}
+
+	hosts := make([]map[string]interface{}, 0)
+
+	for _, host := range allHosts {
+		if string(host.ClusterID()) == clusterID {
+			hostMap := make(map[string]interface{}, 0)
+			hostMap["id"] = host.ID()
+			hostMap["status"] = host.Status()
+			hosts = append(hosts, hostMap)
+		}
+	}
+
+	if err := data.Set("hosts", hosts); err != nil {
+		return errorToDiags("set hosts", err)
+	}
+
+	data.SetId(clusterID)
+
+	return nil
+}

ovirt/data_source_ovirt_cluster_hosts_test.go

@@ -0,0 +1,51 @@
+package ovirt
+
+import (
+	"fmt"
+	"testing"
+
+	"github.com/hashicorp/terraform-plugin-sdk/v2/helper/resource"
+	"github.com/hashicorp/terraform-plugin-sdk/v2/terraform"
+)
+
+func TestClusterHostsDataSource(t *testing.T) {
+	p := newProvider(newTestLogger(t))
+
+	config :=
+		fmt.Sprintf(`
+			provider "ovirt" {
+				mock = true
+			}
+
+			data "ovirt_cluster_hosts" "list" {
+				cluster_id = "%s"
+			}
+
+			output "hosts_list" {
+				value = data.ovirt_cluster_hosts.list
+			}`,
+			p.getTestHelper().GetClusterID())
+
+	resource.UnitTest(t, resource.TestCase{
+		ProviderFactories: p.getProviderFactories(),
+		Steps: []resource.TestStep{
+			{
+				Config: config,
+				Check: resource.ComposeTestCheckFunc(
+					func(s *terraform.State) error {
+						v := s.RootModule().Outputs["hosts_list"].Value.(map[string]interface{})
+						hosts, ok := v["hosts"]
+						if !ok {
+							return fmt.Errorf("missing key 'hosts' in output")
+						}
+						hostSize := len(hosts.([]interface{}))
+						if hostSize != 1 {
+							return fmt.Errorf("expected 1 hosts, but got only %d", hostSize)
+						}
+						return nil
+					},
+				),
+			},
+		},
+	})
+}

ovirt/provider.go

@@ -129,6 +129,7 @@ func (p *provider) getProvider() *schema.Provider {
 		},
 		DataSourcesMap: map[string]*schema.Resource{
 			"ovirt_disk_attachments": p.diskAttachmentsDataSource(),
+			"ovirt_cluster_hosts":    p.clusterHostsDataSource(),
 		},
 	}
 }