88 lines
2.1 KiB
Go
88 lines
2.1 KiB
Go
package config
|
|
|
|
import (
|
|
"os"
|
|
"testing"
|
|
|
|
"github.com/cloudflare/cloudflared/logger"
|
|
"github.com/cloudflare/cloudflared/watcher"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
type mockNotifier struct {
|
|
configs []Root
|
|
}
|
|
|
|
func (n *mockNotifier) ConfigDidUpdate(c Root) {
|
|
n.configs = append(n.configs, c)
|
|
}
|
|
|
|
type mockFileWatcher struct {
|
|
path string
|
|
notifier watcher.Notification
|
|
ready chan struct{}
|
|
}
|
|
|
|
func (w *mockFileWatcher) Start(n watcher.Notification) {
|
|
w.notifier = n
|
|
w.ready <- struct{}{}
|
|
}
|
|
|
|
func (w *mockFileWatcher) Add(string) error {
|
|
return nil
|
|
}
|
|
|
|
func (w *mockFileWatcher) Shutdown() {
|
|
|
|
}
|
|
|
|
func (w *mockFileWatcher) TriggerChange() {
|
|
w.notifier.WatcherItemDidChange(w.path)
|
|
}
|
|
|
|
func TestConfigChanged(t *testing.T) {
|
|
filePath := "config.yaml"
|
|
f, err := os.Create(filePath)
|
|
assert.NoError(t, err)
|
|
defer func() {
|
|
f.Close()
|
|
os.Remove(filePath)
|
|
}()
|
|
c := &Root{
|
|
Forwarders: []Forwarder{
|
|
{
|
|
URL: "test.daltoniam.com",
|
|
Listener: "127.0.0.1:8080",
|
|
},
|
|
},
|
|
}
|
|
configRead := func(configPath string) (Root, error) {
|
|
return *c, nil
|
|
}
|
|
wait := make(chan struct{})
|
|
w := &mockFileWatcher{path: filePath, ready: wait}
|
|
|
|
logger := logger.NewOutputWriter(logger.NewMockWriteManager())
|
|
|
|
service, err := NewFileManager(w, filePath, logger)
|
|
service.ReadConfig = configRead
|
|
assert.NoError(t, err)
|
|
|
|
n := &mockNotifier{}
|
|
go service.Start(n)
|
|
|
|
<-wait
|
|
c.Forwarders = append(c.Forwarders, Forwarder{URL: "add.daltoniam.com", Listener: "127.0.0.1:8081"})
|
|
w.TriggerChange()
|
|
|
|
service.Shutdown()
|
|
|
|
assert.Len(t, n.configs, 2, "did not get 2 config updates as expected")
|
|
assert.Len(t, n.configs[0].Forwarders, 1, "not the amount of forwarders expected")
|
|
assert.Len(t, n.configs[1].Forwarders, 2, "not the amount of forwarders expected")
|
|
|
|
assert.Equal(t, n.configs[0].Forwarders[0].Hash(), c.Forwarders[0].Hash(), "forwarder hashes don't match")
|
|
assert.Equal(t, n.configs[1].Forwarders[0].Hash(), c.Forwarders[0].Hash(), "forwarder hashes don't match")
|
|
assert.Equal(t, n.configs[1].Forwarders[1].Hash(), c.Forwarders[1].Hash(), "forwarder hashes don't match")
|
|
}
|