mirror of
https://github.com/slackhq/nebula.git
synced 2025-11-08 18:13:57 +01:00
879852c32a
Some checks failed
gofmt / Run gofmt (push) Successful in 37s
smoke-extra / Run extra smoke tests (push) Failing after 20s
smoke / Run multi node smoke test (push) Failing after 1m25s
Build and test / Build all and test on ubuntu-linux (push) Failing after 18m51s
Build and test / Build and test on linux with boringcrypto (push) Failing after 2m44s
Build and test / Build and test on linux with pkcs11 (push) Failing after 2m27s
Build and test / Build and test on macos-latest (push) Has been cancelled
Build and test / Build and test on windows-latest (push) Has been cancelled
* upgrade to yaml.v3 The main nice fix here is that maps unmarshal into `map[string]any` instead of `map[any]any`, so it cleans things up a bit. * add config.AsBool Since yaml.v3 doesn't automatically convert yes to bool now, for backwards compat * use type aliases for m * more cleanup * more cleanup * more cleanup * go mod cleanup
79 lines
1.8 KiB
Go
79 lines
1.8 KiB
Go
package nebula
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/slackhq/nebula/config"
|
|
"github.com/slackhq/nebula/test"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestNewPunchyFromConfig(t *testing.T) {
|
|
l := test.NewLogger()
|
|
c := config.NewC(l)
|
|
|
|
// Test defaults
|
|
p := NewPunchyFromConfig(l, c)
|
|
assert.False(t, p.GetPunch())
|
|
assert.False(t, p.GetRespond())
|
|
assert.Equal(t, time.Second, p.GetDelay())
|
|
assert.Equal(t, 5*time.Second, p.GetRespondDelay())
|
|
|
|
// punchy deprecation
|
|
c.Settings["punchy"] = true
|
|
p = NewPunchyFromConfig(l, c)
|
|
assert.True(t, p.GetPunch())
|
|
|
|
// punchy.punch
|
|
c.Settings["punchy"] = map[string]any{"punch": true}
|
|
p = NewPunchyFromConfig(l, c)
|
|
assert.True(t, p.GetPunch())
|
|
|
|
// punch_back deprecation
|
|
c.Settings["punch_back"] = true
|
|
p = NewPunchyFromConfig(l, c)
|
|
assert.True(t, p.GetRespond())
|
|
|
|
// punchy.respond
|
|
c.Settings["punchy"] = map[string]any{"respond": true}
|
|
c.Settings["punch_back"] = false
|
|
p = NewPunchyFromConfig(l, c)
|
|
assert.True(t, p.GetRespond())
|
|
|
|
// punchy.delay
|
|
c.Settings["punchy"] = map[string]any{"delay": "1m"}
|
|
p = NewPunchyFromConfig(l, c)
|
|
assert.Equal(t, time.Minute, p.GetDelay())
|
|
|
|
// punchy.respond_delay
|
|
c.Settings["punchy"] = map[string]any{"respond_delay": "1m"}
|
|
p = NewPunchyFromConfig(l, c)
|
|
assert.Equal(t, time.Minute, p.GetRespondDelay())
|
|
}
|
|
|
|
func TestPunchy_reload(t *testing.T) {
|
|
l := test.NewLogger()
|
|
c := config.NewC(l)
|
|
delay, _ := time.ParseDuration("1m")
|
|
require.NoError(t, c.LoadString(`
|
|
punchy:
|
|
delay: 1m
|
|
respond: false
|
|
`))
|
|
p := NewPunchyFromConfig(l, c)
|
|
assert.Equal(t, delay, p.GetDelay())
|
|
assert.False(t, p.GetRespond())
|
|
|
|
newDelay, _ := time.ParseDuration("10m")
|
|
require.NoError(t, c.ReloadConfigString(`
|
|
punchy:
|
|
delay: 10m
|
|
respond: true
|
|
`))
|
|
p.reload(c, false)
|
|
assert.Equal(t, newDelay, p.GetDelay())
|
|
assert.True(t, p.GetRespond())
|
|
}
|