92 lines
2.7 KiB
Go
92 lines
2.7 KiB
Go
|
package config
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"os"
|
||
|
|
"testing"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"github.com/hashicorp/go-hclog"
|
||
|
|
|
||
|
|
"github.com/hashicorp/consul/sdk/testutil"
|
||
|
|
"github.com/stretchr/testify/require"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestNewRateLimitedWatcher(t *testing.T) {
|
||
|
|
w, err := NewRateLimitedFileWatcher([]string{}, hclog.New(&hclog.LoggerOptions{}), 1*time.Nanosecond)
|
||
|
|
require.NoError(t, err)
|
||
|
|
require.NotNil(t, w)
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestRateLimitedWatcherRenameEvent(t *testing.T) {
|
||
|
|
|
||
|
|
fileTmp := createTempConfigFile(t, "temp_config3")
|
||
|
|
filepaths := []string{createTempConfigFile(t, "temp_config1"), createTempConfigFile(t, "temp_config2")}
|
||
|
|
w, err := NewRateLimitedFileWatcher(filepaths, hclog.New(&hclog.LoggerOptions{}), 1*time.Nanosecond)
|
||
|
|
|
||
|
|
require.NoError(t, err)
|
||
|
|
w.Start(context.Background())
|
||
|
|
defer func() {
|
||
|
|
_ = w.Stop()
|
||
|
|
}()
|
||
|
|
|
||
|
|
require.NoError(t, err)
|
||
|
|
err = os.Rename(fileTmp, filepaths[0])
|
||
|
|
time.Sleep(timeoutDuration + 50*time.Millisecond)
|
||
|
|
require.NoError(t, err)
|
||
|
|
require.NoError(t, assertEvent(filepaths[0], w.EventsCh(), defaultTimeout))
|
||
|
|
// make sure we consume all events
|
||
|
|
_ = assertEvent(filepaths[0], w.EventsCh(), defaultTimeout)
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestRateLimitedWatcherAddNotExist(t *testing.T) {
|
||
|
|
|
||
|
|
file := testutil.TempFile(t, "temp_config")
|
||
|
|
filename := file.Name() + randomStr(16)
|
||
|
|
w, err := NewRateLimitedFileWatcher([]string{filename}, hclog.New(&hclog.LoggerOptions{}), 1*time.Nanosecond)
|
||
|
|
require.Error(t, err, "no such file or directory")
|
||
|
|
require.Nil(t, w)
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestEventRateLimitedWatcherWrite(t *testing.T) {
|
||
|
|
|
||
|
|
file := testutil.TempFile(t, "temp_config")
|
||
|
|
_, err := file.WriteString("test config")
|
||
|
|
require.NoError(t, err)
|
||
|
|
err = file.Sync()
|
||
|
|
require.NoError(t, err)
|
||
|
|
w, err := NewRateLimitedFileWatcher([]string{file.Name()}, hclog.New(&hclog.LoggerOptions{}), 1*time.Nanosecond)
|
||
|
|
require.NoError(t, err)
|
||
|
|
w.Start(context.Background())
|
||
|
|
defer func() {
|
||
|
|
_ = w.Stop()
|
||
|
|
}()
|
||
|
|
|
||
|
|
_, err = file.WriteString("test config 2")
|
||
|
|
require.NoError(t, err)
|
||
|
|
err = file.Sync()
|
||
|
|
require.NoError(t, err)
|
||
|
|
require.NoError(t, assertEvent(file.Name(), w.EventsCh(), defaultTimeout))
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestEventRateLimitedWatcherMove(t *testing.T) {
|
||
|
|
|
||
|
|
filepath := createTempConfigFile(t, "temp_config1")
|
||
|
|
|
||
|
|
w, err := NewRateLimitedFileWatcher([]string{filepath}, hclog.New(&hclog.LoggerOptions{}), 1*time.Second)
|
||
|
|
require.NoError(t, err)
|
||
|
|
w.Start(context.Background())
|
||
|
|
defer func() {
|
||
|
|
_ = w.Stop()
|
||
|
|
}()
|
||
|
|
|
||
|
|
for i := 0; i < 10; i++ {
|
||
|
|
filepath2 := createTempConfigFile(t, "temp_config2")
|
||
|
|
err = os.Rename(filepath2, filepath)
|
||
|
|
time.Sleep(timeoutDuration + 50*time.Millisecond)
|
||
|
|
require.NoError(t, err)
|
||
|
|
}
|
||
|
|
require.NoError(t, assertEvent(filepath, w.EventsCh(), defaultTimeout))
|
||
|
|
require.Error(t, assertEvent(filepath, w.EventsCh(), defaultTimeout), "expected timeout error")
|
||
|
|
}
|