Skip to content

Commit cd13249

Browse files
Copilottobio
andcommitted
Apply code review suggestions: improve duration conversion and simplify test helpers
Co-authored-by: tobio <[email protected]>
1 parent d448c54 commit cd13249

File tree

2 files changed

+8
-7
lines changed

2 files changed

+8
-7
lines changed

internal/fleet/agent_policy/models.go

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,7 @@ import (
44
"context"
55
"fmt"
66
"slices"
7+
"time"
78

89
"github.com/elastic/terraform-provider-elasticstack/generated/kbapi"
910
"github.com/elastic/terraform-provider-elasticstack/internal/utils"
@@ -78,8 +79,8 @@ func (model *agentPolicyModel) populateFromAPI(ctx context.Context, data *kbapi.
7879
model.SupportsAgentless = types.BoolPointerValue(data.SupportsAgentless)
7980
if data.InactivityTimeout != nil {
8081
// Convert seconds to duration string
81-
durationStr := fmt.Sprintf("%.0fs", *data.InactivityTimeout)
82-
model.InactivityTimeout = customtypes.NewDurationValue(durationStr)
82+
d := time.Duration(*data.InactivityTimeout * float32(time.Second)).Truncate(time.Second)
83+
model.InactivityTimeout = customtypes.NewDurationValue(d.String())
8384
} else {
8485
model.InactivityTimeout = customtypes.NewDurationNull()
8586
}

internal/fleet/agent_policy/resource_test.go

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -145,7 +145,7 @@ func TestAccResourceAgentPolicy(t *testing.T) {
145145
},
146146
{
147147
SkipFunc: versionutils.CheckIfVersionIsUnsupported(agent_policy.MinVersionInactivityTimeout),
148-
Config: testAccResourceAgentPolicyCreateWithInactivityTimeout(policyName, false, "2m"),
148+
Config: testAccResourceAgentPolicyCreateWithInactivityTimeout(policyName),
149149
Check: resource.ComposeTestCheckFunc(
150150
resource.TestCheckResourceAttr("elasticstack_fleet_agent_policy.test_policy", "name", fmt.Sprintf("Policy %s", policyName)),
151151
resource.TestCheckResourceAttr("elasticstack_fleet_agent_policy.test_policy", "namespace", "default"),
@@ -308,7 +308,7 @@ data "elasticstack_fleet_enrollment_tokens" "test_policy" {
308308
`, fmt.Sprintf("Policy %s", id), skipDestroy)
309309
}
310310

311-
func testAccResourceAgentPolicyCreateWithInactivityTimeout(id string, skipDestroy bool, inactivityTimeout string) string {
311+
func testAccResourceAgentPolicyCreateWithInactivityTimeout(id string) string {
312312
return fmt.Sprintf(`
313313
provider "elasticstack" {
314314
elasticsearch {}
@@ -321,15 +321,15 @@ resource "elasticstack_fleet_agent_policy" "test_policy" {
321321
description = "Test Agent Policy with Inactivity Timeout"
322322
monitor_logs = true
323323
monitor_metrics = false
324-
skip_destroy = %t
325-
inactivity_timeout = "%s"
324+
skip_destroy = false
325+
inactivity_timeout = "2m"
326326
}
327327
328328
data "elasticstack_fleet_enrollment_tokens" "test_policy" {
329329
policy_id = elasticstack_fleet_agent_policy.test_policy.policy_id
330330
}
331331
332-
`, fmt.Sprintf("Policy %s", id), skipDestroy, inactivityTimeout)
332+
`, fmt.Sprintf("Policy %s", id))
333333
}
334334

335335
func testAccResourceAgentPolicyCreateWithBadGlobalDataTags(id string, skipDestroy bool) string {

0 commit comments

Comments
 (0)