83 lines
1.9 KiB
Go
83 lines
1.9 KiB
Go
|
//
|
||
|
|
// context_test.go
|
||
|
|
// Copyright (C) 2023 tiglog <me@tiglog.com>
|
||
|
|
//
|
||
|
|
// Distributed under terms of the MIT license.
|
||
|
|
//
|
||
|
|
|
||
|
|
//go:build integration
|
||
|
|
// +build integration
|
||
|
|
|
||
|
|
package sqldb_test
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"testing"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"github.com/stretchr/testify/assert"
|
||
|
|
)
|
||
|
|
|
||
|
|
// Drivers that don't support cancellation.
|
||
|
|
var unsupportedDrivers map[string]bool = map[string]bool{
|
||
|
|
"mymysql": true,
|
||
|
|
}
|
||
|
|
|
||
|
|
type SleepDialect interface {
|
||
|
|
// string to sleep for d duration
|
||
|
|
SleepClause(d time.Duration) string
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestWithNotCanceledContext(t *testing.T) {
|
||
|
|
dbmap := initDBMap(t)
|
||
|
|
defer dropAndClose(dbmap)
|
||
|
|
|
||
|
|
ctx, cancel := context.WithTimeout(context.Background(), 100*time.Millisecond)
|
||
|
|
defer cancel()
|
||
|
|
|
||
|
|
withCtx := dbmap.WithContext(ctx)
|
||
|
|
|
||
|
|
_, err := withCtx.Exec("SELECT 1")
|
||
|
|
assert.Nil(t, err)
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestWithCanceledContext(t *testing.T) {
|
||
|
|
dialect, driver := dialectAndDriver()
|
||
|
|
if unsupportedDrivers[driver] {
|
||
|
|
t.Skipf("Cancellation is not yet supported by all drivers. Not known to be supported in %s.", driver)
|
||
|
|
}
|
||
|
|
|
||
|
|
sleepDialect, ok := dialect.(SleepDialect)
|
||
|
|
if !ok {
|
||
|
|
t.Skipf("Sleep is not supported in all dialects. Not known to be supported in %s.", driver)
|
||
|
|
}
|
||
|
|
|
||
|
|
dbmap := initDBMap(t)
|
||
|
|
defer dropAndClose(dbmap)
|
||
|
|
|
||
|
|
ctx, cancel := context.WithTimeout(context.Background(), 100*time.Millisecond)
|
||
|
|
defer cancel()
|
||
|
|
|
||
|
|
withCtx := dbmap.WithContext(ctx)
|
||
|
|
|
||
|
|
startTime := time.Now()
|
||
|
|
|
||
|
|
_, err := withCtx.Exec("SELECT " + sleepDialect.SleepClause(1*time.Second))
|
||
|
|
|
||
|
|
if d := time.Since(startTime); d > 500*time.Millisecond {
|
||
|
|
t.Errorf("too long execution time: %s", d)
|
||
|
|
}
|
||
|
|
|
||
|
|
switch driver {
|
||
|
|
case "postgres":
|
||
|
|
// pq doesn't return standard deadline exceeded error
|
||
|
|
if err.Error() != "pq: canceling statement due to user request" {
|
||
|
|
t.Errorf("expected context.DeadlineExceeded, got %v", err)
|
||
|
|
}
|
||
|
|
default:
|
||
|
|
if err != context.DeadlineExceeded {
|
||
|
|
t.Errorf("expected context.DeadlineExceeded, got %v", err)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|