fixup! remove fakeAcker in favour of generated Acker mock

Author: pchila

internal/pkg/agent/application/upgrade/upgrade_test.go

@@ -36,8 +36,8 @@ import (
 	"github.com/elastic/elastic-agent/pkg/core/logger"
 	"github.com/elastic/elastic-agent/pkg/core/logger/loggertest"
 	agtversion "github.com/elastic/elastic-agent/pkg/version"
-	mockack "github.com/elastic/elastic-agent/testing/mocks/internal_/pkg/fleetapi/acker"
-	mocks "github.com/elastic/elastic-agent/testing/mocks/pkg/control/v2/client"
+	ackermocks "github.com/elastic/elastic-agent/testing/mocks/internal_/pkg/fleetapi/acker"
+	clientmocks "github.com/elastic/elastic-agent/testing/mocks/pkg/control/v2/client"
 )
 
 func Test_CopyFile(t *testing.T) {
@@ -239,7 +239,7 @@ func TestIsInProgress(t *testing.T) {
 		t.Run(name, func(t *testing.T) {
 			// Expect client.State() call to be made only if no Upgrade Watcher PIDs
 			// are returned (i.e. no Upgrade Watcher is found to be running).
-			mc := mocks.NewClient(t)
+			mc := clientmocks.NewClient(t)
 			if test.watcherPIDsFetcher != nil {
 				pids, _ := test.watcherPIDsFetcher()
 				if len(pids) == 0 {
@@ -293,15 +293,15 @@ func TestUpgraderAckAction(t *testing.T) {
 		require.Nil(t, u.AckAction(t.Context(), nil, action))
 	})
 	t.Run("AckAction with acker", func(t *testing.T) {
-		mockAcker := &mockack.Acker{}
+		mockAcker := ackermocks.NewAcker(t)
 		mockAcker.EXPECT().Ack(mock.Anything, action).Return(nil)
 		mockAcker.EXPECT().Commit(mock.Anything).Return(nil)
 
 		require.Nil(t, u.AckAction(t.Context(), mockAcker, action))
 	})
 
 	t.Run("AckAction with acker - failing commit", func(t *testing.T) {
-		mockAcker := &mockack.Acker{}
+		mockAcker := ackermocks.NewAcker(t)
 
 		errCommit := errors.New("failed commit")
 		mockAcker.EXPECT().Ack(mock.Anything, action).Return(nil)
@@ -311,12 +311,12 @@ func TestUpgraderAckAction(t *testing.T) {
 	})
 
 	t.Run("AckAction with acker - failed ack", func(t *testing.T) {
-		mockAcker := &mockack.Acker{}
+		mockAcker := ackermocks.NewAcker(t)
 
 		errAck := errors.New("ack error")
 		mockAcker.EXPECT().Ack(mock.Anything, action).Return(errAck)
-		mockAcker.EXPECT().Commit(mock.Anything).Return(nil)
-
+		// no expectation on Commit() since it shouldn't be called after an error during Ack()
+		
 		require.ErrorIs(t, u.AckAction(t.Context(), mockAcker, action), errAck)
 	})
 }