terraform-provider-openstack/openstack/data_source_openstack_compute_hypervisor_v2_test.go at master · rcreasey/terraform-provider-openstack

Repository navigation

package openstack

import (

"fmt"

"testing"

"github.com/hashicorp/terraform-plugin-sdk/v2/helper/resource"

"github.com/hashicorp/terraform-plugin-sdk/v2/terraform"

)

func testAccHypervisorDataSource() string {

return fmt.Sprintf(`

data "openstack_compute_hypervisor_v2" "host01" {

hostname = "%s"

}

`, osHypervisorEnvironment)

}

func TestAccComputeHypervisorV2DataSource(t *testing.T) {

resource.Test(t, resource.TestCase{

PreCheck: func() {

testAccPreCheckAdminOnly(t)

testAccPreCheckHypervisor(t)

ProviderFactories: testAccProviders,

Steps: []resource.TestStep{

{

Config: testAccHypervisorDataSource(),

Check: resource.ComposeTestCheckFunc(

testAccCheckComputeHypervisorV2DataSourceID("data.openstack_compute_hypervisor_v2.host01"),

resource.TestCheckResourceAttr("data.openstack_compute_hypervisor_v2.host01", "hostname", osHypervisorEnvironment),

})

}

func testAccCheckComputeHypervisorV2DataSourceID(n string) resource.TestCheckFunc {

return func(s *terraform.State) error {

rs, ok := s.RootModule().Resources[n]

if !ok {

return fmt.Errorf("Can't find data source: %s", n)

}

if rs.Primary.ID == "" {

return fmt.Errorf("Data source ID not set")

}

return nil

}