MySQL update statement tests.

2019-08-02 11:08:24 +02:00 · 2019-08-02 11:08:24 +02:00 · 7660bdd8b5
commit 7660bdd8b5
parent a46e8c1c51
12 changed files with 406 additions and 115 deletions
--- a/tests/mysql/update_test.go
+++ b/tests/mysql/update_test.go
@ -0,0 +1,224 @@
+package mysql
+
+import (
+	"context"
+	. "github.com/go-jet/jet"
+	"github.com/go-jet/jet/internal/testutils"
+	"github.com/go-jet/jet/tests/.gentestdata/mysql/test_sample/model"
+	. "github.com/go-jet/jet/tests/.gentestdata/mysql/test_sample/table"
+	"gotest.tools/assert"
+	"testing"
+	"time"
+)
+
+func TestUpdateValues(t *testing.T) {
+	setupLinkTableForUpdateTest(t)
+
+	query := Link.
+		UPDATE(Link.Name, Link.URL).
+		SET("Bong", "http://bong.com").
+		WHERE(Link.Name.EQ(String("Bing")))
+
+	var expectedSQL = `
+UPDATE test_sample.link
+SET name = 'Bong', url = 'http://bong.com'
+WHERE link.name = 'Bing';
+`
+
+	testutils.AssertDebugStatementSql(t, query, expectedSQL, "Bong", "http://bong.com", "Bing")
+
+	testutils.AssertExec(t, query, db)
+
+	links := []model.Link{}
+
+	err := Link.
+		SELECT(Link.AllColumns).
+		WHERE(Link.Name.EQ(String("Bong"))).
+		Query(db, &links)
+
+	assert.NilError(t, err)
+	assert.Equal(t, len(links), 1)
+	assert.DeepEqual(t, links[0], model.Link{
+		ID:   204,
+		URL:  "http://bong.com",
+		Name: "Bong",
+	})
+}
+
+func TestUpdateWithSubQueries(t *testing.T) {
+	setupLinkTableForUpdateTest(t)
+
+	query := Link.
+		UPDATE(Link.Name, Link.URL).
+		SET(
+			SELECT(String("Bong")),
+			SELECT(Link2.URL).
+				FROM(Link2).
+				WHERE(Link2.Name.EQ(String("Youtube"))),
+		).
+		WHERE(Link.Name.EQ(String("Bing")))
+
+	expectedSQL := `
+UPDATE test_sample.link
+SET name = (
+     SELECT ?
+), url = (
+     SELECT link2.url AS "link2.url"
+     FROM test_sample.link2
+     WHERE link2.name = ?
+)
+WHERE link.name = ?;
+`
+	testutils.AssertStatementSql(t, query, expectedSQL, "Bong", "Youtube", "Bing")
+
+	testutils.AssertExec(t, query, db)
+}
+
+func TestUpdateAndReturning(t *testing.T) {
+	defer func() {
+		r := recover()
+		assert.Equal(t, r, "jet: MySQL dialect does not support RETURNING.")
+	}()
+
+	stmt := Link.
+		UPDATE(Link.Name, Link.URL).
+		SET("DuckDuckGo", "http://www.duckduckgo.com").
+		WHERE(Link.Name.EQ(String("Ask"))).
+		RETURNING(Link.AllColumns)
+
+	stmt.Query(db, &struct{}{})
+}
+
+func TestUpdateWithModelData(t *testing.T) {
+	setupLinkTableForUpdateTest(t)
+
+	link := model.Link{
+		ID:   201,
+		URL:  "http://www.duckduckgo.com",
+		Name: "DuckDuckGo",
+	}
+
+	stmt := Link.
+		UPDATE(Link.AllColumns).
+		MODEL(link).
+		WHERE(Link.ID.EQ(Int(int64(link.ID))))
+
+	expectedSQL := `
+UPDATE test_sample.link
+SET id = ?, url = ?, name = ?, description = ?
+WHERE link.id = ?;
+`
+	testutils.AssertStatementSql(t, stmt, expectedSQL, int32(201), "http://www.duckduckgo.com", "DuckDuckGo", nil, int64(201))
+
+	testutils.AssertExec(t, stmt, db)
+}
+
+func TestUpdateWithModelDataAndPredefinedColumnList(t *testing.T) {
+
+	setupLinkTableForUpdateTest(t)
+
+	link := model.Link{
+		ID:   201,
+		URL:  "http://www.duckduckgo.com",
+		Name: "DuckDuckGo",
+	}
+
+	updateColumnList := ColumnList{Link.Description, Link.Name, Link.URL}
+
+	stmt := Link.
+		UPDATE(updateColumnList).
+		MODEL(link).
+		WHERE(Link.ID.EQ(Int(int64(link.ID))))
+
+	var expectedSQL = `
+UPDATE test_sample.link
+SET description = NULL, name = 'DuckDuckGo', url = 'http://www.duckduckgo.com'
+WHERE link.id = 201;
+`
+	//fmt.Println(stmt.DebugSql())
+
+	testutils.AssertDebugStatementSql(t, stmt, expectedSQL, nil, "DuckDuckGo", "http://www.duckduckgo.com", int64(201))
+
+	testutils.AssertExec(t, stmt, db)
+}
+
+func TestUpdateWithInvalidModelData(t *testing.T) {
+	defer func() {
+		r := recover()
+
+		assert.Equal(t, r, "missing struct field for column : id")
+	}()
+
+	setupLinkTableForUpdateTest(t)
+
+	link := struct {
+		Ident       int
+		URL         string
+		Name        string
+		Description *string
+		Rel         *string
+	}{
+		Ident: 201,
+		URL:   "http://www.duckduckgo.com",
+		Name:  "DuckDuckGo",
+	}
+
+	stmt := Link.
+		UPDATE(Link.AllColumns).
+		MODEL(link).
+		WHERE(Link.ID.EQ(Int(int64(link.Ident))))
+
+	stmt.Sql()
+}
+
+func TestUpdateQueryContext(t *testing.T) {
+	setupLinkTableForUpdateTest(t)
+
+	updateStmt := Link.
+		UPDATE(Link.Name, Link.URL).
+		SET("Bong", "http://bong.com").
+		WHERE(Link.Name.EQ(String("Bing")))
+
+	ctx, cancel := context.WithTimeout(context.Background(), 1*time.Microsecond)
+	defer cancel()
+
+	time.Sleep(10 * time.Millisecond)
+
+	dest := []model.Link{}
+	err := updateStmt.QueryContext(ctx, db, &dest)
+
+	assert.Error(t, err, "context deadline exceeded")
+}
+
+func TestUpdateExecContext(t *testing.T) {
+	setupLinkTableForUpdateTest(t)
+
+	updateStmt := Link.
+		UPDATE(Link.Name, Link.URL).
+		SET("Bong", "http://bong.com").
+		WHERE(Link.Name.EQ(String("Bing")))
+
+	ctx, cancel := context.WithTimeout(context.Background(), 1*time.Microsecond)
+	defer cancel()
+
+	time.Sleep(10 * time.Millisecond)
+
+	_, err := updateStmt.ExecContext(ctx, db)
+
+	assert.Error(t, err, "context deadline exceeded")
+}
+
+func setupLinkTableForUpdateTest(t *testing.T) {
+
+	cleanUpLinkTable(t)
+
+	_, err := Link.INSERT(Link.ID, Link.URL, Link.Name, Link.Description).
+		VALUES(200, "http://www.postgresqltutorial.com", "PostgreSQL Tutorial", DEFAULT).
+		VALUES(201, "http://www.ask.com", "Ask", DEFAULT).
+		VALUES(202, "http://www.ask.com", "Ask", DEFAULT).
+		VALUES(203, "http://www.yahoo.com", "Yahoo", DEFAULT).
+		VALUES(204, "http://www.bing.com", "Bing", DEFAULT).
+		Exec(db)
+
+	assert.NilError(t, err)
+}