2021-01-13 16:59:08 +00:00
|
|
|
package sink_test
|
2021-01-02 16:24:06 +00:00
|
|
|
|
|
|
|
import (
|
2021-01-13 16:59:08 +00:00
|
|
|
"crypto/tls"
|
|
|
|
"net"
|
|
|
|
"net/http"
|
2021-01-02 16:24:06 +00:00
|
|
|
"sync"
|
|
|
|
"testing"
|
|
|
|
"time"
|
|
|
|
|
|
|
|
"github.com/golang/mock/gomock"
|
|
|
|
audit_mock "gitlab.com/inetmock/inetmock/internal/mock/audit"
|
|
|
|
"gitlab.com/inetmock/inetmock/pkg/audit"
|
2021-01-13 16:59:08 +00:00
|
|
|
"gitlab.com/inetmock/inetmock/pkg/audit/details"
|
|
|
|
"gitlab.com/inetmock/inetmock/pkg/audit/sink"
|
2021-01-02 16:24:06 +00:00
|
|
|
"gitlab.com/inetmock/inetmock/pkg/logging"
|
2021-01-13 16:59:08 +00:00
|
|
|
"gitlab.com/inetmock/inetmock/pkg/wait"
|
|
|
|
)
|
|
|
|
|
|
|
|
var (
|
|
|
|
testEvents = []*audit.Event{
|
|
|
|
{
|
|
|
|
Transport: audit.TransportProtocol_TCP,
|
|
|
|
Application: audit.AppProtocol_HTTP,
|
|
|
|
SourceIP: net.ParseIP("127.0.0.1").To4(),
|
|
|
|
DestinationIP: net.ParseIP("127.0.0.1").To4(),
|
|
|
|
SourcePort: 32344,
|
|
|
|
DestinationPort: 80,
|
|
|
|
TLS: &audit.TLSDetails{
|
|
|
|
Version: tls.VersionTLS13,
|
|
|
|
CipherSuite: tls.TLS_ECDHE_ECDSA_WITH_AES_256_CBC_SHA,
|
|
|
|
ServerName: "localhost",
|
|
|
|
},
|
|
|
|
ProtocolDetails: details.HTTP{
|
|
|
|
Method: "GET",
|
|
|
|
Host: "localhost",
|
|
|
|
URI: "http://localhost/asdf",
|
|
|
|
Proto: "HTTP 1.1",
|
|
|
|
Headers: http.Header{
|
|
|
|
"Accept": []string{"application/json"},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
{
|
|
|
|
Transport: audit.TransportProtocol_TCP,
|
|
|
|
Application: audit.AppProtocol_DNS,
|
|
|
|
SourceIP: net.ParseIP("::1").To16(),
|
|
|
|
DestinationIP: net.ParseIP("::1").To16(),
|
|
|
|
SourcePort: 32344,
|
|
|
|
DestinationPort: 80,
|
|
|
|
},
|
|
|
|
}
|
2021-01-02 16:24:06 +00:00
|
|
|
)
|
|
|
|
|
|
|
|
func Test_writerCloserSink_OnSubscribe(t *testing.T) {
|
|
|
|
type testCase struct {
|
|
|
|
name string
|
2021-01-07 21:00:12 +00:00
|
|
|
events []*audit.Event
|
2021-01-02 16:24:06 +00:00
|
|
|
}
|
|
|
|
tests := []testCase{
|
|
|
|
{
|
|
|
|
name: "Get a single event",
|
|
|
|
events: testEvents[:1],
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "Get multiple events",
|
|
|
|
events: testEvents,
|
|
|
|
},
|
|
|
|
}
|
|
|
|
scenario := func(tt testCase) func(t *testing.T) {
|
|
|
|
return func(t *testing.T) {
|
|
|
|
wg := new(sync.WaitGroup)
|
|
|
|
wg.Add(len(tt.events))
|
|
|
|
|
|
|
|
ctrl := gomock.NewController(t)
|
|
|
|
t.Cleanup(ctrl.Finish)
|
|
|
|
writerMock := audit_mock.NewMockWriter(ctrl)
|
|
|
|
writerMock.
|
|
|
|
EXPECT().
|
|
|
|
Write(gomock.Any()).
|
|
|
|
Do(func(_ *audit.Event) {
|
|
|
|
wg.Done()
|
|
|
|
}).
|
|
|
|
Times(len(tt.events))
|
|
|
|
|
2021-01-13 16:59:08 +00:00
|
|
|
writerCloserSink := sink.NewWriterSink("WriterMock", writerMock, sink.WithCloseOnExit)
|
2021-01-02 16:24:06 +00:00
|
|
|
var evs audit.EventStream
|
|
|
|
var err error
|
|
|
|
|
|
|
|
if evs, err = audit.NewEventStream(logging.CreateTestLogger(t)); err != nil {
|
|
|
|
t.Errorf("NewEventStream() error = %v", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
if err = evs.RegisterSink(writerCloserSink); err != nil {
|
|
|
|
t.Errorf("RegisterSink() error = %v", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
for _, ev := range tt.events {
|
2021-01-07 21:00:12 +00:00
|
|
|
evs.Emit(*ev)
|
2021-01-02 16:24:06 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
select {
|
|
|
|
case <-time.After(100 * time.Millisecond):
|
|
|
|
t.Errorf("not all events recorded in time")
|
2021-01-13 16:59:08 +00:00
|
|
|
case <-wait.ForWaitGroupDone(wg):
|
2021-01-02 16:24:06 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
for _, tt := range tests {
|
|
|
|
t.Run(tt.name, scenario(tt))
|
|
|
|
}
|
|
|
|
}
|