enricher_test.go

package main

import (
	"bytes"
	"encoding/json"
	"os"
	"testing"

	"github.com/stretchr/testify/require"
)

func TestEnricher(t *testing.T) {
	table := []struct {
		file string
	}{
		{file: "testdata/record.txt"},
		{file: "testdata/record1.txt"},
		{file: "testdata/record2.txt"},
		{file: "testdata/record3.txt"},
		{file: "testdata/record4.txt"},
	}

	for _, tt := range table {
		sc, err := newEnricher(
			enricherWithCRS("crs/coraza.conf", "crs/crs-setup.conf", "crs/rules/*.conf"),
			enricherWithOptionalGeoIP("testdata/GeoLite2-City-Test.mmdb"),
		)
		require.Nil(t, err, "unexpected error in instantiating scorer")

		f, err := os.ReadFile(tt.file)
		require.Nil(t, err, "unexpected error in reading test data")

		s, err := sc.EnrichRecord(bytes.NewReader(f))
		require.NoError(t, err, "should not return error")
		require.NotNil(t, s, "should produce non nil score")
		defer s.Close()
		ecs, err := s.toECS()
		require.NoError(t, err, "should not return error")
		require.NotNil(t, ecs, "should produce non nil score")
		b, _ := json.Marshal(ecs)
		t.Log(string(b))
	}
}