122 lines
2.3 KiB
Go
122 lines
2.3 KiB
Go
package all
|
|
|
|
import (
|
|
"errors"
|
|
"sync"
|
|
"testing"
|
|
|
|
"github.com/FreifunkBremen/yanic/output"
|
|
"github.com/FreifunkBremen/yanic/runtime"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
type testOutput struct {
|
|
output.Output
|
|
countSave int
|
|
sync.Mutex
|
|
}
|
|
|
|
func (c *testOutput) Save(nodes *runtime.Nodes) {
|
|
c.Lock()
|
|
c.countSave++
|
|
c.Unlock()
|
|
}
|
|
func (c *testOutput) Get() int {
|
|
c.Lock()
|
|
defer c.Unlock()
|
|
return c.countSave
|
|
}
|
|
|
|
func TestStart(t *testing.T) {
|
|
assert := assert.New(t)
|
|
|
|
nodes := &runtime.Nodes{}
|
|
|
|
globalOutput := &testOutput{}
|
|
output.RegisterAdapter("a", func(config map[string]interface{}) (output.Output, error) {
|
|
return globalOutput, nil
|
|
})
|
|
output.RegisterAdapter("b", func(config map[string]interface{}) (output.Output, error) {
|
|
return globalOutput, nil
|
|
})
|
|
output.RegisterAdapter("c", func(config map[string]interface{}) (output.Output, error) {
|
|
return globalOutput, nil
|
|
})
|
|
output.RegisterAdapter("d", func(config map[string]interface{}) (output.Output, error) {
|
|
return nil, nil
|
|
})
|
|
output.RegisterAdapter("e", func(config map[string]interface{}) (output.Output, error) {
|
|
return nil, errors.New("blub")
|
|
})
|
|
allOutput, err := Register(map[string]interface{}{
|
|
"a": []map[string]interface{}{
|
|
{
|
|
"enable": false,
|
|
"path": "a1",
|
|
},
|
|
{
|
|
"path": "a2",
|
|
},
|
|
{
|
|
"enable": true,
|
|
"path": "a3",
|
|
},
|
|
},
|
|
"b": nil,
|
|
"c": []map[string]interface{}{
|
|
{
|
|
"path": "c1",
|
|
"filter": map[string]interface{}{},
|
|
},
|
|
},
|
|
// fetch continue command in Connect
|
|
"d": []map[string]interface{}{
|
|
{
|
|
"path": "d0",
|
|
},
|
|
},
|
|
})
|
|
assert.NoError(err)
|
|
|
|
assert.Equal(0, globalOutput.Get())
|
|
allOutput.Save(nodes)
|
|
assert.Equal(3, globalOutput.Get())
|
|
|
|
// wrong format - map
|
|
_, err = Register(map[string]interface{}{
|
|
"e": []interface{}{
|
|
false,
|
|
},
|
|
})
|
|
assert.Error(err)
|
|
|
|
// wrong format - array
|
|
_, err = Register(map[string]interface{}{
|
|
"e": true,
|
|
})
|
|
assert.Error(err)
|
|
|
|
// output error
|
|
_, err = Register(map[string]interface{}{
|
|
"e": []interface{}{
|
|
map[string]interface{}{
|
|
"enable": true,
|
|
},
|
|
},
|
|
})
|
|
assert.Error(err)
|
|
|
|
// output error invalid config of filter
|
|
_, err = Register(map[string]interface{}{
|
|
"a": []interface{}{
|
|
map[string]interface{}{
|
|
"enable": true,
|
|
"filter": map[string]interface{}{
|
|
"blocklist": true,
|
|
},
|
|
},
|
|
},
|
|
})
|
|
assert.Error(err)
|
|
}
|