executor: Add tests for tidb_constraint_check_in_place for local te…

…mporary table (pingcap#27271)

executor/write_test.go

@@ -2863,6 +2863,7 @@ func (s *testSuite7) TestDeferConstraintCheckForInsert(c *C) {
 
 	// Cover the temporary table.
 	tk.MustExec("set tidb_enable_global_temporary_table=true")
+	tk.MustExec("set tidb_enable_noop_functions=true")
 	for val := range []int{0, 1} {
 		tk.MustExec("set tidb_constraint_check_in_place = ?", val)
 
@@ -2891,6 +2892,53 @@ func (s *testSuite7) TestDeferConstraintCheckForInsert(c *C) {
 		_, err = tk.Exec("insert into t values (3, 1) on duplicated key update b = b + 1")
 		c.Assert(err, NotNil)
 		tk.MustExec("commit")
+
+		// cases for temporary table
+		tk.MustExec("drop table if exists tl")
+		tk.MustExec("create temporary table tl (a int primary key, b int)")
+		tk.MustExec("begin")
+		tk.MustExec("insert into tl values (1, 1)")
+		_, err = tk.Exec(`insert into tl values (1, 3)`)
+		c.Assert(err, NotNil)
+		tk.MustExec("insert into tl values (2, 2)")
+		_, err = tk.Exec("update tl set a = a + 1 where a = 1")
+		c.Assert(err, NotNil)
+		_, err = tk.Exec("insert into tl values (1, 3) on duplicated key update a = a + 1")
+		c.Assert(err, NotNil)
+		tk.MustExec("commit")
+
+		tk.MustExec("begin")
+		tk.MustQuery("select * from tl").Check(testkit.Rows("1 1", "2 2"))
+		_, err = tk.Exec(`insert into tl values (1, 3)`)
+		c.Assert(err, NotNil)
+		_, err = tk.Exec("update tl set a = a + 1 where a = 1")
+		c.Assert(err, NotNil)
+		_, err = tk.Exec("insert into tl values (1, 3) on duplicated key update a = a + 1")
+		c.Assert(err, NotNil)
+		tk.MustExec("rollback")
+
+		tk.MustExec("drop table tl")
+		tk.MustExec("create temporary table tl (a int, b int unique)")
+		tk.MustExec("begin")
+		tk.MustExec("insert into tl values (1, 1)")
+		_, err = tk.Exec(`insert into tl values (3, 1)`)
+		c.Assert(err, NotNil)
+		tk.MustExec("insert into tl values (2, 2)")
+		_, err = tk.Exec("update tl set b = b + 1 where a = 1")
+		c.Assert(err, NotNil)
+		_, err = tk.Exec("insert into tl values (3, 1) on duplicated key update b = b + 1")
+		c.Assert(err, NotNil)
+		tk.MustExec("commit")
+
+		tk.MustExec("begin")
+		tk.MustQuery("select * from tl").Check(testkit.Rows("1 1", "2 2"))
+		_, err = tk.Exec(`insert into tl values (3, 1)`)
+		c.Assert(err, NotNil)
+		_, err = tk.Exec("update tl set b = b + 1 where a = 1")
+		c.Assert(err, NotNil)
+		_, err = tk.Exec("insert into tl values (3, 1) on duplicated key update b = b + 1")
+		c.Assert(err, NotNil)
+		tk.MustExec("rollback")
 	}
 }