124 lines
2.7 KiB
Go
124 lines
2.7 KiB
Go
package honeybeetest
|
|
|
|
import (
|
|
"context"
|
|
"git.wisehodl.dev/jay/go-honeybee/types"
|
|
"log/slog"
|
|
"net/http"
|
|
"sync"
|
|
"time"
|
|
)
|
|
|
|
// Dialer Mocks
|
|
|
|
type MockDialer struct {
|
|
DialContextFunc func(
|
|
context.Context, string, http.Header,
|
|
) (types.Socket, *http.Response, error)
|
|
}
|
|
|
|
func (m *MockDialer) DialContext(
|
|
ctx context.Context, url string, h http.Header,
|
|
) (types.Socket, *http.Response, error) {
|
|
return m.DialContextFunc(ctx, url, h)
|
|
}
|
|
|
|
// Socket Mocks
|
|
|
|
type MockSocket struct {
|
|
WriteMessageFunc func(int, []byte) error
|
|
SetReadDeadlineFunc func(t time.Time) error
|
|
SetWriteDeadlineFunc func(t time.Time) error
|
|
ReadMessageFunc func() (int, []byte, error)
|
|
CloseFunc func() error
|
|
SetCloseHandlerFunc func(func(int, string) error)
|
|
Closed chan struct{}
|
|
Once sync.Once
|
|
Mu sync.Mutex
|
|
}
|
|
|
|
func NewMockSocket() *MockSocket {
|
|
return &MockSocket{
|
|
WriteMessageFunc: func(int, []byte) error { return nil },
|
|
ReadMessageFunc: func() (int, []byte, error) { return 0, []byte("message"), nil },
|
|
CloseFunc: func() error { return nil },
|
|
|
|
SetReadDeadlineFunc: func(time.Time) error { return nil },
|
|
SetWriteDeadlineFunc: func(time.Time) error { return nil },
|
|
SetCloseHandlerFunc: func(func(int, string) error) {},
|
|
|
|
Closed: make(chan struct{}),
|
|
}
|
|
|
|
}
|
|
|
|
func (m *MockSocket) WriteMessage(t int, d []byte) error {
|
|
return m.WriteMessageFunc(t, d)
|
|
}
|
|
|
|
func (m *MockSocket) ReadMessage() (int, []byte, error) {
|
|
return m.ReadMessageFunc()
|
|
}
|
|
|
|
func (m *MockSocket) Close() error {
|
|
return m.CloseFunc()
|
|
}
|
|
|
|
func (m *MockSocket) SetReadDeadline(t time.Time) error {
|
|
return m.SetReadDeadlineFunc(t)
|
|
}
|
|
|
|
func (m *MockSocket) SetWriteDeadline(t time.Time) error {
|
|
return m.SetWriteDeadlineFunc(t)
|
|
}
|
|
|
|
func (m *MockSocket) SetCloseHandler(h func(code int, text string) error) {
|
|
m.SetCloseHandlerFunc(h)
|
|
}
|
|
|
|
// Logging mocks
|
|
|
|
type MockSlogHandler struct {
|
|
records []slog.Record
|
|
mu sync.RWMutex
|
|
}
|
|
|
|
func NewMockSlogHandler() *MockSlogHandler {
|
|
return &MockSlogHandler{
|
|
records: make([]slog.Record, 0),
|
|
}
|
|
}
|
|
|
|
func (m *MockSlogHandler) Handle(ctx context.Context, record slog.Record) error {
|
|
m.mu.Lock()
|
|
defer m.mu.Unlock()
|
|
m.records = append(m.records, record)
|
|
return nil
|
|
}
|
|
|
|
func (m *MockSlogHandler) Enabled(ctx context.Context, level slog.Level) bool {
|
|
return true
|
|
}
|
|
|
|
func (m *MockSlogHandler) WithAttrs(attrs []slog.Attr) slog.Handler {
|
|
return m
|
|
}
|
|
|
|
func (m *MockSlogHandler) WithGroup(name string) slog.Handler {
|
|
return m
|
|
}
|
|
|
|
func (m *MockSlogHandler) GetRecords() []slog.Record {
|
|
m.mu.RLock()
|
|
defer m.mu.RUnlock()
|
|
result := make([]slog.Record, len(m.records))
|
|
copy(result, m.records)
|
|
return result
|
|
}
|
|
|
|
func (m *MockSlogHandler) Clear() {
|
|
m.mu.Lock()
|
|
defer m.mu.Unlock()
|
|
m.records = make([]slog.Record, 0)
|
|
}
|