0ct0pu5/crowdsec
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

fix tests

Browse source
This commit is contained in:
alteredCoder 2023-09-29 19:14:26 +02:00
parent c7614a327a
commit 1eaaa2075d
2 changed files with 61 additions and 22 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

71
pkg/alertcontext/alertcontext_test.go
View file

@ -2,24 +2,34 @@ package alertcontext
import ( import (
"fmt" "fmt"
"path/filepath"
"testing" "testing"
"github.com/crowdsecurity/crowdsec/pkg/csconfig"
"github.com/crowdsecurity/crowdsec/pkg/models" "github.com/crowdsecurity/crowdsec/pkg/models"
"github.com/crowdsecurity/crowdsec/pkg/types" "github.com/crowdsecurity/crowdsec/pkg/types"
"github.com/stretchr/testify/assert" "github.com/stretchr/testify/assert"
"github.com/stretchr/testify/require"
) )
func TestNewAlertContext(t *testing.T) { func TestNewAlertContext(t *testing.T) {
contextFileFullPath, err := filepath.Abs("./tests/context.yaml")
require.NoError(t, err)
tests := []struct { tests := []struct {
name string name string
contextToSend map[string][]string contextToSend []csconfig.ContextToSend
valueLength int valueLength int
expectedErr error expectedErr error
}{ }{
{ {
name: "basic config test", name: "basic config test",
contextToSend: map[string][]string{ contextToSend: []csconfig.ContextToSend{
"test": {"evt.Parsed.source_ip"}, csconfig.ContextToSend{
SourceFile: contextFileFullPath,
Context: map[string][]string{
"source_ip": {"evt.Parsed.source_ip"},
},
},
}, },
valueLength: 100, valueLength: 100,
expectedErr: nil, expectedErr: nil,
@ -35,18 +45,26 @@ func TestNewAlertContext(t *testing.T) {
} }
func TestEventToContext(t *testing.T) { func TestEventToContext(t *testing.T) {
contextFileFullPath, err := filepath.Abs("./tests/context.yaml")
require.NoError(t, err)
tests := []struct { tests := []struct {
name string name string
contextToSend map[string][]string contextToSend []csconfig.ContextToSend
valueLength int valueLength int
events []types.Event events []types.Event
expectedResult models.Meta expectedResult models.Meta
}{ }{
{ {
name: "basic test", name: "basic test",
contextToSend: map[string][]string{ contextToSend: []csconfig.ContextToSend{
"source_ip": {"evt.Parsed.source_ip"}, csconfig.ContextToSend{
"nonexistent_field": {"evt.Parsed.nonexist"}, SourceFile: contextFileFullPath,
Context: map[string][]string{
"source_ip": {"evt.Parsed.source_ip"},
"nonexistent_field": {"evt.Parsed.nonexist"},
},
},
}, },
valueLength: 100, valueLength: 100,
events: []types.Event{ events: []types.Event{
@ -66,10 +84,14 @@ func TestEventToContext(t *testing.T) {
}, },
{ {
name: "test many events", name: "test many events",
contextToSend: map[string][]string{ contextToSend: []csconfig.ContextToSend{
"source_ip": {"evt.Parsed.source_ip"}, csconfig.ContextToSend{
"source_machine": {"evt.Parsed.source_machine"}, SourceFile: contextFileFullPath,
"cve": {"evt.Parsed.cve"}, Context: map[string][]string{
"source_ip": {"evt.Parsed.source_ip"},
"source_machine": {"evt.Parsed.source_machine"},
"cve": {"evt.Parsed.cve"}},
},
}, },
valueLength: 100, valueLength: 100,
events: []types.Event{ events: []types.Event{
@ -112,11 +134,17 @@ func TestEventToContext(t *testing.T) {
}, },
{ {
name: "test many events with result above max length (need truncate, keep only 2 on 3 elements)", name: "test many events with result above max length (need truncate, keep only 2 on 3 elements)",
contextToSend: map[string][]string{ contextToSend: []csconfig.ContextToSend{
"source_ip": {"evt.Parsed.source_ip"}, csconfig.ContextToSend{
"source_machine": {"evt.Parsed.source_machine"}, SourceFile: contextFileFullPath,
"uri": {"evt.Parsed.uri"}, Context: map[string][]string{
"source_ip": {"evt.Parsed.source_ip"},
"source_machine": {"evt.Parsed.source_machine"},
"uri": {"evt.Parsed.uri"},
},
},
}, },
valueLength: 100, valueLength: 100,
events: []types.Event{ events: []types.Event{
{ {
@ -158,10 +186,15 @@ func TestEventToContext(t *testing.T) {
}, },
{ {
name: "test one events with result above max length (need truncate on one element)", name: "test one events with result above max length (need truncate on one element)",
contextToSend: map[string][]string{ contextToSend: []csconfig.ContextToSend{
"source_ip": {"evt.Parsed.source_ip"}, csconfig.ContextToSend{
"source_machine": {"evt.Parsed.source_machine"}, SourceFile: contextFileFullPath,
"uri": {"evt.Parsed.uri"}, Context: map[string][]string{
"source_ip": {"evt.Parsed.source_ip"},
"source_machine": {"evt.Parsed.source_machine"},
"uri": {"evt.Parsed.uri"},
},
},
}, },
valueLength: 100, valueLength: 100,
events: []types.Event{ events: []types.Event{

12
pkg/csconfig/crowdsec_service_test.go
View file

@ -36,6 +36,9 @@ func TestLoadCrowdsec(t *testing.T) {
contextFileFullPath, err := filepath.Abs("./tests/context.yaml") contextFileFullPath, err := filepath.Abs("./tests/context.yaml")
require.NoError(t, err) require.NoError(t, err)
contextDirFullPath, err := filepath.Abs("./tests/context/")
require.NoError(t, err)
tests := []struct { tests := []struct {
name string name string
input *Config input *Config
@ -66,6 +69,7 @@ func TestLoadCrowdsec(t *testing.T) {
Enable: ptr.Of(true), Enable: ptr.Of(true),
AcquisitionDirPath: "", AcquisitionDirPath: "",
ContextPath: contextFileFullPath, ContextPath: contextFileFullPath,
ContextDir: contextDirFullPath,
AcquisitionFilePath: acquisFullPath, AcquisitionFilePath: acquisFullPath,
ConfigDir: configDirFullPath, ConfigDir: configDirFullPath,
DataDir: dataFullPath, DataDir: dataFullPath,
@ -79,7 +83,7 @@ func TestLoadCrowdsec(t *testing.T) {
SimulationFilePath: "./tests/simulation.yaml", SimulationFilePath: "./tests/simulation.yaml",
ContextToSend: []ContextToSend{ ContextToSend: []ContextToSend{
ContextToSend{ ContextToSend{
SourceFile: "./tests/context.yaml", SourceFile: contextFileFullPath,
Context: map[string][]string{ Context: map[string][]string{
"source_ip": {"evt.Parsed.source_ip"}, "source_ip": {"evt.Parsed.source_ip"},
}, },
@ -115,6 +119,7 @@ func TestLoadCrowdsec(t *testing.T) {
AcquisitionDirPath: acquisDirFullPath, AcquisitionDirPath: acquisDirFullPath,
AcquisitionFilePath: acquisFullPath, AcquisitionFilePath: acquisFullPath,
ContextPath: contextFileFullPath, ContextPath: contextFileFullPath,
ContextDir: contextDirFullPath,
ConfigDir: configDirFullPath, ConfigDir: configDirFullPath,
HubIndexFile: hubIndexFileFullPath, HubIndexFile: hubIndexFileFullPath,
DataDir: dataFullPath, DataDir: dataFullPath,
@ -126,7 +131,7 @@ func TestLoadCrowdsec(t *testing.T) {
AcquisitionFiles: []string{acquisFullPath, acquisInDirFullPath}, AcquisitionFiles: []string{acquisFullPath, acquisInDirFullPath},
ContextToSend: []ContextToSend{ ContextToSend: []ContextToSend{
ContextToSend{ ContextToSend{
SourceFile: "./tests/context.yaml", SourceFile: contextFileFullPath,
Context: map[string][]string{ Context: map[string][]string{
"source_ip": {"evt.Parsed.source_ip"}, "source_ip": {"evt.Parsed.source_ip"},
}, },
@ -169,11 +174,12 @@ func TestLoadCrowdsec(t *testing.T) {
ParserRoutinesCount: 1, ParserRoutinesCount: 1,
OutputRoutinesCount: 1, OutputRoutinesCount: 1,
ContextValueLength: 10, ContextValueLength: 10,
ContextDir: contextDirFullPath,
AcquisitionFiles: []string{}, AcquisitionFiles: []string{},
SimulationFilePath: "", SimulationFilePath: "",
ContextToSend: []ContextToSend{ ContextToSend: []ContextToSend{
ContextToSend{ ContextToSend{
SourceFile: "./tests/context.yaml", SourceFile: contextFileFullPath,
Context: map[string][]string{ Context: map[string][]string{
"source_ip": {"evt.Parsed.source_ip"}, "source_ip": {"evt.Parsed.source_ip"},
}, },