1
0
Fork 0

update test to handle ro timeout

master
Mauri de Souza Meneguzzo 8 months ago
parent 1c3268be78
commit 5ea8233424

@ -2,7 +2,6 @@ package db
import ( import (
"errors" "errors"
"fmt"
"os" "os"
"strings" "strings"
"testing" "testing"
@ -13,30 +12,20 @@ import (
) )
func testBusyTimeout(t *testing.T, db *DB) { func testBusyTimeout(t *testing.T, db *DB) {
rbt := random.Intn(10000) wantTimeout := random.Intn(10000)
_, err := db.ExecuteStringStmt(fmt.Sprintf("PRAGMA busy_timeout=%d", rbt))
err := db.SetBusyTimeout(wantTimeout, wantTimeout)
if err != nil { if err != nil {
t.Fatalf("failed to set busy_timeout: %s", err.Error()) t.Fatalf("failed to set busy_timeout: %s", err.Error())
} }
rw, _, err := db.BusyTimeout() gotrw, gotro, err := db.BusyTimeout()
if err != nil { if err != nil {
t.Fatalf("failed to get busy_timeout: %s", err.Error()) t.Fatalf("failed to get busy_timeout: %s", err.Error())
} }
if exp, got := rbt, rw; exp != got {
t.Fatalf("expected busy_timeout %d, got %d", exp, got)
}
rw2 := random.Intn(10000) if gotrw != wantTimeout || gotro != wantTimeout {
if err := db.SetBusyTimeout(rw2, 0); err != nil { t.Fatalf("want busy_timeout=%d, got rw=%d, ro=%d", wantTimeout, gotrw, gotro)
t.Fatalf("failed to set busy_timeout: %s", err.Error())
}
rw, _, err = db.BusyTimeout()
if err != nil {
t.Fatalf("failed to get busy_timeout: %s", err.Error())
}
if exp, got := rw2, rw; exp != got {
t.Fatalf("expected busy_timeout %d, got %d", exp, got)
} }
} }

Loading…
Cancel
Save