Allow either error message

batch-wip
Jack Christensen 2017-06-02 08:38:27 -05:00
parent dcf3ee2781
commit dfe250c13b
1 changed files with 2 additions and 2 deletions

View File

@ -714,8 +714,8 @@ func TestBeginTxContextCancel(t *testing.T) {
cancelFn()
err = tx.Commit()
if err != context.Canceled {
t.Fatalf("err => %v, want %v", err, context.Canceled)
if err != context.Canceled && err != sql.ErrTxDone {
t.Fatalf("err => %v, want %v or %v", err, context.Canceled, sql.ErrTxDone)
}
var n int