fix import

squaredup · Jan 15, 2025 · 27ea26e · 27ea26e
1 parent 662712c
commit 27ea26e
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 3 deletions.
diff --git a/internal/provider/models.go b/internal/provider/models.go
@@ -14,7 +14,8 @@ type DataSource struct {
 	DisplayName string `json:"displayName"`
 	ID          string `json:"id,omitempty"`
 	Plugin      struct {
-		Name string `json:"name"`
+		Name   string `json:"name"`
+		OnPrem bool   `json:"onPrem"`
 	} `json:"plugin"`
 	AgentGroupID string `json:"agentGroupId,omitempty"`
 }

diff --git a/internal/provider/resource_data_source.go b/internal/provider/resource_data_source.go
@@ -140,7 +140,7 @@ func (r *dataSourceResource) Create(ctx context.Context, req resource.CreateRequ
 
 	state := dataSource{
 		DisplayName:  types.StringValue(newDataSource.DisplayName),
-		OnPrem:       types.BoolPointerValue(plan.OnPrem.ValueBoolPointer()),
+		OnPrem:       types.BoolPointerValue(&newDataSource.Plugin.OnPrem),
 		Name:         types.StringValue(newDataSource.Plugin.Name),
 		AgentGroupID: types.StringValue(newDataSource.AgentGroupID),
 		ID:           types.StringValue(newDataSource.ID),
@@ -176,6 +176,7 @@ func (r *dataSourceResource) Read(ctx context.Context, req resource.ReadRequest,
 	}
 
 	state.DisplayName = types.StringValue(readDataSource.DisplayName)
+	state.OnPrem = types.BoolValue(readDataSource.Plugin.OnPrem)
 	state.Name = types.StringValue(readDataSource.Plugin.Name)
 	state.AgentGroupID = types.StringValue(readDataSource.AgentGroupID)
 	state.ID = types.StringValue(readDataSource.ID)
@@ -245,7 +246,7 @@ func (r *dataSourceResource) Update(ctx context.Context, req resource.UpdateRequ
 
 	state = dataSource{
 		DisplayName:  types.StringValue(getDataSource.DisplayName),
-		OnPrem:       types.BoolPointerValue(plan.OnPrem.ValueBoolPointer()),
+		OnPrem:       types.BoolPointerValue(&getDataSource.Plugin.OnPrem),
 		Name:         types.StringValue(getDataSource.Plugin.Name),
 		AgentGroupID: types.StringValue(getDataSource.AgentGroupID),
 		ID:           types.StringValue(getDataSource.ID),

diff --git a/internal/provider/resource_data_source_test.go b/internal/provider/resource_data_source_test.go
@@ -28,6 +28,7 @@ resource "squaredup_datasource" "sample_data_source" {
 					resource.TestCheckResourceAttr("squaredup_datasource.sample_data_source", "display_name", "Sample Data - DataSource Test - "+uuid),
 					resource.TestCheckResourceAttrSet("squaredup_datasource.sample_data_source", "id"),
 					resource.TestCheckResourceAttrSet("squaredup_datasource.sample_data_source", "last_updated"),
+					resource.TestCheckResourceAttr("squaredup_datasource.sample_data_source", "on_prem", "false"),
 				),
 			},
 			//Update DataSource Test