mirror of
https://github.com/siderolabs/talos.git
synced 2025-08-18 04:27:06 +02:00
9379cf9ee1
This change is only moving packages and updating import paths. Goal: expose `internal/pkg/provision` as `pkg/provision` to enable other projects to import Talos provisioning library. As cluster checks are almost always required as part of provisioning process, package `internal/pkg/cluster` was also made public as `pkg/cluster`. Other changes were direct dependencies discovered by `importvet` which were updated. Public packages (useful, general purpose packages with stable API): * `internal/pkg/conditions` -> `pkg/conditions` * `internal/pkg/tail` -> `pkg/tail` Private packages (used only on provisioning library internally): * `internal/pkg/inmemhttp` -> `pkg/provision/internal/inmemhttp` * `internal/pkg/kernel/vmlinuz` -> `pkg/provision/internal/vmlinuz` * `internal/pkg/cniutils` -> `pkg/provision/internal/cniutils` Signed-off-by: Andrey Smirnov <smirnov.andrey@gmail.com>
149 lines
3.0 KiB
Go
149 lines
3.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 conditions_test
|
|
|
|
import (
|
|
"context"
|
|
"io/ioutil"
|
|
"os"
|
|
"path/filepath"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/suite"
|
|
|
|
"github.com/talos-systems/talos/pkg/conditions"
|
|
)
|
|
|
|
type FilesSuite struct {
|
|
suite.Suite
|
|
|
|
tempDir string
|
|
}
|
|
|
|
func (suite *FilesSuite) SetupSuite() {
|
|
var err error
|
|
suite.tempDir, err = ioutil.TempDir("", "talos")
|
|
suite.Require().NoError(err)
|
|
}
|
|
|
|
func (suite *FilesSuite) TearDownSuite() {
|
|
suite.Require().NoError(os.RemoveAll(suite.tempDir))
|
|
}
|
|
|
|
func (suite *FilesSuite) createFile(name string) (path string) {
|
|
path = filepath.Join(suite.tempDir, name)
|
|
f, err := os.Create(path)
|
|
suite.Require().NoError(err)
|
|
|
|
suite.Require().NoError(f.Close())
|
|
|
|
return
|
|
}
|
|
|
|
func (suite *FilesSuite) TestString() {
|
|
suite.Require().Equal("file \"abc.txt\" to exist", conditions.WaitForFileToExist("abc.txt").String())
|
|
}
|
|
|
|
func (suite *FilesSuite) TestWaitForFileToExist() {
|
|
path := suite.createFile("w.txt")
|
|
|
|
err := conditions.WaitForFileToExist(path).Wait(context.Background())
|
|
suite.Require().NoError(err)
|
|
|
|
suite.Require().NoError(os.Remove(path))
|
|
|
|
errCh := make(chan error)
|
|
|
|
go func() {
|
|
errCh <- conditions.WaitForFileToExist(path).Wait(context.Background())
|
|
}()
|
|
|
|
time.Sleep(50 * time.Millisecond)
|
|
|
|
select {
|
|
case <-errCh:
|
|
suite.Fail("unexpected return")
|
|
default:
|
|
}
|
|
|
|
suite.createFile("w.txt")
|
|
|
|
suite.Require().NoError(<-errCh)
|
|
|
|
suite.Require().NoError(os.Remove(path))
|
|
|
|
ctx, ctxCancel := context.WithCancel(context.Background())
|
|
|
|
go func() {
|
|
errCh <- conditions.WaitForFileToExist(path).Wait(ctx)
|
|
}()
|
|
|
|
time.Sleep(50 * time.Millisecond)
|
|
|
|
select {
|
|
case <-errCh:
|
|
suite.Fail("unexpected return")
|
|
default:
|
|
}
|
|
|
|
ctxCancel()
|
|
|
|
suite.Require().EqualError(<-errCh, context.Canceled.Error())
|
|
}
|
|
|
|
func (suite *FilesSuite) TestWaitForAllFilesToExist() {
|
|
pathA := suite.createFile("wA.txt")
|
|
pathB := suite.createFile("wB.txt")
|
|
|
|
err := conditions.WaitForFilesToExist(pathA, pathB).Wait(context.Background())
|
|
suite.Require().NoError(err)
|
|
|
|
suite.Require().NoError(os.Remove(pathB))
|
|
|
|
errCh := make(chan error)
|
|
|
|
go func() {
|
|
errCh <- conditions.WaitForFilesToExist(pathA, pathB).Wait(context.Background())
|
|
}()
|
|
|
|
time.Sleep(50 * time.Millisecond)
|
|
|
|
select {
|
|
case <-errCh:
|
|
suite.Fail("unexpected return")
|
|
default:
|
|
}
|
|
|
|
suite.createFile("wB.txt")
|
|
|
|
suite.Require().NoError(<-errCh)
|
|
|
|
suite.Require().NoError(os.Remove(pathA))
|
|
suite.Require().NoError(os.Remove(pathB))
|
|
|
|
ctx, ctxCancel := context.WithCancel(context.Background())
|
|
|
|
go func() {
|
|
errCh <- conditions.WaitForFilesToExist(pathA, pathB).Wait(ctx)
|
|
}()
|
|
|
|
time.Sleep(50 * time.Millisecond)
|
|
|
|
select {
|
|
case <-errCh:
|
|
suite.Fail("unexpected return")
|
|
default:
|
|
}
|
|
|
|
ctxCancel()
|
|
|
|
suite.Require().EqualError(<-errCh, context.Canceled.Error())
|
|
}
|
|
|
|
func TestFilesSuite(t *testing.T) {
|
|
suite.Run(t, new(FilesSuite))
|
|
}
|