96aa9638f7
There's a cyclic dependency on siderolink library which imports talos machinery back. We will fix that after we get talos pushed under a new name. Signed-off-by: Andrey Smirnov <andrey.smirnov@talos-systems.com>
142 lines
2.8 KiB
Go
142 lines
2.8 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"
|
|
"os"
|
|
"path/filepath"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/suite"
|
|
|
|
"github.com/siderolabs/talos/pkg/conditions"
|
|
)
|
|
|
|
type FilesSuite struct {
|
|
suite.Suite
|
|
|
|
tempDir string
|
|
}
|
|
|
|
func (suite *FilesSuite) SetupSuite() {
|
|
suite.tempDir = suite.T().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))
|
|
}
|