Merge pull request #489 from Projectplace/fix-cancel-race

Fix race in ExecContext

1 files changed, 21 insertions, 0 deletions

diff --git a/sqlite3_go18_test.go b/sqlite3_go18_test.go
index 2662fcf..44fc4df 100644
--- a/sqlite3_go18_test.go
+++ b/sqlite3_go18_test.go
@@ -134,3 +134,24 @@ func TestShortTimeout(t *testing.T) {
 		t.Fatal(ctx.Err())
 	}
 }
+
+func TestExecCancel(t *testing.T) {
+	db, err := sql.Open("sqlite3", ":memory:")
+	if err != nil {
+		t.Fatal(err)
+	}
+	defer db.Close()
+
+	if _, err = db.Exec("create table foo (id integer primary key)"); err != nil {
+		t.Fatal(err)
+	}
+
+	for n := 0; n < 100; n++ {
+		ctx, cancel := context.WithCancel(context.Background())
+		_, err = db.ExecContext(ctx, "insert into foo (id) values (?)", n)
+		cancel()
+		if err != nil {
+			t.Fatal(err)
+		}
+	}
+}