Allow either error message

2017-06-02 08:38:27 -05:00
parent dcf3ee2781
commit dfe250c13b
1 changed files with 2 additions and 2 deletions
@@ -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