talos/pkg/archiver/walker_test.go
Andrew Rynhard 90c91807bd refactor: restructure the project layout
This change moves packages into more appropriate places.

Signed-off-by: Andrew Rynhard <andrew@andrewrynhard.com>
2019-08-01 22:19:42 -07:00

83 lines
2.0 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"
"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) 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) 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))
}