package audit_test

import (
	"sync"
	"testing"
	"time"

	"github.com/golang/mock/gomock"
	logging_mock "gitlab.com/inetmock/inetmock/internal/mock/logging"
	"gitlab.com/inetmock/inetmock/pkg/audit"
	"gitlab.com/inetmock/inetmock/pkg/logging"
	"go.uber.org/zap"
)

//nolint:dupl
func Test_logSink_OnSubscribe(t *testing.T) {
	type fields struct {
		loggerSetup func(t *testing.T, wg *sync.WaitGroup) logging.Logger
	}
	type testCase struct {
		name   string
		fields fields
		events []audit.Event
	}
	tests := []testCase{
		{
			name: "Get a single log line",
			fields: fields{
				loggerSetup: func(t *testing.T, wg *sync.WaitGroup) logging.Logger {
					ctrl := gomock.NewController(t)
					t.Cleanup(ctrl.Finish)
					loggerMock := logging_mock.NewMockLogger(ctrl)

					loggerMock.
						EXPECT().
						With(gomock.Any()).
						Return(loggerMock)

					loggerMock.
						EXPECT().
						Info("handled request", gomock.Any()).
						Do(func(_ string, _ ...zap.Field) {
							wg.Done()
						}).
						Times(1)

					return loggerMock
				},
			},
			events: testEvents[:1],
		},
		{
			name: "Get multiple events",
			fields: fields{
				loggerSetup: func(t *testing.T, wg *sync.WaitGroup) logging.Logger {
					ctrl := gomock.NewController(t)
					t.Cleanup(ctrl.Finish)
					loggerMock := logging_mock.NewMockLogger(ctrl)

					loggerMock.
						EXPECT().
						With(gomock.Any()).
						Return(loggerMock)

					loggerMock.
						EXPECT().
						Info("handled request", gomock.Any()).
						Do(func(_ string, _ ...zap.Field) {
							wg.Done()
						}).
						Times(2)

					return loggerMock
				},
			},
			events: testEvents,
		},
	}
	scenario := func(tt testCase) func(t *testing.T) {
		return func(t *testing.T) {
			wg := new(sync.WaitGroup)
			wg.Add(len(tt.events))

			logSink := audit.NewLogSink(tt.fields.loggerSetup(t, wg))
			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(logSink); err != nil {
				t.Errorf("RegisterSink() error = %v", err)
			}

			for _, ev := range tt.events {
				evs.Emit(ev)
			}

			select {
			case <-time.After(100 * time.Millisecond):
				t.Errorf("not all events recorded in time")
			case <-waitGroupDone(wg):
			}
		}
	}

	for _, tt := range tests {
		t.Run(tt.name, scenario(tt))
	}
}