talos/pkg/archiver/walker_test.go
Andrey Smirnov bc9e0c0dba fix: re-implement upgrade (install) with preserve
For 0.6 -> 0.7 upgrade, in any case config.yaml is preserved and moved
from `/boot` to `/system/state`.

For single node upgrade, `EPHEMERAL` partition is not touched and other
partitions are re-created as needed.

Bump provision tests to 0.6/0.7 upgrades as we get closer to the new
release.

Signed-off-by: Andrey Smirnov <smirnov.andrey@gmail.com>
2020-10-28 07:25:26 -07:00

137 lines
3.3 KiB
Go

// This Source Code Form is subject to the terms of the Mozilla Public
// License, v. 2.0. If a copy of the MPL was not distributed with this
// file, You can obtain one at http://mozilla.org/MPL/2.0/.
// Package archiver provides a service to archive part of the filesystem into tar archive
package archiver_test
import (
"context"
"io/ioutil"
"os"
"path/filepath"
"testing"
"github.com/stretchr/testify/suite"
"github.com/talos-systems/talos/pkg/archiver"
)
type WalkerSuite struct {
CommonSuite
}
func (suite *WalkerSuite) TestIterationDir() {
ch, err := archiver.Walker(context.Background(), suite.tmpDir, archiver.WithSkipRoot())
suite.Require().NoError(err)
relPaths := []string(nil)
for fi := range ch {
suite.Require().NoError(fi.Error)
relPaths = append(relPaths, fi.RelPath)
if fi.RelPath == "usr/bin/mv" {
suite.Assert().Equal("/usr/bin/cp", fi.Link)
}
}
suite.Assert().Equal([]string{
"dev", "dev/random",
"etc", "etc/certs", "etc/certs/ca.crt", "etc/hostname",
"lib", "lib/dynalib.so",
"usr", "usr/bin", "usr/bin/cp", "usr/bin/mv",
},
relPaths)
}
func (suite *WalkerSuite) TestIterationFilter() {
ch, err := archiver.Walker(context.Background(), suite.tmpDir, archiver.WithSkipRoot(), archiver.WithFnmatchPatterns("dev/*", "lib"))
suite.Require().NoError(err)
relPaths := []string(nil)
for fi := range ch {
suite.Require().NoError(fi.Error)
relPaths = append(relPaths, fi.RelPath)
if fi.RelPath == "usr/bin/mv" {
suite.Assert().Equal("/usr/bin/cp", fi.Link)
}
}
suite.Assert().Equal([]string{
"dev/random",
"lib",
},
relPaths)
}
func (suite *WalkerSuite) TestIterationMaxRecurseDepth() {
ch, err := archiver.Walker(context.Background(), suite.tmpDir, archiver.WithMaxRecurseDepth(1))
suite.Require().NoError(err)
relPaths := []string(nil)
for fi := range ch {
suite.Require().NoError(fi.Error)
relPaths = append(relPaths, fi.RelPath)
}
suite.Assert().Equal([]string{
".", "dev", "etc", "lib", "usr",
},
relPaths)
}
func (suite *WalkerSuite) TestIterationFile() {
ch, err := archiver.Walker(context.Background(), filepath.Join(suite.tmpDir, "usr/bin/cp"))
suite.Require().NoError(err)
relPaths := []string(nil)
for fi := range ch {
suite.Require().NoError(fi.Error)
relPaths = append(relPaths, fi.RelPath)
}
suite.Assert().Equal([]string{"cp"},
relPaths)
}
func (suite *WalkerSuite) TestIterationSymlink() {
original := filepath.Join(suite.tmpDir, "original")
err := os.Mkdir(original, 0o755)
suite.Require().NoError(err)
newname := filepath.Join(suite.tmpDir, "new")
// NB: We make this a relative symlink to make the test more complete.
err = os.Symlink("original", newname)
suite.Require().NoError(err)
err = ioutil.WriteFile(filepath.Join(original, "original.txt"), []byte{}, 0o666)
suite.Require().NoError(err)
ch, err := archiver.Walker(context.Background(), newname)
suite.Require().NoError(err)
relPaths := []string(nil)
for fi := range ch {
suite.Require().NoError(fi.Error)
relPaths = append(relPaths, fi.RelPath)
}
suite.Assert().Equal([]string{".", "original.txt"}, relPaths)
}
func (suite *WalkerSuite) TestIterationNotFound() {
_, err := archiver.Walker(context.Background(), filepath.Join(suite.tmpDir, "doesntlivehere"))
suite.Require().Error(err)
}
func TestWalkerSuite(t *testing.T) {
suite.Run(t, new(WalkerSuite))
}