51 lines
1.4 KiB
Go
51 lines
1.4 KiB
Go
package service
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
func TestIsRecoverableIPCError(t *testing.T) {
|
|
cases := []error{
|
|
errors.New("write websocket frame: write tcp 127.0.0.1:64954->127.0.0.1:36510: use of closed network connection"),
|
|
errors.New("broken pipe"),
|
|
errors.New("Lingma IPC notification stream closed"),
|
|
}
|
|
for _, err := range cases {
|
|
if !isRecoverableIPCError(err) {
|
|
t.Fatalf("expected recoverable error: %v", err)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestIsRecoverableIPCErrorIgnoresModelErrors(t *testing.T) {
|
|
if isRecoverableIPCError(errors.New("timed out while waiting for Lingma IPC to finish responding")) {
|
|
t.Fatal("timeout should not be treated as an immediate reconnect retry")
|
|
}
|
|
}
|
|
|
|
func TestNewKeepsZeroTimeoutUnlimited(t *testing.T) {
|
|
svc := New(Config{Timeout: 0})
|
|
if svc.cfg.Timeout != 0 {
|
|
t.Fatalf("timeout = %v, want 0", svc.cfg.Timeout)
|
|
}
|
|
}
|
|
|
|
func TestContextWithOptionalTimeoutZeroDoesNotSetDeadline(t *testing.T) {
|
|
ctx, cancel := contextWithOptionalTimeout(context.Background(), 0)
|
|
defer cancel()
|
|
if _, ok := ctx.Deadline(); ok {
|
|
t.Fatal("zero timeout should not set a deadline")
|
|
}
|
|
}
|
|
|
|
func TestContextWithOptionalTimeoutPositiveSetsDeadline(t *testing.T) {
|
|
ctx, cancel := contextWithOptionalTimeout(context.Background(), time.Second)
|
|
defer cancel()
|
|
if _, ok := ctx.Deadline(); !ok {
|
|
t.Fatal("positive timeout should set a deadline")
|
|
}
|
|
}
|