88 lines
2.2 KiB
Go
88 lines
2.2 KiB
Go
|
package grafana
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
"regexp"
|
||
|
|
"strconv"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
gapi "github.com/apparentlymart/go-grafana-api"
|
||
|
|
|
||
|
|
"github.com/hashicorp/terraform/helper/resource"
|
||
|
|
"github.com/hashicorp/terraform/terraform"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestAccDataSource_basic(t *testing.T) {
|
||
|
|
var dataSource gapi.DataSource
|
||
|
|
|
||
|
|
resource.Test(t, resource.TestCase{
|
||
|
|
PreCheck: func() { testAccPreCheck(t) },
|
||
|
|
Providers: testAccProviders,
|
||
|
|
CheckDestroy: testAccDataSourceCheckDestroy(&dataSource),
|
||
|
|
Steps: []resource.TestStep{
|
||
|
|
resource.TestStep{
|
||
|
|
Config: testAccDataSourceConfig_basic,
|
||
|
|
Check: resource.ComposeTestCheckFunc(
|
||
|
|
testAccDataSourceCheckExists("grafana_data_source.test", &dataSource),
|
||
|
|
resource.TestCheckResourceAttr(
|
||
|
|
"grafana_data_source.test", "type", "influxdb",
|
||
|
|
),
|
||
|
|
resource.TestMatchResourceAttr(
|
||
|
|
"grafana_data_source.test", "id", regexp.MustCompile(`\d+`),
|
||
|
|
),
|
||
|
|
),
|
||
|
|
},
|
||
|
|
},
|
||
|
|
})
|
||
|
|
}
|
||
|
|
|
||
|
|
func testAccDataSourceCheckExists(rn string, dataSource *gapi.DataSource) resource.TestCheckFunc {
|
||
|
|
return func(s *terraform.State) error {
|
||
|
|
rs, ok := s.RootModule().Resources[rn]
|
||
|
|
if !ok {
|
||
|
|
return fmt.Errorf("resource not found: %s", rn)
|
||
|
|
}
|
||
|
|
|
||
|
|
if rs.Primary.ID == "" {
|
||
|
|
return fmt.Errorf("resource id not set")
|
||
|
|
}
|
||
|
|
|
||
|
|
id, err := strconv.ParseInt(rs.Primary.ID, 10, 64)
|
||
|
|
if err != nil {
|
||
|
|
return fmt.Errorf("resource id is malformed")
|
||
|
|
}
|
||
|
|
|
||
|
|
client := testAccProvider.Meta().(*gapi.Client)
|
||
|
|
gotDataSource, err := client.DataSource(id)
|
||
|
|
if err != nil {
|
||
|
|
return fmt.Errorf("error getting data source: %s", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
*dataSource = *gotDataSource
|
||
|
|
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func testAccDataSourceCheckDestroy(dataSource *gapi.DataSource) resource.TestCheckFunc {
|
||
|
|
return func(s *terraform.State) error {
|
||
|
|
client := testAccProvider.Meta().(*gapi.Client)
|
||
|
|
_, err := client.DataSource(dataSource.Id)
|
||
|
|
if err == nil {
|
||
|
|
return fmt.Errorf("data source still exists")
|
||
|
|
}
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
const testAccDataSourceConfig_basic = `
|
||
|
|
resource "grafana_data_source" "test" {
|
||
|
|
type = "influxdb"
|
||
|
|
name = "terraform-acc-test"
|
||
|
|
database_name = "terraform-acc-test"
|
||
|
|
url = "http://terraform-acc-test.invalid/"
|
||
|
|
username = "terraform_user"
|
||
|
|
password = "terraform_password"
|
||
|
|
}
|
||
|
|
`
|