-
Notifications
You must be signed in to change notification settings - Fork 19
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge branch 'main' into awalden/migrate-azure-devops-repository-data…
…source
- Loading branch information
Showing
7 changed files
with
196 additions
and
71 deletions.
There are no files selected for viewing
79 changes: 79 additions & 0 deletions
79
pkg/framework/objects/azure_dev_ops_project/data_source.go
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,79 @@ | ||
package azure_dev_ops_project | ||
|
||
import ( | ||
"context" | ||
"fmt" | ||
"github.com/dbt-labs/terraform-provider-dbtcloud/pkg/dbt_cloud" | ||
"github.com/hashicorp/terraform-plugin-framework/datasource" | ||
"github.com/hashicorp/terraform-plugin-framework/types" | ||
) | ||
|
||
var ( | ||
_ datasource.DataSource = &azureDevOpsProjectDataSource{} | ||
_ datasource.DataSourceWithConfigure = &azureDevOpsProjectDataSource{} | ||
) | ||
|
||
func AzureDevOpsProjectDataSource() datasource.DataSource { | ||
return &azureDevOpsProjectDataSource{} | ||
} | ||
|
||
type azureDevOpsProjectDataSource struct { | ||
client *dbt_cloud.Client | ||
} | ||
|
||
func (d *azureDevOpsProjectDataSource) Metadata( | ||
_ context.Context, | ||
req datasource.MetadataRequest, | ||
resp *datasource.MetadataResponse, | ||
) { | ||
resp.TypeName = req.ProviderTypeName + "_azure_dev_ops_project" | ||
} | ||
|
||
func (d *azureDevOpsProjectDataSource) Read( | ||
ctx context.Context, | ||
req datasource.ReadRequest, | ||
resp *datasource.ReadResponse, | ||
) { | ||
var state AzureDevOpsProjectDataSourceModel | ||
|
||
resp.Diagnostics.Append(req.Config.Get(ctx, &state)...) | ||
|
||
projectName := state.Name.ValueString() | ||
|
||
azureDevOpsProject, err := d.client.GetAzureDevOpsProject(projectName) | ||
|
||
if err != nil { | ||
resp.Diagnostics.AddError( | ||
fmt.Sprintf("Did not find Azure DevOps Project with name: %s", state.Name), | ||
err.Error(), | ||
) | ||
return | ||
} | ||
|
||
state.Name = types.StringValue(azureDevOpsProject.Name) | ||
state.ID = types.StringValue(azureDevOpsProject.ID) | ||
state.URL = types.StringValue(azureDevOpsProject.URL) | ||
|
||
diags := resp.State.Set(ctx, &state) | ||
resp.Diagnostics.Append(diags...) | ||
if resp.Diagnostics.HasError() { | ||
return | ||
} | ||
} | ||
|
||
func (d *azureDevOpsProjectDataSource) Configure( | ||
_ context.Context, | ||
req datasource.ConfigureRequest, | ||
resp *datasource.ConfigureResponse, | ||
) { | ||
switch c := req.ProviderData.(type) { | ||
case nil: // do nothing | ||
case *dbt_cloud.Client: | ||
d.client = c | ||
default: | ||
resp.Diagnostics.AddError( | ||
"Missing client", | ||
"A client is required to configure the Azure DevOps Project data source", | ||
) | ||
} | ||
} |
70 changes: 70 additions & 0 deletions
70
pkg/framework/objects/azure_dev_ops_project/data_source_acceptance_test.go
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,70 @@ | ||
package azure_dev_ops_project_test | ||
|
||
import ( | ||
"os" | ||
"testing" | ||
|
||
"github.com/dbt-labs/terraform-provider-dbtcloud/pkg/framework/acctest_helper" | ||
"github.com/hashicorp/terraform-plugin-testing/config" | ||
"github.com/hashicorp/terraform-plugin-testing/helper/resource" | ||
) | ||
|
||
func TestAccDbtCloudAzureDevOpsProject(t *testing.T) { | ||
//TODO: Remove both env var checks when this gets configurted in CI and the variables are parameterized | ||
if os.Getenv("CI") != "" { | ||
t.Skip("Skipping Azure DevOps Project datasource test in CI " + | ||
"until Azure integration and a personal access token are available") | ||
} | ||
|
||
if os.Getenv("DBT_CLOUD_PERSONAL_ACCESS_TOKEN") == "" { | ||
t.Skip("Skipping Azure DevOps Project datasource because no personal access token is available") | ||
} | ||
|
||
//TODO: Parameterize these values when a standard method is available for parameterization | ||
adoProjectName := "dbt-cloud-ado-project" | ||
personalAccessToken := os.Getenv("DBT_CLOUD_PERSONAL_ACCESS_TOKEN") | ||
|
||
resource.ParallelTest(t, resource.TestCase{ | ||
PreCheck: func() { acctest_helper.TestAccPreCheck(t) }, | ||
|
||
ProtoV6ProviderFactories: acctest_helper.TestAccProtoV6ProviderFactories, | ||
Steps: []resource.TestStep{ | ||
{ | ||
ConfigVariables: config.Variables{ | ||
"dbt_token": config.StringVariable(personalAccessToken), | ||
"ado_project_name": config.StringVariable(adoProjectName), | ||
}, | ||
Config: ` | ||
variable "dbt_token" { | ||
type = string | ||
sensitive = true | ||
} | ||
provider "dbtcloud" { | ||
token = var.dbt_token | ||
} | ||
variable "ado_project_name" { | ||
type = string | ||
} | ||
data dbtcloud_azure_dev_ops_project test { | ||
name = var.ado_project_name | ||
} | ||
`, | ||
// we check the computed values, for the other ones the test suite already checks that the plan and state are the same | ||
Check: resource.ComposeTestCheckFunc( | ||
resource.TestCheckResourceAttrSet( | ||
"data.dbtcloud_azure_dev_ops_project.test", | ||
"id", | ||
), | ||
resource.TestCheckResourceAttrSet( | ||
"data.dbtcloud_azure_dev_ops_project.test", | ||
"url", | ||
), | ||
), | ||
}, | ||
}, | ||
}) | ||
|
||
} |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,9 @@ | ||
package azure_dev_ops_project | ||
|
||
import "github.com/hashicorp/terraform-plugin-framework/types" | ||
|
||
type AzureDevOpsProjectDataSourceModel struct { | ||
ID types.String `tfsdk:"id"` | ||
Name types.String `tfsdk:"name"` | ||
URL types.String `tfsdk:"url"` | ||
} |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,35 @@ | ||
package azure_dev_ops_project | ||
|
||
import ( | ||
"context" | ||
|
||
"github.com/hashicorp/terraform-plugin-framework/datasource" | ||
"github.com/hashicorp/terraform-plugin-framework/datasource/schema" | ||
) | ||
|
||
func (r *azureDevOpsProjectDataSource) Schema( | ||
_ context.Context, | ||
_ datasource.SchemaRequest, | ||
resp *datasource.SchemaResponse, | ||
) { | ||
resp.Schema = schema.Schema{ | ||
Description: `Use this data source to retrieve the ID of an Azure Dev Ops project | ||
based on its name. | ||
This data source requires connecting with a user token and doesn't work with a service token.`, | ||
Attributes: map[string]schema.Attribute{ | ||
"id": schema.StringAttribute{ | ||
Computed: true, | ||
Description: "The internal Azure Dev Ops ID of the ADO Project", | ||
}, | ||
"name": schema.StringAttribute{ | ||
Required: true, | ||
Description: "The name of the ADO project", | ||
}, | ||
"url": schema.StringAttribute{ | ||
Computed: true, | ||
Description: "The URL of the ADO project", | ||
}, | ||
}, | ||
} | ||
} |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file was deleted.
Oops, something went wrong.