Skip to content

Commit 764b7ad

Browse files
committed
fixup! Introduce WatcherHelper
1 parent 6e3ebbc commit 764b7ad

File tree

5 files changed

+271
-285
lines changed

5 files changed

+271
-285
lines changed

.mockery.yaml

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -40,3 +40,10 @@ packages:
4040
github.com/elastic/elastic-agent/internal/pkg/agent/application/upgrade:
4141
interfaces:
4242
WatcherHelper:
43+
config:
44+
inpackage: True
45+
with-expecter: True
46+
dir: "{{.InterfaceDirRelative}}"
47+
mockname: "{{.Mock}}{{.InterfaceName}}"
48+
outpkg: "{{.PackageName}}"
49+
filename: "mocks_test.go"

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

Lines changed: 262 additions & 0 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

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

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -33,7 +33,6 @@ import (
3333
"github.com/elastic/elastic-agent/pkg/core/logger"
3434
"github.com/elastic/elastic-agent/pkg/core/logger/loggertest"
3535
infomocks "github.com/elastic/elastic-agent/testing/mocks/internal_/pkg/agent/application/info"
36-
upgrademocks "github.com/elastic/elastic-agent/testing/mocks/internal_/pkg/agent/application/upgrade"
3736
ackermocks "github.com/elastic/elastic-agent/testing/mocks/internal_/pkg/fleetapi/acker"
3837
clientmocks "github.com/elastic/elastic-agent/testing/mocks/pkg/control/v2/client"
3938
)
@@ -1036,7 +1035,7 @@ func TestIsSameReleaseVersion(t *testing.T) {
10361035
func TestManualRollback(t *testing.T) {
10371036
log, _ := loggertest.New(t.Name())
10381037
mockAgentInfo := infomocks.NewAgent(t)
1039-
upgrader, err := NewUpgrader(log, &artifact.Config{}, &configuration.UpgradeConfig{}, mockAgentInfo, upgrademocks.NewWatcherHelper(t))
1038+
upgrader, err := NewUpgrader(log, &artifact.Config{}, &configuration.UpgradeConfig{}, mockAgentInfo, NewMockWatcherHelper(t))
10401039
require.NoError(t, err, "error creating upgrader")
10411040

10421041
upgrader.Upgrade(t.Context(), "", true, "", nil, nil, true, false)

magefile.go

Lines changed: 1 addition & 25 deletions
Original file line numberDiff line numberDiff line change
@@ -262,31 +262,7 @@ func (Dev) RegenerateMocks() error {
262262
return fmt.Errorf("generating mocks: %w", err)
263263
}
264264

265-
// change CWD
266-
workingDir, err := os.Getwd()
267-
if err != nil {
268-
return fmt.Errorf("retrieving CWD: %w", err)
269-
}
270-
// restore the working directory when exiting the function
271-
defer func() {
272-
err := os.Chdir(workingDir)
273-
if err != nil {
274-
panic(fmt.Errorf("failed to restore working dir %q: %w", workingDir, err))
275-
}
276-
}()
277-
278-
mPath, err := mocksPath()
279-
if err != nil {
280-
return fmt.Errorf("retrieving mocks path: %w", err)
281-
}
282-
283-
err = os.Chdir(mPath)
284-
if err != nil {
285-
return fmt.Errorf("changing current directory to %q: %w", mPath, err)
286-
}
287-
288-
mg.Deps(devtools.AddLicenseHeaders)
289-
mg.Deps(devtools.GoImports)
265+
mg.Deps(devtools.Format)
290266
return nil
291267
}
292268

0 commit comments

Comments
 (0)