Jay
b44a46ed2f
Replaces the flat key-value logging scheme with component-based structured logging via go-mana-component. Each layer (pool, worker, connection) builds its own component identity and derives a *slog.Logger from a caller-supplied slog.Handler. - Delete logging/ package (logging.go, logging_test.go) - Strip LoggingEnabled and LogLevel from ConnectionConfig, PoolConfig, WorkerConfig; remove associated option funcs - Change NewConnection and NewConnectionFromSocket to accept ctx and slog.Handler instead of *slog.Logger; constructors build component identity via MustNew/MustExtend internally - Change WorkerFactory, NewWorker, connect, and RunDialer to carry slog.Handler; remove PoolPlugin.Handler - Change NewPool to establish pool component identity via MustNew; remove pool_id field, PoolPlugin.ID, and ErrInvalidPoolID - Fix data race in MockSlogHandler: WithAttrs now shares parent mutex pointer rather than allocating a new one per child - Run go fix
127 lines
2.7 KiB
Go
127 lines
2.7 KiB
Go
package honeybee
|
|
|
|
import (
|
|
"git.wisehodl.dev/jay/go-honeybee/transport"
|
|
"github.com/stretchr/testify/assert"
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
func TestNewPoolConfig(t *testing.T) {
|
|
conf, err := NewPoolConfig()
|
|
assert.NoError(t, err)
|
|
|
|
assert.Equal(t, conf, &PoolConfig{
|
|
InboxBufferSize: 256,
|
|
EventsBufferSize: 10,
|
|
ConnectionConfig: nil,
|
|
WorkerConfig: nil,
|
|
WorkerFactory: nil,
|
|
})
|
|
}
|
|
|
|
func TestDefaultPoolConfig(t *testing.T) {
|
|
conf := GetDefaultPoolConfig()
|
|
|
|
assert.Equal(t, conf, &PoolConfig{
|
|
InboxBufferSize: 256,
|
|
EventsBufferSize: 10,
|
|
ConnectionConfig: nil,
|
|
WorkerConfig: nil,
|
|
WorkerFactory: nil,
|
|
})
|
|
}
|
|
|
|
func TestApplyPoolOptions(t *testing.T) {
|
|
conf := &PoolConfig{}
|
|
err := applyPoolOptions(
|
|
conf,
|
|
WithConnectionConfig(&transport.ConnectionConfig{}),
|
|
)
|
|
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, 0*time.Second, conf.ConnectionConfig.WriteTimeout)
|
|
}
|
|
|
|
func TestWithBufferSizes(t *testing.T) {
|
|
conf := &PoolConfig{}
|
|
|
|
err := applyPoolOptions(conf,
|
|
WithInboxBufferSize(100),
|
|
WithEventsBufferSize(20),
|
|
)
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, 100, conf.InboxBufferSize)
|
|
assert.Equal(t, 20, conf.EventsBufferSize)
|
|
}
|
|
|
|
func TestWithConnectionConfig(t *testing.T) {
|
|
conf := &PoolConfig{}
|
|
opt := WithConnectionConfig(&transport.ConnectionConfig{WriteTimeout: 1 * time.Second})
|
|
err := applyPoolOptions(conf, opt)
|
|
assert.NoError(t, err)
|
|
assert.NotNil(t, conf.ConnectionConfig)
|
|
assert.Equal(t, 1*time.Second, conf.ConnectionConfig.WriteTimeout)
|
|
|
|
// invalid config is rejected
|
|
conf = &PoolConfig{}
|
|
opt = WithConnectionConfig(&transport.ConnectionConfig{WriteTimeout: -1 * time.Second})
|
|
err = applyPoolOptions(conf, opt)
|
|
assert.Error(t, err)
|
|
}
|
|
|
|
func TestValidatePoolConfig(t *testing.T) {
|
|
cases := []struct {
|
|
name string
|
|
conf PoolConfig
|
|
wantErr error
|
|
wantErrText string
|
|
}{
|
|
{
|
|
name: "valid empty",
|
|
conf: *&PoolConfig{},
|
|
},
|
|
{
|
|
name: "valid defaults",
|
|
conf: *GetDefaultPoolConfig(),
|
|
},
|
|
{
|
|
name: "valid complete",
|
|
conf: PoolConfig{
|
|
ConnectionConfig: &transport.ConnectionConfig{},
|
|
},
|
|
},
|
|
{
|
|
name: "invalid connection config",
|
|
conf: PoolConfig{
|
|
ConnectionConfig: &transport.ConnectionConfig{
|
|
Retry: &transport.RetryConfig{
|
|
InitialDelay: 10 * time.Second,
|
|
MaxDelay: 1 * time.Second,
|
|
},
|
|
},
|
|
},
|
|
wantErrText: "initial delay may not exceed maximum delay",
|
|
},
|
|
}
|
|
|
|
for _, tc := range cases {
|
|
t.Run(tc.name, func(t *testing.T) {
|
|
err := ValidatePoolConfig(&tc.conf)
|
|
|
|
if tc.wantErr != nil || tc.wantErrText != "" {
|
|
if tc.wantErr != nil {
|
|
assert.ErrorIs(t, err, tc.wantErr)
|
|
}
|
|
|
|
if tc.wantErrText != "" {
|
|
assert.ErrorContains(t, err, tc.wantErrText)
|
|
}
|
|
return
|
|
}
|
|
|
|
assert.NoError(t, err)
|
|
})
|
|
}
|
|
}
|