github/kairos-agent
7
0
Fork 0
mirror of https://github.com/kairos-io/kairos-agent.git synced 2025-06-03 01:44:53 +00:00
Code Issues Packages Projects Releases Wiki Activity

Address PR comments

Browse Source 
Signed-off-by: Dimitris Karakasilis <dimitris@karakasilis.me>
This commit is contained in:
Dimitris Karakasilis 2024-09-04 12:51:19 +03:00
parent 8bf8fd8e6e
commit 774547e150
No known key found for this signature in database
GPG Key ID: 286DCAFD2C97DDE3
1 changed files with 7 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
pkg/action/upgrade_test.go
View File

@ -160,7 +160,7 @@ var _ = Describe("Runtime Actions", func() {
Expect(err).ToNot(HaveOccurred())
cliConfig := string(d)
config, err := agentConfig.Scan(collector.Directories(), collector.Readers(strings.NewReader(cliConfig)))
config, err := agentConfig.Scan(collector.Readers(strings.NewReader(cliConfig)))
Expect(err).ToNot(HaveOccurred())
agentConfig.WithFs(fs)(config)
@ -670,27 +670,15 @@ func createDummyFile(fs vfs.FS, sizeMb int64) string {
tmpFile.Close()
os.RemoveAll(tmpName)
dummyData := []byte("1234567890ABCDEF")
dummyLength := int64(len(dummyData))
var written int64
data := []byte{}
for written < fileSize {
bytesToWrite := dummyLength
if written+bytesToWrite > fileSize {
bytesToWrite = fileSize - written
}
data = append(data, dummyData[:bytesToWrite]...)
Expect(err).ToNot(HaveOccurred())
written += bytesToWrite
}
dir := filepath.Dir(tmpName)
err = fs.Mkdir(dir, os.ModePerm)
Expect(err).ToNot(HaveOccurred())
err = fs.WriteFile(tmpName, data, os.ModePerm)
Expect(err).ToNot(HaveOccurred())
f, err := fs.Create(tmpName)
Expect(err).ShouldNot(HaveOccurred())
err = f.Truncate(fileSize)
Expect(err).ShouldNot(HaveOccurred())
f.Close()
return tmpName
}