47 lines
811 B
Go
47 lines
811 B
Go
// Copyright (c) HashiCorp, Inc.
|
|
// SPDX-License-Identifier: MPL-2.0
|
|
|
|
package testutil
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"path/filepath"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestWait_WaitForFilesUntil(t *testing.T) {
|
|
|
|
N := 10
|
|
|
|
tmpDir := t.TempDir()
|
|
|
|
var files []string
|
|
for i := 1; i < N; i++ {
|
|
files = append(files, filepath.Join(
|
|
tmpDir, fmt.Sprintf("test%d.txt", i),
|
|
))
|
|
}
|
|
|
|
go func() {
|
|
for _, file := range files {
|
|
t.Logf("Creating file %s ...", file)
|
|
fh, createErr := os.Create(file)
|
|
require.NoError(t, createErr)
|
|
|
|
closeErr := fh.Close()
|
|
require.NoError(t, closeErr)
|
|
require.FileExists(t, file)
|
|
|
|
time.Sleep(250 * time.Millisecond)
|
|
}
|
|
}()
|
|
|
|
duration := 5 * time.Second
|
|
t.Log("Waiting 5 seconds for files ...")
|
|
WaitForFilesUntil(t, files, duration)
|
|
}
|