From c0c419cce8255b22b69e5387987e36319d458d0e Mon Sep 17 00:00:00 2001 From: Dimitris Karakasilis Date: Tue, 13 Aug 2024 14:51:42 +0300 Subject: [PATCH] Fix unit tests Signed-off-by: Dimitris Karakasilis --- pkg/action/upgrade_test.go | 9 ++++----- pkg/types/v1/config_test.go | 5 +++-- 2 files changed, 7 insertions(+), 7 deletions(-) diff --git a/pkg/action/upgrade_test.go b/pkg/action/upgrade_test.go index a53edd6..3254716 100644 --- a/pkg/action/upgrade_test.go +++ b/pkg/action/upgrade_test.go @@ -19,9 +19,10 @@ package action_test import ( "bytes" "fmt" - sdkTypes "github.com/kairos-io/kairos-sdk/types" "path/filepath" + sdkTypes "github.com/kairos-io/kairos-sdk/types" + agentConfig "github.com/kairos-io/kairos-agent/v2/pkg/config" fsutils "github.com/kairos-io/kairos-agent/v2/pkg/utils/fs" @@ -449,8 +450,7 @@ var _ = Describe("Runtime Actions", func() { spec.Active.Size = 10 spec.Passive.Size = 10 spec.Recovery.Size = 10 - - spec.RecoveryUpgrade = true + spec.Entry = constants.BootEntryRecovery err = fsutils.MkdirAll(config.Fs, "/proc", constants.DirPerm) Expect(err).ShouldNot(HaveOccurred()) @@ -542,8 +542,7 @@ var _ = Describe("Runtime Actions", func() { spec.Active.Size = 10 spec.Passive.Size = 10 spec.Recovery.Size = 10 - - spec.RecoveryUpgrade = true + spec.Entry = constants.BootEntryRecovery err = fsutils.MkdirAll(config.Fs, "/proc", constants.DirPerm) Expect(err).ShouldNot(HaveOccurred()) diff --git a/pkg/types/v1/config_test.go b/pkg/types/v1/config_test.go index 877e7c1..04dd223 100644 --- a/pkg/types/v1/config_test.go +++ b/pkg/types/v1/config_test.go @@ -17,11 +17,12 @@ limitations under the License. package v1_test import ( + "path/filepath" + "github.com/kairos-io/kairos-agent/v2/pkg/constants" v1 "github.com/kairos-io/kairos-agent/v2/pkg/types/v1" . "github.com/onsi/ginkgo/v2" . "github.com/onsi/gomega" - "path/filepath" ) var _ = Describe("Types", Label("types", "config"), func() { @@ -444,7 +445,7 @@ var _ = Describe("Types", Label("types", "config"), func() { }) Describe("Recovery upgrade", func() { BeforeEach(func() { - spec.RecoveryUpgrade = true + spec.Entry = constants.BootEntryRecovery }) It("fails with empty source", func() { err := spec.Sanitize()