mirror of
https://gitee.com/wanwujie/sub2api
synced 2026-04-18 05:44:46 +08:00
23 lines
662 B
Go
23 lines
662 B
Go
|
package repository
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"fmt"
|
||
|
|
"testing"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestIsQueryTimeoutErr(t *testing.T) {
|
||
|
|
if !isQueryTimeoutErr(context.DeadlineExceeded) {
|
||
|
|
t.Fatalf("context.DeadlineExceeded should be treated as query timeout")
|
||
|
|
}
|
||
|
|
if !isQueryTimeoutErr(fmt.Errorf("wrapped: %w", context.DeadlineExceeded)) {
|
||
|
|
t.Fatalf("wrapped context.DeadlineExceeded should be treated as query timeout")
|
||
|
|
}
|
||
|
|
if isQueryTimeoutErr(context.Canceled) {
|
||
|
|
t.Fatalf("context.Canceled should not be treated as query timeout")
|
||
|
|
}
|
||
|
|
if isQueryTimeoutErr(fmt.Errorf("wrapped: %w", context.Canceled)) {
|
||
|
|
t.Fatalf("wrapped context.Canceled should not be treated as query timeout")
|
||
|
|
}
|
||
|
|
}
|