412 lines
11 KiB
Go
412 lines
11 KiB
Go
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)
|
|
}
|
|
}
|