beenull/crowdsec
1
0
Fork 0
Code Issues Pull requests Packages Projects Releases Wiki Activity
crowdsec/pkg/csconfig/config_test.go
AlteredCoder 794d3221d0
add tests in pkg/csconfig and improve pkg/exprhelpers tests (#150) 
* add tests for csconfig & improve exprhelpers tests
2020-07-28 15:38:48 +02:00

412 lines
11 KiB
Go
Raw Blame History

package csconfig
import (
"flag"
"os"
"testing"
"github.com/crowdsecurity/crowdsec/pkg/outputs"
log "github.com/sirupsen/logrus"
"github.com/stretchr/testify/assert"
)
func TestNewCrowdSecConfig(t *testing.T) {
tests := []struct {
name string
expectedResult *CrowdSec
err string
}{
{
name: "new configuration: basic",
expectedResult: &CrowdSec{
LogLevel: log.InfoLevel,
Daemonize: false,
Profiling: false,
WorkingFolder: "/tmp/",
DataFolder: "/var/lib/crowdsec/data/",
ConfigFolder: "/etc/crowdsec/config/",
PIDFolder: "/var/run/",
LogFolder: "/var/log/",
LogMode: "stdout",
APIMode: false,
NbParsers: 1,
Prometheus: false,
HTTPListen: "127.0.0.1:6060",
},
err: "",
},
}
for _, test := range tests {
result := NewCrowdSecConfig()
isOk := assert.Equal(t, test.expectedResult, result)
if !isOk {
t.Fatalf("test '%s' failed", test.name)
}
log.Infof("test '%s' : OK", test.name)
}
}
func TestLoadConfig(t *testing.T) {
tests := []struct {
name string
expectedResult *CrowdSec
Args []string
err string
}{
{
name: "load configuration: basic",
expectedResult: &CrowdSec{
LogLevel: log.InfoLevel,
Daemonize: true,
Profiling: true,
WorkingFolder: "./tests/",
DataFolder: "./tests/",
ConfigFolder: "./tests/",
PIDFolder: "./tests/",
LogFolder: "./tests/",
LogMode: "stdout",
APIMode: true,
NbParsers: 1,
Prometheus: true,
HTTPListen: "127.0.0.1:6060",
AcquisitionFile: "tests/acquis.yaml",
CsCliFolder: "./tests/cscli/",
SimulationCfg: &SimulationConfig{
Simulation: false,
Exclusions: nil,
},
SimulationCfgPath: "./tests/simulation.yaml",
OutputConfig: &outputs.OutputFactory{
BackendFolder: "./tests/plugins/backend",
MaxRecords: "",
MaxRecordsAge: "720h",
Flush: false,
Debug: false,
},
},
Args: []string{
"crowdsec",
"-c",
"./tests/config.yaml",
},
err: "",
},
{
name: "load configuration: with -file",
expectedResult: &CrowdSec{
LogLevel: log.InfoLevel,
SingleFile: "./tests/test.file",
SingleFileLabel: "test",
Daemonize: true,
Profiling: true,
WorkingFolder: "./tests/",
DataFolder: "./tests/",
ConfigFolder: "./tests/",
PIDFolder: "./tests/",
LogFolder: "./tests/",
LogMode: "stdout",
APIMode: true,
NbParsers: 1,
Prometheus: true,
HTTPListen: "127.0.0.1:6060",
AcquisitionFile: "tests/acquis.yaml",
CsCliFolder: "./tests/cscli/",
SimulationCfg: &SimulationConfig{
Simulation: false,
Exclusions: nil,
},
SimulationCfgPath: "./tests/simulation.yaml",
OutputConfig: &outputs.OutputFactory{
BackendFolder: "./tests/plugins/backend",
MaxRecords: "",
MaxRecordsAge: "720h",
Flush: false,
Debug: false,
},
},
Args: []string{
"crowdsec",
"-c",
"./tests/config.yaml",
"-file",
"./tests/test.file",
"-type",
"test",
},
err: "",
},
{
name: "load configuration: with -file without -type",
expectedResult: &CrowdSec{
LogLevel: log.InfoLevel,
Daemonize: false,
Profiling: false,
WorkingFolder: "/tmp/",
DataFolder: "/var/lib/crowdsec/data/",
ConfigFolder: "/etc/crowdsec/config/",
PIDFolder: "/var/run/",
LogFolder: "/var/log/",
LogMode: "stdout",
APIMode: false,
NbParsers: 1,
Prometheus: false,
HTTPListen: "127.0.0.1:6060",
},
Args: []string{
"crowdsec",
"-c",
"./tests/config.yaml",
"-file",
"./tests/test.file",
},
err: "-file requires -type",
},
{
name: "load configuration: all flags set",
expectedResult: &CrowdSec{
LogLevel: log.TraceLevel,
Daemonize: true,
Profiling: true,
WorkingFolder: "./tests/",
DataFolder: "./tests/",
ConfigFolder: "./tests/",
PIDFolder: "./tests/",
LogFolder: "./tests/",
LogMode: "stdout",
APIMode: true,
Linter: true,
NbParsers: 1,
Prometheus: true,
HTTPListen: "127.0.0.1:6060",
AcquisitionFile: "./tests/acquis.yaml",
CsCliFolder: "./tests/cscli/",
SimulationCfg: &SimulationConfig{
Simulation: false,
Exclusions: nil,
},
SimulationCfgPath: "./tests/simulation.yaml",
OutputConfig: &outputs.OutputFactory{
BackendFolder: "./tests/plugins/backend",
MaxRecords: "",
MaxRecordsAge: "720h",
Flush: false,
Debug: false,
},
RestoreMode: "./tests/states.json",
DumpBuckets: true,
},
Args: []string{
"crowdsec",
"-c",
"./tests/config.yaml",
"-acquis",
"./tests/acquis.yaml",
"-dump-state",
"-prometheus-metrics",
"-t",
"-daemon",
"-profile",
"-debug",
"-trace",
"-info",
"-restore-state",
"./tests/states.json",
"-api",
},
err: "",
},
{
name: "load configuration: bad config file",
expectedResult: &CrowdSec{
LogLevel: log.InfoLevel,
Daemonize: true,
Profiling: true,
WorkingFolder: "./tests/",
DataFolder: "./tests/",
ConfigFolder: "./tests/",
PIDFolder: "./tests/",
LogFolder: "./tests/",
LogMode: "stdout",
APIMode: true,
Linter: false,
NbParsers: 1,
Prometheus: true,
HTTPListen: "127.0.0.1:6060",
CsCliFolder: "./tests/cscli/",
SimulationCfgPath: "./tests/simulation.yaml",
OutputConfig: &outputs.OutputFactory{
BackendFolder: "./tests/plugins/backend",
MaxRecords: "",
MaxRecordsAge: "720h",
Flush: false,
Debug: false,
},
},
Args: []string{
"crowdsec",
"-c",
"./tests/bad_config.yaml",
},
err: "Error while loading configuration : parse './tests/bad_config.yaml' : yaml: unmarshal errors:\n line 1: field non_existing_field not found in type csconfig.CrowdSec",
},
{
name: "load configuration: bad simulation file",
expectedResult: &CrowdSec{
LogLevel: log.InfoLevel,
Daemonize: true,
Profiling: true,
WorkingFolder: "./tests/",
DataFolder: "./tests/",
ConfigFolder: "./tests/",
PIDFolder: "./tests/",
LogFolder: "./tests/",
LogMode: "stdout",
APIMode: true,
Linter: false,
NbParsers: 1,
Prometheus: true,
AcquisitionFile: "tests/acquis.yaml",
HTTPListen: "127.0.0.1:6060",
CsCliFolder: "./tests/cscli/",
SimulationCfgPath: "./tests/bad_simulation.yaml",
OutputConfig: &outputs.OutputFactory{
BackendFolder: "./tests/plugins/backend",
MaxRecords: "",
MaxRecordsAge: "720h",
Flush: false,
Debug: false,
},
},
Args: []string{
"crowdsec",
"-c",
"./tests/bad_config_simulation.yaml",
},
err: `Error while loading configuration : loading simulation config : while parsing './tests/bad_simulation.yaml' : yaml: unmarshal errors:
line 1: field test not found in type csconfig.SimulationConfig`,
},
{
name: "load configuration: bad config file",
expectedResult: &CrowdSec{
LogLevel: log.InfoLevel,
Daemonize: true,
Profiling: true,
WorkingFolder: "./tests/",
DataFolder: "./tests/",
ConfigFolder: "./tests/",
PIDFolder: "./tests/",
LogFolder: "./tests/",
LogMode: "stdout",
APIMode: true,
Linter: false,
NbParsers: 1,
Prometheus: true,
HTTPListen: "127.0.0.1:6060",
CsCliFolder: "./tests/cscli/",
SimulationCfgPath: "./tests/simulation.yaml",
OutputConfig: &outputs.OutputFactory{
BackendFolder: "./tests/plugins/backend",
MaxRecords: "",
MaxRecordsAge: "720h",
Flush: false,
Debug: false,
},
},
Args: []string{
"crowdsec",
"-c",
"./tests/bad_config.yaml",
},
err: "Error while loading configuration : parse './tests/bad_config.yaml' : yaml: unmarshal errors:\n line 1: field non_existing_field not found in type csconfig.CrowdSec",
},
{
name: "load configuration: non exist simulation file",
expectedResult: &CrowdSec{
LogLevel: log.InfoLevel,
Daemonize: true,
Profiling: true,
WorkingFolder: "./tests/",
DataFolder: "./tests/",
ConfigFolder: "./tests/",
PIDFolder: "./tests/",
LogFolder: "./tests/",
LogMode: "stdout",
APIMode: true,
Linter: false,
NbParsers: 1,
Prometheus: true,
AcquisitionFile: "tests/acquis.yaml",
HTTPListen: "127.0.0.1:6060",
CsCliFolder: "./tests/cscli/",
SimulationCfgPath: "./tests/non_exist.yaml",
OutputConfig: &outputs.OutputFactory{
BackendFolder: "./tests/plugins/backend",
MaxRecords: "",
MaxRecordsAge: "720h",
Flush: false,
Debug: false,
},
},
Args: []string{
"crowdsec",
"-c",
"./tests/bad_config_simulation_1.yaml",
},
err: "Error while loading configuration : loading simulation config : while reading './tests/non_exist.yaml' : open ./tests/non_exist.yaml: no such file or directory",
},
{
name: "load configuration: non existent configuration file",
expectedResult: &CrowdSec{
LogLevel: log.InfoLevel,
Daemonize: false,
Profiling: false,
WorkingFolder: "/tmp/",
DataFolder: "/var/lib/crowdsec/data/",
ConfigFolder: "/etc/crowdsec/config/",
PIDFolder: "/var/run/",
LogFolder: "/var/log/",
LogMode: "stdout",
APIMode: false,
NbParsers: 1,
Prometheus: false,
HTTPListen: "127.0.0.1:6060",
},
Args: []string{
"crowdsec",
"-c",
"./tests/non_exist.yaml",
},
err: "Error while loading configuration : read './tests/non_exist.yaml' : open ./tests/non_exist.yaml: no such file or directory",
},
}
oldArgs := os.Args
defer func() { os.Args = oldArgs }()
for _, test := range tests {
log.Printf("testing '%s'", test.name)
flag.CommandLine = flag.NewFlagSet(test.Args[0], flag.ExitOnError)
result := NewCrowdSecConfig()
os.Args = test.Args
err := result.LoadConfig()
if test.err != "" {
if err == nil {
t.Fatalf("test '%s' should returned an error", test.name)
}
isOk := assert.EqualErrorf(t, err, test.err, "")
if !isOk {
t.Fatalf("test '%s' failed", test.name)
}
}
if test.err == "" && err != nil {
t.Fatalf("test '%s' return an error : %s", test.name, err)
}
isOk := assert.Equal(t, test.expectedResult, result)
if !isOk {
t.Fatalf("test '%s' failed", test.name)
}
log.Infof("test '%s' : OK", test.name)
}
}
Reference in a new issue View git blame Copy permalink