Merge pull request e-breuninger#122 from madnutter56/feature/cluster_type_site_data_source

add data source for cluster type and site

Author: fbreckle

netbox/data_source_netbox_cluster_type.go

@@ -0,0 +1,53 @@
+package netbox
+
+import (
+	"errors"
+	"strconv"
+
+	"github.com/fbreckle/go-netbox/netbox/client"
+	"github.com/fbreckle/go-netbox/netbox/client/virtualization"
+	"github.com/hashicorp/terraform-plugin-sdk/v2/helper/schema"
+)
+
+func dataSourceNetboxClusterType() *schema.Resource {
+	return &schema.Resource{
+		Read: dataSourceNetboxClusterTypeRead,
+		Schema: map[string]*schema.Schema{
+			"cluster_type_id": &schema.Schema{
+				Type:     schema.TypeInt,
+				Computed: true,
+			},
+			"name": &schema.Schema{
+				Type:     schema.TypeString,
+				Required: true,
+			},
+		},
+	}
+}
+
+func dataSourceNetboxClusterTypeRead(d *schema.ResourceData, m interface{}) error {
+	api := m.(*client.NetBoxAPI)
+
+	name := d.Get("name").(string)
+	params := virtualization.NewVirtualizationClusterTypesListParams()
+	params.Name = &name
+	limit := int64(2) // Limit of 2 is enough
+	params.Limit = &limit
+
+	res, err := api.Virtualization.VirtualizationClusterTypesList(params, nil)
+	if err != nil {
+		return err
+	}
+
+	if *res.GetPayload().Count > int64(1) {
+		return errors.New("More than one result. Specify a more narrow filter")
+	}
+	if *res.GetPayload().Count == int64(0) {
+		return errors.New("No result")
+	}
+	result := res.GetPayload().Results[0]
+	d.Set("cluster_type_id", result.ID)
+	d.SetId(strconv.FormatInt(result.ID, 10))
+	d.Set("name", result.Name)
+	return nil
+}

netbox/data_source_netbox_cluster_type_test.go

@@ -0,0 +1,34 @@
+package netbox
+
+import (
+	"fmt"
+	"testing"
+
+	"github.com/hashicorp/terraform-plugin-sdk/v2/helper/resource"
+)
+
+func TestAccNetboxClusterTypeDataSource_basic(t *testing.T) {
+
+	testSlug := "clstrtyp_ds_basic"
+	testName := testAccGetTestName(testSlug)
+	resource.ParallelTest(t, resource.TestCase{
+		Providers: testAccProviders,
+		Steps: []resource.TestStep{
+			{
+				Config: fmt.Sprintf(`
+resource "netbox_cluster_type" "test" {
+  name = "%[1]s"
+}
+
+data "netbox_cluster_type" "test" {
+  depends_on = [netbox_cluster_type.test]
+  name = "%[1]s"
+}`, testName),
+				Check: resource.ComposeTestCheckFunc(
+					resource.TestCheckResourceAttrPair("data.netbox_cluster_type.test", "cluster_type_id", "netbox_cluster_type.test", "id"),
+					resource.TestCheckResourceAttrPair("data.netbox_cluster_type.test", "id", "netbox_cluster_type.test", "id"),
+				),
+			},
+		},
+	})
+}

netbox/data_source_netbox_site.go

@@ -0,0 +1,53 @@
+package netbox
+
+import (
+	"errors"
+	"strconv"
+
+	"github.com/fbreckle/go-netbox/netbox/client"
+	"github.com/fbreckle/go-netbox/netbox/client/dcim"
+	"github.com/hashicorp/terraform-plugin-sdk/v2/helper/schema"
+)
+
+func dataSourceNetboxSite() *schema.Resource {
+	return &schema.Resource{
+		Read: dataSourceNetboxSiteRead,
+		Schema: map[string]*schema.Schema{
+			"site_id": &schema.Schema{
+				Type:     schema.TypeInt,
+				Computed: true,
+			},
+			"name": &schema.Schema{
+				Type:     schema.TypeString,
+				Required: true,
+			},
+		},
+	}
+}
+
+func dataSourceNetboxSiteRead(d *schema.ResourceData, m interface{}) error {
+	api := m.(*client.NetBoxAPI)
+
+	name := d.Get("name").(string)
+	params := dcim.NewDcimSitesListParams()
+	params.Name = &name
+	limit := int64(2) // Limit of 2 is enough
+	params.Limit = &limit
+
+	res, err := api.Dcim.DcimSitesList(params, nil)
+	if err != nil {
+		return err
+	}
+
+	if *res.GetPayload().Count > int64(1) {
+		return errors.New("More than one result. Specify a more narrow filter")
+	}
+	if *res.GetPayload().Count == int64(0) {
+		return errors.New("No result")
+	}
+	result := res.GetPayload().Results[0]
+	d.Set("site_id", result.ID)
+	d.SetId(strconv.FormatInt(result.ID, 10))
+	d.Set("name", result.Name)
+	return nil
+}

netbox/data_source_netbox_site_test.go

@@ -0,0 +1,35 @@
+package netbox
+
+import (
+	"fmt"
+	"testing"
+
+	"github.com/hashicorp/terraform-plugin-sdk/v2/helper/resource"
+)
+
+func TestAccNetboxSiteDataSource_basic(t *testing.T) {
+
+	testSlug := "site_ds_basic"
+	testName := testAccGetTestName(testSlug)
+	resource.ParallelTest(t, resource.TestCase{
+		Providers: testAccProviders,
+		Steps: []resource.TestStep{
+			{
+				Config: fmt.Sprintf(`
+resource "netbox_site" "test" {
+  name = "%[1]s"
+  status = "active"
+}
+data "netbox_site" "test" {
+  depends_on = [netbox_site.test]
+  name = "%[1]s"
+}`, testName),
+				Check: resource.ComposeTestCheckFunc(
+					resource.TestCheckResourceAttrPair("data.netbox_site.test", "status", "netbox_site.test", "active"),
+					resource.TestCheckResourceAttrPair("data.netbox_site.test", "site_id", "netbox_site.test", "id"),
+					resource.TestCheckResourceAttrPair("data.netbox_site.test", "id", "netbox_site.test", "id"),
+				),
+			},
+		},
+	})
+}

netbox/provider.go

@@ -40,12 +40,14 @@ func Provider() *schema.Provider {
 		DataSourcesMap: map[string]*schema.Resource{
 			"netbox_cluster":          dataSourceNetboxCluster(),
 			"netbox_cluster_group":    dataSourceNetboxClusterGroup(),
+			"netbox_cluster_type":     dataSourceNetboxClusterType(),
 			"netbox_tenant":           dataSourceNetboxTenant(),
 			"netbox_tenant_group":     dataSourceNetboxTenantGroup(),
 			"netbox_vrf":              dataSourceNetboxVrf(),
 			"netbox_platform":         dataSourceNetboxPlatform(),
 			"netbox_prefix":           dataSourceNetboxPrefix(),
 			"netbox_device_role":      dataSourceNetboxDeviceRole(),
+			"netbox_site":             dataSourceNetboxSite(),
 			"netbox_tag":              dataSourceNetboxTag(),
 			"netbox_virtual_machines": dataSourceNetboxVirtualMachine(),
 			"netbox_interfaces":       dataSourceNetboxInterfaces(),