123 lines
2.5 KiB
Go
123 lines
2.5 KiB
Go
|
package mcfg
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
. "testing"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"github.com/stretchr/testify/assert"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestHook(t *T) {
|
||
|
|
{ // test Then
|
||
|
|
aCh := make(chan bool)
|
||
|
|
bCh := make(chan bool)
|
||
|
|
h := Nop()
|
||
|
|
h.Then(func(context.Context) error {
|
||
|
|
aCh <- true
|
||
|
|
<-aCh
|
||
|
|
return nil
|
||
|
|
})
|
||
|
|
h.Then(func(context.Context) error {
|
||
|
|
bCh <- true
|
||
|
|
<-bCh
|
||
|
|
return nil
|
||
|
|
})
|
||
|
|
errCh := make(chan error)
|
||
|
|
go func() {
|
||
|
|
errCh <- h(nil)
|
||
|
|
}()
|
||
|
|
|
||
|
|
assert.True(t, <-aCh)
|
||
|
|
// make sure bCh isn't being written to till aCh is closed
|
||
|
|
select {
|
||
|
|
case <-bCh:
|
||
|
|
assert.Fail(t, "bCh shouldn't be written to yet")
|
||
|
|
case <-time.After(250 * time.Millisecond):
|
||
|
|
close(aCh)
|
||
|
|
}
|
||
|
|
assert.True(t, <-bCh)
|
||
|
|
// make sure errCh isn't being written to till bCh is closed
|
||
|
|
select {
|
||
|
|
case <-errCh:
|
||
|
|
assert.Fail(t, "errCh shouldn't be written to yet")
|
||
|
|
case <-time.After(250 * time.Millisecond):
|
||
|
|
close(bCh)
|
||
|
|
}
|
||
|
|
assert.Nil(t, <-errCh)
|
||
|
|
}
|
||
|
|
|
||
|
|
{ // test Also
|
||
|
|
aCh := make(chan bool)
|
||
|
|
bCh := make(chan bool)
|
||
|
|
h := Nop()
|
||
|
|
h.Also(func(context.Context) error {
|
||
|
|
aCh <- true
|
||
|
|
<-aCh
|
||
|
|
return nil
|
||
|
|
})
|
||
|
|
h.Also(func(context.Context) error {
|
||
|
|
bCh <- true
|
||
|
|
<-bCh
|
||
|
|
return nil
|
||
|
|
})
|
||
|
|
errCh := make(chan error)
|
||
|
|
go func() {
|
||
|
|
errCh <- h(nil)
|
||
|
|
}()
|
||
|
|
|
||
|
|
// both channels should get written to, then closed, then errCh should
|
||
|
|
// get written to
|
||
|
|
assert.True(t, <-aCh)
|
||
|
|
assert.True(t, <-bCh)
|
||
|
|
// make sure errCh isn't being written to till both channels are written
|
||
|
|
select {
|
||
|
|
case <-errCh:
|
||
|
|
assert.Fail(t, "errCh shouldn't be written to yet")
|
||
|
|
case <-time.After(250 * time.Millisecond):
|
||
|
|
close(aCh)
|
||
|
|
close(bCh)
|
||
|
|
}
|
||
|
|
assert.Nil(t, <-errCh)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestPopulateParams(t *T) {
|
||
|
|
{
|
||
|
|
cfg := New()
|
||
|
|
a := cfg.ParamInt("a", 0, "")
|
||
|
|
cfgChild := cfg.Child("foo")
|
||
|
|
b := cfgChild.ParamInt("b", 0, "")
|
||
|
|
c := cfgChild.ParamInt("c", 0, "")
|
||
|
|
|
||
|
|
err := cfg.populateParams(SourceCLI{
|
||
|
|
Args: []string{"--a=1", "--foo-b=2"},
|
||
|
|
})
|
||
|
|
assert.NoError(t, err)
|
||
|
|
assert.Equal(t, 1, *a)
|
||
|
|
assert.Equal(t, 2, *b)
|
||
|
|
assert.Equal(t, 0, *c)
|
||
|
|
}
|
||
|
|
|
||
|
|
{ // test that required params are enforced
|
||
|
|
cfg := New()
|
||
|
|
a := cfg.ParamInt("a", 0, "")
|
||
|
|
cfgChild := cfg.Child("foo")
|
||
|
|
b := cfgChild.ParamInt("b", 0, "")
|
||
|
|
c := cfgChild.ParamRequiredInt("c", "")
|
||
|
|
|
||
|
|
err := cfg.populateParams(SourceCLI{
|
||
|
|
Args: []string{"--a=1", "--foo-b=2"},
|
||
|
|
})
|
||
|
|
assert.Error(t, err)
|
||
|
|
|
||
|
|
err = cfg.populateParams(SourceCLI{
|
||
|
|
Args: []string{"--a=1", "--foo-b=2", "--foo-c=3"},
|
||
|
|
})
|
||
|
|
assert.NoError(t, err)
|
||
|
|
assert.Equal(t, 1, *a)
|
||
|
|
assert.Equal(t, 2, *b)
|
||
|
|
assert.Equal(t, 3, *c)
|
||
|
|
}
|
||
|
|
}
|