57 lines
1 KiB
Go
57 lines
1 KiB
Go
|
// Copyright 2016 The Chihaya Authors. All rights reserved.
|
||
|
|
// Use of this source code is governed by the BSD 2-Clause license,
|
||
|
|
// which can be found in the LICENSE file.
|
||
|
|
|
||
|
|
package infohash
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"gopkg.in/yaml.v2"
|
||
|
|
|
||
|
|
"github.com/chihaya/chihaya"
|
||
|
|
"github.com/stretchr/testify/assert"
|
||
|
|
)
|
||
|
|
|
||
|
|
var (
|
||
|
|
configTemplate = `name: foo
|
||
|
|
config:
|
||
|
|
%s: %s`
|
||
|
|
|
||
|
|
data = []testData{
|
||
|
|
{"mode", "block", false, ModeBlock},
|
||
|
|
{"mode", "filter", false, ModeFilter},
|
||
|
|
{"some", "stuff", true, ModeBlock},
|
||
|
|
}
|
||
|
|
)
|
||
|
|
|
||
|
|
type testData struct {
|
||
|
|
key string
|
||
|
|
value string
|
||
|
|
err bool
|
||
|
|
expected Mode
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestNewConfig(t *testing.T) {
|
||
|
|
var mwconfig chihaya.MiddlewareConfig
|
||
|
|
|
||
|
|
cfg, err := newConfig(mwconfig)
|
||
|
|
assert.NotNil(t, err)
|
||
|
|
assert.Nil(t, cfg)
|
||
|
|
|
||
|
|
for _, test := range data {
|
||
|
|
config := fmt.Sprintf(configTemplate, test.key, test.value)
|
||
|
|
err = yaml.Unmarshal([]byte(config), &mwconfig)
|
||
|
|
assert.Nil(t, err)
|
||
|
|
|
||
|
|
cfg, err = newConfig(mwconfig)
|
||
|
|
if test.err {
|
||
|
|
assert.NotNil(t, err)
|
||
|
|
continue
|
||
|
|
}
|
||
|
|
assert.Nil(t, err)
|
||
|
|
assert.Equal(t, test.expected, cfg.Mode)
|
||
|
|
}
|
||
|
|
}
|