kairos/kairos-agent
1
0
Fork 0
mirror of https://github.com/kairos-io/kairos-agent.git synced 2025-09-04 18:40:04 +00:00
Code Issues Packages Projects Releases Wiki Activity

Respect user defined/default sizes on upgrade (#587)

Browse Source
This commit is contained in:
Itxaka
2024-11-06 11:34:01 +01:00
committed by GitHub
parent 87fca9570f
commit 6634e18aa3
7 changed files with 73 additions and 71 deletions
Download Patch File Download Diff File Expand all files Collapse all files

40
pkg/config/spec_test.go
View File

@@ -20,6 +20,7 @@ import (
"bytes"
"os"
"path/filepath"
"strings"
"github.com/kairos-io/kairos-agent/v2/pkg/config"
"github.com/kairos-io/kairos-agent/v2/pkg/constants"
@@ -449,6 +450,45 @@ var _ = Describe("Types", Label("types", "config"), func() {
Expect(spec.Recovery.Source.IsEmpty()).To(BeTrue())
Expect(spec.Recovery.FS).To(Equal(constants.SquashFs))
})
It("sets image size to default value if not set", func() {
spec, err := config.NewUpgradeSpec(c)
Expect(err).ShouldNot(HaveOccurred())
Expect(spec.Active.Size).To(Equal(constants.ImgSize))
})
It("sets image size to provided value if set in the config and image is smaller", func() {
cfg, err := config.ScanNoLogs(collector.Readers(strings.NewReader("#cloud-config\nupgrade:\n system:\n size: 666\n")))
// Set manually the config collector in the cfg file before unmarshalling the spec
c.Config = cfg.Config
spec, err := config.NewUpgradeSpec(c)
Expect(err).ShouldNot(HaveOccurred())
Expect(spec.Active.Size).To(Equal(uint(666)))
})
It("sets image size to default value if not set in the config and image is smaller", func() {
cfg, err := config.ScanNoLogs(collector.Readers(strings.NewReader("#cloud-config\nupgrade:\n system:\n uri: dir:/\n")))
// Set manually the config collector in the cfg file before unmarshalling the spec
c.Config = cfg.Config
spec, err := config.NewUpgradeSpec(c)
Expect(err).ShouldNot(HaveOccurred())
Expect(spec.Active.Size).To(Equal(constants.ImgSize))
})
It("sets image size to the source if default is smaller", func() {
cfg, err := config.ScanNoLogs(collector.Readers(strings.NewReader("#cloud-config\nupgrade:\n system:\n uri: file:/tmp/waka\n")))
// Set manually the config collector in the cfg file before unmarshalling the spec
c.Config = cfg.Config
Expect(c.Fs.Mkdir("/tmp", 0777)).ShouldNot(HaveOccurred())
Expect(c.Fs.WriteFile("/tmp/waka", []byte("waka"), 0777)).ShouldNot(HaveOccurred())
Expect(c.Fs.Truncate("/tmp/waka", 5120*1024*1024)).ShouldNot(HaveOccurred())
spec, err := config.NewUpgradeSpec(c)
Expect(err).ShouldNot(HaveOccurred())
f, err := c.Fs.Stat("/tmp/waka")
Expect(err).ShouldNot(HaveOccurred())
// Make the same calculation as the code
Expect(spec.Active.Size).To(Equal(uint(f.Size()/1000/1000) + 100))
})
})
})
Describe("Config from cloudconfig", Label("cloud-config"), func() {