test: use a lower unenroll timeout in TestActionUnenrollHandler (#5465)

* test: use a lower unenroll timeout in TestActionUnenrollHandler

auto unenroll with no running components is timing out after 15s due to
hardcoded unenrollTimeout.
There is no need to use a huge timeout in test.
Pass the timeout as a method arg and use a lower timeout.

Test time went from 15s to 0.1s

* lint: fix linter issues

internal/pkg/agent/application/actions/handlers/handler_action_unenroll.go

@@ -61,6 +61,10 @@ func NewUnenroll(
 
 // Handle handles UNENROLL action.
 func (h *Unenroll) Handle(ctx context.Context, a fleetapi.Action, acker acker.Acker) error {
+	return h.handle(ctx, a, acker, unenrollTimeout)
+}
+
+func (h *Unenroll) handle(ctx context.Context, a fleetapi.Action, acker acker.Acker, timeout time.Duration) error {
 	h.log.Debugf("handlerUnenroll: action '%+v' received", a)
 	action, ok := a.(*fleetapi.ActionUnenroll)
 	if !ok {
@@ -98,7 +102,7 @@ func (h *Unenroll) Handle(ctx context.Context, a fleetapi.Action, acker acker.Ac
 		h.log.Warnf("Failed to update state store: %v", err)
 	}
 
-	unenrollCtx, cancel := context.WithTimeout(ctx, unenrollTimeout)
+	unenrollCtx, cancel := context.WithTimeout(ctx, timeout)
 	defer cancel()
 
 	unenrollPolicy.WaitAck(unenrollCtx)

internal/pkg/agent/application/actions/handlers/handler_action_unenroll_test.go

@@ -9,6 +9,7 @@ import (
 	"path/filepath"
 	"sync/atomic"
 	"testing"
+	"time"
 
 	"github.com/elastic/elastic-agent-client/v7/pkg/client"
 	"github.com/elastic/elastic-agent-client/v7/pkg/proto"
@@ -219,7 +220,7 @@ func TestActionUnenrollHandler(t *testing.T) {
 			if tc.autoUnenroll {
 				a.IsDetected = true
 			}
-			err := handler.Handle(ctx, a, acker)
+			err := handler.handle(ctx, a, acker, 100*time.Millisecond)
 
 			require.ErrorIs(t, err, tc.wantErr)