65 lines
1.5 KiB
Go
65 lines
1.5 KiB
Go
package csconfig
|
|
|
|
import (
|
|
"fmt"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/crowdsecurity/crowdsec/pkg/types"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestLoadDBConfig(t *testing.T) {
|
|
tests := []struct {
|
|
name string
|
|
Input *Config
|
|
expectedResult *DatabaseCfg
|
|
err string
|
|
}{
|
|
{
|
|
name: "basic valid configuration",
|
|
Input: &Config{
|
|
DbConfig: &DatabaseCfg{
|
|
Type: "sqlite",
|
|
DbPath: "./tests/test.db",
|
|
MaxOpenConns: types.IntPtr(10),
|
|
},
|
|
Cscli: &CscliCfg{},
|
|
API: &APICfg{
|
|
Server: &LocalApiServerCfg{},
|
|
},
|
|
},
|
|
expectedResult: &DatabaseCfg{
|
|
Type: "sqlite",
|
|
DbPath: "./tests/test.db",
|
|
MaxOpenConns: types.IntPtr(10),
|
|
},
|
|
},
|
|
{
|
|
name: "no configuration path",
|
|
Input: &Config{},
|
|
expectedResult: nil,
|
|
},
|
|
}
|
|
|
|
for idx, test := range tests {
|
|
err := test.Input.LoadDBConfig()
|
|
if err == nil && test.err != "" {
|
|
fmt.Printf("TEST '%s': NOK\n", test.name)
|
|
t.Fatalf("%d/%d expected error, didn't get it", idx, len(tests))
|
|
} else if test.err != "" {
|
|
if !strings.HasPrefix(fmt.Sprintf("%s", err), test.err) {
|
|
fmt.Printf("TEST '%s': NOK\n", test.name)
|
|
t.Fatalf("%d/%d expected '%s' got '%s'", idx, len(tests),
|
|
test.err,
|
|
fmt.Sprintf("%s", err))
|
|
}
|
|
}
|
|
isOk := assert.Equal(t, test.expectedResult, test.Input.DbConfig)
|
|
if !isOk {
|
|
t.Fatalf("TEST '%s': NOK", test.name)
|
|
} else {
|
|
fmt.Printf("TEST '%s': OK\n", test.name)
|
|
}
|
|
}
|
|
}
|