datasource_digitalocean_projects_test.go

package digitalocean

import (
	"fmt"
	"testing"

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

func TestAccDataSourceDigitalOceanProjects_Basic(t *testing.T) {
	prodProjectName := randomName("tf-acc-project-", 6)
	stagingProjectName := randomName("tf-acc-project-", 6)

	resourcesConfig := fmt.Sprintf(`
resource "digitalocean_project" "prod" {
	name = "%s"
	environment = "Production"
}

resource "digitalocean_project" "staging" {
	name = "%s"
	environment = "Staging"
}
`, prodProjectName, stagingProjectName)

	datasourcesConfig := fmt.Sprintf(`
data "digitalocean_projects" "prod" {
	filter {
      key = "environment"
      values = ["Production"]
    }
    filter {
      key = "is_default"
      values = ["false"]
    }
}

data "digitalocean_projects" "staging" {
	filter {
      key = "name"
      values = ["%s"]
    }
    filter {
      key = "is_default"
      values = ["false"]
    }
}

data "digitalocean_projects" "both" {
	filter {
      key = "environment"
      values = ["Production"]
    }
	filter {
      key = "name"
      values = ["%s"]
    }
}
`, stagingProjectName, stagingProjectName)
	resource.Test(t, resource.TestCase{
		PreCheck:          func() { testAccPreCheck(t) },
		ProviderFactories: testAccProviderFactories,
		CheckDestroy:      testAccCheckDigitalOceanProjectDestroy,
		Steps: []resource.TestStep{
			{
				Config: resourcesConfig,
			},
			{
				Config: resourcesConfig + datasourcesConfig,
				Check: resource.ComposeTestCheckFunc(
					resource.TestCheckResourceAttr("data.digitalocean_projects.prod", "projects.#", "1"),
					resource.TestCheckResourceAttr("data.digitalocean_projects.prod", "projects.0.name", prodProjectName),
					resource.TestCheckResourceAttr("data.digitalocean_projects.prod", "projects.0.environment", "Production"),
					resource.TestCheckResourceAttr("data.digitalocean_projects.staging", "projects.#", "1"),
					resource.TestCheckResourceAttr("data.digitalocean_projects.staging", "projects.0.name", stagingProjectName),
					resource.TestCheckResourceAttr("data.digitalocean_projects.staging", "projects.0.environment", "Staging"),
					resource.TestCheckResourceAttr("data.digitalocean_projects.both", "projects.#", "0"),
				),
			},
		},
	})
}