more tests

Author: ncode

pkg/auditserver/server_test.go

@@ -3,7 +3,11 @@ package auditserver
 import (
 	"bytes"
 	"encoding/json"
+	"errors"
 	"fmt"
+	"github.com/ncode/vault-audit-filter/pkg/forwarder"
+	"github.com/ncode/vault-audit-filter/pkg/messaging"
+	"github.com/stretchr/testify/require"
 	"io"
 	"io/ioutil"
 	"net"
@@ -799,3 +803,102 @@ func TestAuditServer_React_WithForwarding(t *testing.T) {
 		})
 	}
 }
+
+type dummyMessenger struct {
+	sendErr error
+	calls   int
+}
+
+func (d *dummyMessenger) Send(_ string) error {
+	d.calls++
+	return d.sendErr
+}
+
+type dummyForwarder struct {
+	forwardErr error
+	calls      int
+}
+
+func (d *dummyForwarder) Forward(_ []byte) error {
+	d.calls++
+	return d.forwardErr
+}
+
+func mustAuditLogBytes(t *testing.T) []byte {
+	t.Helper()
+	json := `{"type":"request","time":"2000-01-01T00:00:00Z","auth":{},"request":{},"response":{}}`
+	return []byte(json)
+}
+
+func newRuleGroup(match bool, msgr messaging.Messenger, fwd forwarder.Forwarder, wr bytes.Buffer) RuleGroup {
+	rule := CompiledRule{}
+	if !match {
+		// compiledRules ≠ 0 + rule that always returns false
+		rule = CompiledRule{Program: nil}
+	}
+	return RuleGroup{
+		Name:          "grp",
+		CompiledRules: []CompiledRule{rule},
+		Messenger:     msgr,
+		Forwarder:     fwd,
+		Writer:        &wr,
+	}
+}
+
+func TestReact_Branches(t *testing.T) {
+	frame := mustAuditLogBytes(t)
+
+	tests := []struct {
+		name         string
+		group        RuleGroup
+		wantAction   gnet.Action
+		wantMsgCalls int
+		wantFwdCalls int
+	}{
+		{
+			name:       "match_no_side_effects_returns_None",
+			group:      newRuleGroup(true, nil, nil, bytes.Buffer{}),
+			wantAction: gnet.None,
+		},
+		{
+			name:         "forwarder_ok_triggers_Close",
+			group:        newRuleGroup(true, nil, &dummyForwarder{}, bytes.Buffer{}),
+			wantAction:   gnet.Close,
+			wantFwdCalls: 1,
+		},
+		{
+			name:         "forwarder_error_triggers_Close",
+			group:        newRuleGroup(true, nil, &dummyForwarder{forwardErr: errors.New("x")}, bytes.Buffer{}),
+			wantAction:   gnet.Close,
+			wantFwdCalls: 1,
+		},
+		{
+			name:         "messenger_error_triggers_Close",
+			group:        newRuleGroup(true, &dummyMessenger{sendErr: errors.New("x")}, nil, bytes.Buffer{}),
+			wantAction:   gnet.Close,
+			wantMsgCalls: 1,
+		},
+		{
+			name:       "no_match_triggers_Close",
+			group:      newRuleGroup(false, nil, nil, bytes.Buffer{}),
+			wantAction: gnet.Close,
+		},
+	}
+
+	for _, tc := range tests {
+		tc := tc // capture
+		t.Run(tc.name, func(t *testing.T) {
+			srv := &AuditServer{ruleGroups: []RuleGroup{tc.group}}
+			_, act := srv.React(frame, nil)
+
+			require.Equal(t, tc.wantAction, act)
+
+			if dm, ok := tc.group.Messenger.(*dummyMessenger); ok {
+				require.Equal(t, tc.wantMsgCalls, dm.calls)
+			}
+			if df, ok := tc.group.Forwarder.(*dummyForwarder); ok {
+				require.Equal(t, tc.wantFwdCalls, df.calls)
+			}
+		})
+	}
+}