diff options
Diffstat (limited to 'internal/events/bus_test.go')
| -rw-r--r-- | internal/events/bus_test.go | 746 |
1 files changed, 746 insertions, 0 deletions
diff --git a/internal/events/bus_test.go b/internal/events/bus_test.go new file mode 100644 index 0000000..633d950 --- /dev/null +++ b/internal/events/bus_test.go @@ -0,0 +1,746 @@ +package events_test + +import ( + "net/http" + "net/http/httptest" + "sync" + "sync/atomic" + "testing" + "time" + + "github.com/shiv/internal/events" + "github.com/stretchr/testify/assert" + "github.com/stretchr/testify/require" +) + +// ── helpers ─────────────────────────────────────────────────────────────────── + +func newRequest(t *testing.T, method, url string) *http.Request { + t.Helper() + return httptest.NewRequest(method, url, nil) +} + +func newRequestEvent(t *testing.T) events.RequestEvent { + t.Helper() + return events.RequestEvent{ + Request: newRequest(t, http.MethodGet, "http://example.com/path"), + Body: []byte("request body"), + } +} + +func newResponseEvent() events.ResponseEvent { + return events.ResponseEvent{ + Timestamp: time.Now(), + Host: "example.com", + Proto: "HTTP/1.1", + Method: "GET", + URL: "http://example.com/path", + StatusCode: 200, + RespBody: []byte("response body"), + DurationMs: 42, + TLS: false, + } +} + +func newWSConnectionEvent() events.WebSocketConnectionEvent { + return events.WebSocketConnectionEvent{ + Host: "example.com:443", + URL: "wss://example.com/ws", + TLS: true, + Timestamp: time.Now(), + } +} + +func newWSFrameEvent(connID uint64) events.WebSocketFrameEvent { + return events.WebSocketFrameEvent{ + ConnectionID: connID, + Timestamp: time.Now(), + Direction: events.WebSocketClient, + Opcode: events.WebSocketText, + Payload: []byte("hello"), + } +} + +// ── Register ────────────────────────────────────────────────────────────────── + +func TestRegister_PanicsOnUnknownType(t *testing.T) { + b := events.NewBus() + assert.Panics(t, func() { + b.Register(struct{}{}) + }) +} + +func TestRegister_AcceptsRequestMiddleware(t *testing.T) { + b := events.NewBus() + require.NotPanics(t, func() { + b.Register(events.RequestMiddlewareFunc(func(e events.RequestEvent) events.RequestResult { + return events.RequestResult{Request: e.Request, Body: e.Body} + })) + }) +} + +func TestRegister_AcceptsResponseObserver(t *testing.T) { + b := events.NewBus() + require.NotPanics(t, func() { + b.Register(events.ResponseObserverFunc(func(e events.ResponseEvent) {})) + }) +} + +func TestRegister_AcceptsWebSocketConnectionObserver(t *testing.T) { + b := events.NewBus() + require.NotPanics(t, func() { + b.Register(events.WebSocketConnectionObserverFunc(func(e events.WebSocketConnectionEvent) uint64 { return 1 })) + }) +} + +func TestRegister_AcceptsWebSocketFrameObserver(t *testing.T) { + b := events.NewBus() + require.NotPanics(t, func() { + b.Register(events.WebSocketFrameObserverFunc(func(e events.WebSocketFrameEvent) events.WebSocketFrameResult { + return events.WebSocketFrameResult{Payload: e.Payload} + })) + }) +} + +func TestRegister_AcceptsPluginLogObserver(t *testing.T) { + b := events.NewBus() + require.NotPanics(t, func() { + b.Register(events.PluginLogObserverFunc(func(e events.PluginLogEvent) {})) + }) +} + +func TestRegister_AcceptsPluginEnabledObserver(t *testing.T) { + b := events.NewBus() + require.NotPanics(t, func() { + b.Register(events.PluginEnabledObserverFunc(func(e events.SetPluginEnabledEvent) {})) + }) +} + +func TestRegister_AcceptsLoadPluginObserver(t *testing.T) { + b := events.NewBus() + require.NotPanics(t, func() { + b.Register(events.LoadPluginObserverFunc(func(e events.LoadPluginEvent) {})) + }) +} + +func TestRegister_MultiInterfaceTypeRegisteredForAll(t *testing.T) { + b := events.NewBus() + + var requestCalled, responseCalled, wsConnCalled, wsFrameCalled bool + var pluginLogCalled, pluginEnabledCalled, loadPluginCalled bool + + all := &allHandler{ + onRequest: func() { requestCalled = true }, + onResponse: func() { responseCalled = true }, + onWSConn: func() { wsConnCalled = true }, + onWSFrame: func() { wsFrameCalled = true }, + onPluginLog: func() { pluginLogCalled = true }, + onPluginEnabled: func() { pluginEnabledCalled = true }, + onLoadPlugin: func() { loadPluginCalled = true }, + } + + b.Register(all) + + b.EmitRequest(newRequestEvent(t)) + b.EmitResponse(newResponseEvent()) + b.EmitWebSocketConnection(newWSConnectionEvent()) + b.EmitWebSocketFrame(newWSFrameEvent(1)) + b.EmitPluginLog(events.PluginLogEvent{Name: "test.lua", Message: "hello"}) + b.EmitSetPluginEnabled(events.SetPluginEnabledEvent{Name: "test.lua", Enabled: true}) + b.EmitLoadPlugin(events.LoadPluginEvent{SourcePath: "/tmp/test.lua"}) + + assert.True(t, requestCalled) + assert.True(t, responseCalled) + assert.True(t, wsConnCalled) + assert.True(t, wsFrameCalled) + assert.True(t, pluginLogCalled) + assert.True(t, pluginEnabledCalled) + assert.True(t, loadPluginCalled) +} + +// ── EmitRequest ─────────────────────────────────────────────────────────────── + +func TestEmitRequest_NoHandlers_PassThrough(t *testing.T) { + b := events.NewBus() + ev := newRequestEvent(t) + result := b.EmitRequest(ev) + assert.False(t, result.Drop) + assert.Equal(t, ev.Request, result.Request) + assert.Equal(t, ev.Body, result.Body) +} + +func TestEmitRequest_SingleMiddleware_PassThrough(t *testing.T) { + b := events.NewBus() + b.Register(events.RequestMiddlewareFunc(func(e events.RequestEvent) events.RequestResult { + return events.RequestResult{Request: e.Request, Body: e.Body} + })) + ev := newRequestEvent(t) + result := b.EmitRequest(ev) + assert.False(t, result.Drop) + assert.Equal(t, ev.Request, result.Request) +} + +func TestEmitRequest_MiddlewareDrop_ReturnsDrop(t *testing.T) { + b := events.NewBus() + b.Register(events.RequestMiddlewareFunc(func(e events.RequestEvent) events.RequestResult { + return events.RequestResult{Drop: true, Request: e.Request, Body: e.Body} + })) + result := b.EmitRequest(newRequestEvent(t)) + assert.True(t, result.Drop) +} + +func TestEmitRequest_FirstDropShortCircuits(t *testing.T) { + b := events.NewBus() + secondCalled := false + b.Register(events.RequestMiddlewareFunc(func(e events.RequestEvent) events.RequestResult { + return events.RequestResult{Drop: true, Request: e.Request, Body: e.Body} + })) + b.Register(events.RequestMiddlewareFunc(func(e events.RequestEvent) events.RequestResult { + secondCalled = true + return events.RequestResult{Request: e.Request, Body: e.Body} + })) + b.EmitRequest(newRequestEvent(t)) + assert.False(t, secondCalled) +} + +func TestEmitRequest_MiddlewareModifiesBody(t *testing.T) { + b := events.NewBus() + b.Register(events.RequestMiddlewareFunc(func(e events.RequestEvent) events.RequestResult { + return events.RequestResult{Request: e.Request, Body: []byte("modified body")} + })) + result := b.EmitRequest(newRequestEvent(t)) + assert.False(t, result.Drop) + assert.Equal(t, []byte("modified body"), result.Body) +} + +func TestEmitRequest_MiddlewareModifiesRequest(t *testing.T) { + b := events.NewBus() + b.Register(events.RequestMiddlewareFunc(func(e events.RequestEvent) events.RequestResult { + e.Request.Header.Set("X-Injected", "yes") + return events.RequestResult{Request: e.Request, Body: e.Body} + })) + result := b.EmitRequest(newRequestEvent(t)) + assert.Equal(t, "yes", result.Request.Header.Get("X-Injected")) +} + +func TestEmitRequest_ChainedMiddlewares_ModificationsAccumulate(t *testing.T) { + b := events.NewBus() + b.Register(events.RequestMiddlewareFunc(func(e events.RequestEvent) events.RequestResult { + return events.RequestResult{Request: e.Request, Body: []byte("step1")} + })) + b.Register(events.RequestMiddlewareFunc(func(e events.RequestEvent) events.RequestResult { + return events.RequestResult{Request: e.Request, Body: append(e.Body, []byte("-step2")...)} + })) + result := b.EmitRequest(newRequestEvent(t)) + assert.Equal(t, []byte("step1-step2"), result.Body) +} + +func TestEmitRequest_MiddlewareCalledInRegistrationOrder(t *testing.T) { + b := events.NewBus() + var order []int + b.Register(events.RequestMiddlewareFunc(func(e events.RequestEvent) events.RequestResult { + order = append(order, 1) + return events.RequestResult{Request: e.Request, Body: e.Body} + })) + b.Register(events.RequestMiddlewareFunc(func(e events.RequestEvent) events.RequestResult { + order = append(order, 2) + return events.RequestResult{Request: e.Request, Body: e.Body} + })) + b.Register(events.RequestMiddlewareFunc(func(e events.RequestEvent) events.RequestResult { + order = append(order, 3) + return events.RequestResult{Request: e.Request, Body: e.Body} + })) + b.EmitRequest(newRequestEvent(t)) + assert.Equal(t, []int{1, 2, 3}, order) +} + +// ── EmitResponse ────────────────────────────────────────────────────────────── + +func TestEmitResponse_NoHandlers_NoPanic(t *testing.T) { + b := events.NewBus() + assert.NotPanics(t, func() { b.EmitResponse(newResponseEvent()) }) +} + +func TestEmitResponse_ObserverReceivesEvent(t *testing.T) { + b := events.NewBus() + var received events.ResponseEvent + b.Register(events.ResponseObserverFunc(func(e events.ResponseEvent) { received = e })) + ev := newResponseEvent() + b.EmitResponse(ev) + assert.Equal(t, ev.Host, received.Host) + assert.Equal(t, ev.StatusCode, received.StatusCode) + assert.Equal(t, ev.DurationMs, received.DurationMs) +} + +func TestEmitResponse_MultipleObserversAllCalled(t *testing.T) { + b := events.NewBus() + var calls int + b.Register(events.ResponseObserverFunc(func(e events.ResponseEvent) { calls++ })) + b.Register(events.ResponseObserverFunc(func(e events.ResponseEvent) { calls++ })) + b.Register(events.ResponseObserverFunc(func(e events.ResponseEvent) { calls++ })) + b.EmitResponse(newResponseEvent()) + assert.Equal(t, 3, calls) +} + +func TestEmitResponse_ObserversCalledInRegistrationOrder(t *testing.T) { + b := events.NewBus() + var order []int + b.Register(events.ResponseObserverFunc(func(e events.ResponseEvent) { order = append(order, 1) })) + b.Register(events.ResponseObserverFunc(func(e events.ResponseEvent) { order = append(order, 2) })) + b.Register(events.ResponseObserverFunc(func(e events.ResponseEvent) { order = append(order, 3) })) + b.EmitResponse(newResponseEvent()) + assert.Equal(t, []int{1, 2, 3}, order) +} + +func TestEmitResponse_IsSynchronous(t *testing.T) { + b := events.NewBus() + done := false + b.Register(events.ResponseObserverFunc(func(e events.ResponseEvent) { done = true })) + b.EmitResponse(newResponseEvent()) + assert.True(t, done) +} + +// ── EmitWebSocketConnection ─────────────────────────────────────────────────── + +func TestEmitWebSocketConnection_NoHandlers_ReturnsZero(t *testing.T) { + b := events.NewBus() + assert.Equal(t, uint64(0), b.EmitWebSocketConnection(newWSConnectionEvent())) +} + +func TestEmitWebSocketConnection_ObserverReceivesEvent(t *testing.T) { + b := events.NewBus() + var received events.WebSocketConnectionEvent + b.Register(events.WebSocketConnectionObserverFunc(func(e events.WebSocketConnectionEvent) uint64 { + received = e + return 1 + })) + ev := newWSConnectionEvent() + b.EmitWebSocketConnection(ev) + assert.Equal(t, ev.Host, received.Host) + assert.Equal(t, ev.URL, received.URL) + assert.Equal(t, ev.TLS, received.TLS) +} + +func TestEmitWebSocketConnection_ReturnsFirstNonZeroID(t *testing.T) { + b := events.NewBus() + b.Register(events.WebSocketConnectionObserverFunc(func(e events.WebSocketConnectionEvent) uint64 { return 0 })) + b.Register(events.WebSocketConnectionObserverFunc(func(e events.WebSocketConnectionEvent) uint64 { return 42 })) + b.Register(events.WebSocketConnectionObserverFunc(func(e events.WebSocketConnectionEvent) uint64 { return 99 })) + assert.Equal(t, uint64(42), b.EmitWebSocketConnection(newWSConnectionEvent())) +} + +func TestEmitWebSocketConnection_AllObserversCalledEvenAfterNonZero(t *testing.T) { + b := events.NewBus() + var calls int + b.Register(events.WebSocketConnectionObserverFunc(func(e events.WebSocketConnectionEvent) uint64 { calls++; return 1 })) + b.Register(events.WebSocketConnectionObserverFunc(func(e events.WebSocketConnectionEvent) uint64 { calls++; return 2 })) + b.EmitWebSocketConnection(newWSConnectionEvent()) + assert.Equal(t, 2, calls) +} + +func TestEmitWebSocketConnection_ZeroReturnNotUsedAsID(t *testing.T) { + b := events.NewBus() + b.Register(events.WebSocketConnectionObserverFunc(func(e events.WebSocketConnectionEvent) uint64 { return 0 })) + assert.Equal(t, uint64(0), b.EmitWebSocketConnection(newWSConnectionEvent())) +} + +// ── EmitWebSocketFrame ──────────────────────────────────────────────────────── + +func TestEmitWebSocketFrame_NoHandlers_ReturnsOriginalPayload(t *testing.T) { + b := events.NewBus() + ev := newWSFrameEvent(1) + result := b.EmitWebSocketFrame(ev) + assert.Equal(t, ev.Payload, result.Payload) +} + +func TestEmitWebSocketFrame_ObserverReceivesEvent(t *testing.T) { + b := events.NewBus() + var received events.WebSocketFrameEvent + b.Register(events.WebSocketFrameObserverFunc(func(e events.WebSocketFrameEvent) events.WebSocketFrameResult { + received = e + return events.WebSocketFrameResult{Payload: e.Payload} + })) + ev := newWSFrameEvent(7) + b.EmitWebSocketFrame(ev) + assert.Equal(t, uint64(7), received.ConnectionID) + assert.Equal(t, events.WebSocketClient, received.Direction) + assert.Equal(t, events.WebSocketText, received.Opcode) + assert.Equal(t, []byte("hello"), received.Payload) +} + +func TestEmitWebSocketFrame_ObserverModifiesPayload(t *testing.T) { + b := events.NewBus() + b.Register(events.WebSocketFrameObserverFunc(func(e events.WebSocketFrameEvent) events.WebSocketFrameResult { + return events.WebSocketFrameResult{Payload: []byte("modified")} + })) + result := b.EmitWebSocketFrame(newWSFrameEvent(1)) + assert.Equal(t, []byte("modified"), result.Payload) +} + +func TestEmitWebSocketFrame_ChainedObservers_ModificationsAccumulate(t *testing.T) { + b := events.NewBus() + b.Register(events.WebSocketFrameObserverFunc(func(e events.WebSocketFrameEvent) events.WebSocketFrameResult { + return events.WebSocketFrameResult{Payload: []byte("step1")} + })) + b.Register(events.WebSocketFrameObserverFunc(func(e events.WebSocketFrameEvent) events.WebSocketFrameResult { + return events.WebSocketFrameResult{Payload: append(e.Payload, []byte("-step2")...)} + })) + result := b.EmitWebSocketFrame(newWSFrameEvent(1)) + assert.Equal(t, []byte("step1-step2"), result.Payload) +} + +func TestEmitWebSocketFrame_NilPayloadResult_KeepsPrevious(t *testing.T) { + b := events.NewBus() + b.Register(events.WebSocketFrameObserverFunc(func(e events.WebSocketFrameEvent) events.WebSocketFrameResult { + return events.WebSocketFrameResult{Payload: []byte("keep-me")} + })) + b.Register(events.WebSocketFrameObserverFunc(func(e events.WebSocketFrameEvent) events.WebSocketFrameResult { + return events.WebSocketFrameResult{Payload: nil} + })) + result := b.EmitWebSocketFrame(newWSFrameEvent(1)) + assert.Equal(t, []byte("keep-me"), result.Payload) +} + +func TestEmitWebSocketFrame_MultipleObserversAllCalled(t *testing.T) { + b := events.NewBus() + var calls int + b.Register(events.WebSocketFrameObserverFunc(func(e events.WebSocketFrameEvent) events.WebSocketFrameResult { + calls++ + return events.WebSocketFrameResult{Payload: e.Payload} + })) + b.Register(events.WebSocketFrameObserverFunc(func(e events.WebSocketFrameEvent) events.WebSocketFrameResult { + calls++ + return events.WebSocketFrameResult{Payload: e.Payload} + })) + b.EmitWebSocketFrame(newWSFrameEvent(1)) + assert.Equal(t, 2, calls) +} + +func TestEmitWebSocketFrame_OrderingPreserved(t *testing.T) { + b := events.NewBus() + var received []uint64 + b.Register(events.WebSocketFrameObserverFunc(func(e events.WebSocketFrameEvent) events.WebSocketFrameResult { + received = append(received, e.ConnectionID) + return events.WebSocketFrameResult{Payload: e.Payload} + })) + b.EmitWebSocketFrame(events.WebSocketFrameEvent{ConnectionID: 1, Payload: []byte("a")}) + b.EmitWebSocketFrame(events.WebSocketFrameEvent{ConnectionID: 2, Payload: []byte("b")}) + b.EmitWebSocketFrame(events.WebSocketFrameEvent{ConnectionID: 3, Payload: []byte("c")}) + assert.Equal(t, []uint64{1, 2, 3}, received) +} + +// ── EmitPluginLog ───────────────────────────────────────────────────────────── + +func TestEmitPluginLog_NoHandlers_NoPanic(t *testing.T) { + b := events.NewBus() + assert.NotPanics(t, func() { + b.EmitPluginLog(events.PluginLogEvent{Name: "test.lua", Message: "hello"}) + }) +} + +func TestEmitPluginLog_ObserverReceivesEvent(t *testing.T) { + b := events.NewBus() + var received events.PluginLogEvent + b.Register(events.PluginLogObserverFunc(func(e events.PluginLogEvent) { received = e })) + b.EmitPluginLog(events.PluginLogEvent{Name: "myplugin.lua", Message: "log line"}) + assert.Equal(t, "myplugin.lua", received.Name) + assert.Equal(t, "log line", received.Message) +} + +func TestEmitPluginLog_MultipleObserversAllCalled(t *testing.T) { + b := events.NewBus() + var calls int + b.Register(events.PluginLogObserverFunc(func(e events.PluginLogEvent) { calls++ })) + b.Register(events.PluginLogObserverFunc(func(e events.PluginLogEvent) { calls++ })) + b.EmitPluginLog(events.PluginLogEvent{Name: "test.lua", Message: "msg"}) + assert.Equal(t, 2, calls) +} + +func TestEmitPluginLog_IsSynchronous(t *testing.T) { + b := events.NewBus() + done := false + b.Register(events.PluginLogObserverFunc(func(e events.PluginLogEvent) { done = true })) + b.EmitPluginLog(events.PluginLogEvent{Name: "test.lua", Message: "msg"}) + assert.True(t, done) +} + +// ── EmitSetPluginEnabled ────────────────────────────────────────────────────── + +func TestEmitSetPluginEnabled_NoHandlers_NoPanic(t *testing.T) { + b := events.NewBus() + assert.NotPanics(t, func() { + b.EmitSetPluginEnabled(events.SetPluginEnabledEvent{Name: "test.lua", Enabled: false}) + }) +} + +func TestEmitSetPluginEnabled_ObserverReceivesEvent(t *testing.T) { + b := events.NewBus() + var received events.SetPluginEnabledEvent + b.Register(events.PluginEnabledObserverFunc(func(e events.SetPluginEnabledEvent) { received = e })) + b.EmitSetPluginEnabled(events.SetPluginEnabledEvent{Name: "myplugin.lua", Enabled: false}) + assert.Equal(t, "myplugin.lua", received.Name) + assert.False(t, received.Enabled) +} + +func TestEmitSetPluginEnabled_MultipleObserversAllCalled(t *testing.T) { + b := events.NewBus() + var calls int + b.Register(events.PluginEnabledObserverFunc(func(e events.SetPluginEnabledEvent) { calls++ })) + b.Register(events.PluginEnabledObserverFunc(func(e events.SetPluginEnabledEvent) { calls++ })) + b.EmitSetPluginEnabled(events.SetPluginEnabledEvent{Name: "test.lua", Enabled: true}) + assert.Equal(t, 2, calls) +} + +func TestEmitSetPluginEnabled_EnabledFlagPropagated(t *testing.T) { + b := events.NewBus() + var gotEnabled bool + b.Register(events.PluginEnabledObserverFunc(func(e events.SetPluginEnabledEvent) { gotEnabled = e.Enabled })) + b.EmitSetPluginEnabled(events.SetPluginEnabledEvent{Name: "test.lua", Enabled: true}) + assert.True(t, gotEnabled) + b.EmitSetPluginEnabled(events.SetPluginEnabledEvent{Name: "test.lua", Enabled: false}) + assert.False(t, gotEnabled) +} + +// ── EmitLoadPlugin ──────────────────────────────────────────────────────────── + +func TestEmitLoadPlugin_NoHandlers_NoPanic(t *testing.T) { + b := events.NewBus() + assert.NotPanics(t, func() { + b.EmitLoadPlugin(events.LoadPluginEvent{SourcePath: "/tmp/test.lua"}) + }) +} + +func TestEmitLoadPlugin_ObserverReceivesEvent(t *testing.T) { + b := events.NewBus() + var received events.LoadPluginEvent + b.Register(events.LoadPluginObserverFunc(func(e events.LoadPluginEvent) { received = e })) + b.EmitLoadPlugin(events.LoadPluginEvent{SourcePath: "/home/user/myplugin.lua"}) + assert.Equal(t, "/home/user/myplugin.lua", received.SourcePath) +} + +func TestEmitLoadPlugin_MultipleObserversAllCalled(t *testing.T) { + b := events.NewBus() + var calls int + b.Register(events.LoadPluginObserverFunc(func(e events.LoadPluginEvent) { calls++ })) + b.Register(events.LoadPluginObserverFunc(func(e events.LoadPluginEvent) { calls++ })) + b.EmitLoadPlugin(events.LoadPluginEvent{SourcePath: "/tmp/test.lua"}) + assert.Equal(t, 2, calls) +} + +func TestEmitLoadPlugin_IsSynchronous(t *testing.T) { + b := events.NewBus() + done := false + b.Register(events.LoadPluginObserverFunc(func(e events.LoadPluginEvent) { done = true })) + b.EmitLoadPlugin(events.LoadPluginEvent{SourcePath: "/tmp/test.lua"}) + assert.True(t, done) +} + +// ── Func adapters ───────────────────────────────────────────────────────────── + +func TestRequestMiddlewareFunc_ImplementsInterface(t *testing.T) { + var _ events.RequestMiddleware = events.RequestMiddlewareFunc(nil) +} + +func TestResponseObserverFunc_ImplementsInterface(t *testing.T) { + var _ events.ResponseObserver = events.ResponseObserverFunc(nil) +} + +func TestWebSocketConnectionObserverFunc_ImplementsInterface(t *testing.T) { + var _ events.WebSocketConnectionObserver = events.WebSocketConnectionObserverFunc(nil) +} + +func TestWebSocketFrameObserverFunc_ImplementsInterface(t *testing.T) { + var _ events.WebSocketFrameObserver = events.WebSocketFrameObserverFunc(nil) +} + +func TestPluginLogObserverFunc_ImplementsInterface(t *testing.T) { + var _ events.PluginLogObserver = events.PluginLogObserverFunc(nil) +} + +func TestPluginEnabledObserverFunc_ImplementsInterface(t *testing.T) { + var _ events.PluginEnabledObserver = events.PluginEnabledObserverFunc(nil) +} + +func TestLoadPluginObserverFunc_ImplementsInterface(t *testing.T) { + var _ events.LoadPluginObserver = events.LoadPluginObserverFunc(nil) +} + +func TestRequestMiddlewareFunc_DelegatesCall(t *testing.T) { + called := false + f := events.RequestMiddlewareFunc(func(e events.RequestEvent) events.RequestResult { + called = true + return events.RequestResult{Request: e.Request, Body: e.Body} + }) + f.HandleRequest(events.RequestEvent{Request: newRequest(t, http.MethodGet, "http://example.com")}) + assert.True(t, called) +} + +func TestResponseObserverFunc_DelegatesCall(t *testing.T) { + called := false + f := events.ResponseObserverFunc(func(e events.ResponseEvent) { called = true }) + f.ObserveResponse(newResponseEvent()) + assert.True(t, called) +} + +func TestWebSocketConnectionObserverFunc_DelegatesCall(t *testing.T) { + f := events.WebSocketConnectionObserverFunc(func(e events.WebSocketConnectionEvent) uint64 { return 55 }) + assert.Equal(t, uint64(55), f.ObserveWebSocketConnection(newWSConnectionEvent())) +} + +func TestWebSocketFrameObserverFunc_DelegatesCall(t *testing.T) { + called := false + f := events.WebSocketFrameObserverFunc(func(e events.WebSocketFrameEvent) events.WebSocketFrameResult { + called = true + return events.WebSocketFrameResult{Payload: e.Payload} + }) + f.ObserveWebSocketFrame(newWSFrameEvent(1)) + assert.True(t, called) +} + +func TestPluginLogObserverFunc_DelegatesCall(t *testing.T) { + called := false + f := events.PluginLogObserverFunc(func(e events.PluginLogEvent) { called = true }) + f.ObservePluginLog(events.PluginLogEvent{Name: "test.lua", Message: "hello"}) + assert.True(t, called) +} + +func TestPluginEnabledObserverFunc_DelegatesCall(t *testing.T) { + var got events.SetPluginEnabledEvent + f := events.PluginEnabledObserverFunc(func(e events.SetPluginEnabledEvent) { got = e }) + f.ObservePluginEnabled(events.SetPluginEnabledEvent{Name: "test.lua", Enabled: true}) + assert.Equal(t, "test.lua", got.Name) + assert.True(t, got.Enabled) +} + +func TestLoadPluginObserverFunc_DelegatesCall(t *testing.T) { + var got events.LoadPluginEvent + f := events.LoadPluginObserverFunc(func(e events.LoadPluginEvent) { got = e }) + f.ObserveLoadPlugin(events.LoadPluginEvent{SourcePath: "/tmp/test.lua"}) + assert.Equal(t, "/tmp/test.lua", got.SourcePath) +} + +// ── Concurrency ─────────────────────────────────────────────────────────────── + +func TestBus_ConcurrentEmitRequest_Safe(t *testing.T) { + b := events.NewBus() + var count atomic.Int64 + b.Register(events.RequestMiddlewareFunc(func(e events.RequestEvent) events.RequestResult { + count.Add(1) + return events.RequestResult{Request: e.Request, Body: e.Body} + })) + const goroutines = 50 + var wg sync.WaitGroup + wg.Add(goroutines) + for range goroutines { + go func() { + defer wg.Done() + b.EmitRequest(events.RequestEvent{Request: httptest.NewRequest(http.MethodGet, "http://example.com", nil)}) + }() + } + wg.Wait() + assert.Equal(t, int64(goroutines), count.Load()) +} + +func TestBus_ConcurrentEmitResponse_Safe(t *testing.T) { + b := events.NewBus() + var count atomic.Int64 + b.Register(events.ResponseObserverFunc(func(e events.ResponseEvent) { count.Add(1) })) + const goroutines = 50 + var wg sync.WaitGroup + wg.Add(goroutines) + for range goroutines { + go func() { + defer wg.Done() + b.EmitResponse(newResponseEvent()) + }() + } + wg.Wait() + assert.Equal(t, int64(goroutines), count.Load()) +} + +func TestBus_ConcurrentEmitPluginLog_Safe(t *testing.T) { + b := events.NewBus() + var count atomic.Int64 + b.Register(events.PluginLogObserverFunc(func(e events.PluginLogEvent) { count.Add(1) })) + const goroutines = 50 + var wg sync.WaitGroup + wg.Add(goroutines) + for range goroutines { + go func() { + defer wg.Done() + b.EmitPluginLog(events.PluginLogEvent{Name: "test.lua", Message: "msg"}) + }() + } + wg.Wait() + assert.Equal(t, int64(goroutines), count.Load()) +} + +func TestBus_ConcurrentRegisterAndEmit_Safe(t *testing.T) { + b := events.NewBus() + var wg sync.WaitGroup + for range 20 { + wg.Add(1) + go func() { + defer wg.Done() + b.EmitRequest(events.RequestEvent{Request: httptest.NewRequest(http.MethodGet, "http://example.com", nil)}) + }() + } + for range 20 { + wg.Add(1) + go func() { + defer wg.Done() + b.Register(events.RequestMiddlewareFunc(func(e events.RequestEvent) events.RequestResult { + return events.RequestResult{Request: e.Request, Body: e.Body} + })) + }() + } + wg.Wait() +} + +// ── WebSocket direction and opcode constants ────────────────────────────────── + +func TestWebSocketDirection_Values(t *testing.T) { + assert.Equal(t, events.WebSocketDirection(0), events.WebSocketClient) + assert.Equal(t, events.WebSocketDirection(1), events.WebSocketServer) +} + +func TestWebSocketOpcode_Values(t *testing.T) { + assert.Equal(t, events.WebSocketOpcode(1), events.WebSocketText) + assert.Equal(t, events.WebSocketOpcode(2), events.WebSocketBinary) + assert.Equal(t, events.WebSocketOpcode(8), events.WebSocketClose) + assert.Equal(t, events.WebSocketOpcode(9), events.WebSocketPing) + assert.Equal(t, events.WebSocketOpcode(10), events.WebSocketPong) +} + +// ── allHandler — test double implementing all seven interfaces ──────────────── + +type allHandler struct { + onRequest func() + onResponse func() + onWSConn func() + onWSFrame func() + onPluginLog func() + onPluginEnabled func() + onLoadPlugin func() +} + +func (h *allHandler) HandleRequest(e events.RequestEvent) events.RequestResult { + h.onRequest() + return events.RequestResult{Request: e.Request, Body: e.Body} +} + +func (h *allHandler) ObserveResponse(e events.ResponseEvent) { h.onResponse() } + +func (h *allHandler) ObserveWebSocketConnection(e events.WebSocketConnectionEvent) uint64 { + h.onWSConn() + return 1 +} + +func (h *allHandler) ObserveWebSocketFrame(e events.WebSocketFrameEvent) events.WebSocketFrameResult { + h.onWSFrame() + return events.WebSocketFrameResult{Payload: e.Payload} +} + +func (h *allHandler) ObservePluginLog(e events.PluginLogEvent) { h.onPluginLog() } +func (h *allHandler) ObservePluginEnabled(e events.SetPluginEnabledEvent) { h.onPluginEnabled() } +func (h *allHandler) ObserveLoadPlugin(e events.LoadPluginEvent) { h.onLoadPlugin() } |