bob/upsert_test.go

package bob_test

import (
	"reflect"
	"testing"

	"github.com/aldy505/bob"
)

func TestUpsert(t *testing.T) {
	t.Run("should be able to generate upsert query for mysql", func(t *testing.T) {
		sql, args, err := bob.
			Upsert("users", bob.MySQL).
			Columns("name", "email").
			Values("John Doe", "john@doe.com").
			Replace("name", "John Does").
			ToSql()
		if err != nil {
			t.Error(err)
		}

		desiredSql := "INSERT INTO \"users\" (\"name\", \"email\") VALUES (?, ?) ON DUPLICATE KEY UPDATE \"name\" = ?;"
		desiredArgs := []interface{}{"John Doe", "john@doe.com", "John Does"}

		if sql != desiredSql {
			t.Error("sql is not the same as result: ", sql)
		}
		if !reflect.DeepEqual(args, desiredArgs) {
			t.Error("args is not the same as result: ", args)
		}
	})

	t.Run("should be able to generate upsert query for postgres", func(t *testing.T) {
		sql, args, err := bob.
			Upsert("users", bob.PostgreSQL).
			Columns("name", "email").
			Values("John Doe", "john@doe.com").
			Key("email").
			Replace("name", "John Does").
			PlaceholderFormat(bob.Dollar).
			ToSql()
		if err != nil {
			t.Error(err)
		}

		desiredSql := "INSERT INTO \"users\" (\"name\", \"email\") VALUES ($1, $2) ON CONFLICT (\"email\") DO UPDATE SET \"name\" = $3;"
		desiredArgs := []interface{}{"John Doe", "john@doe.com", "John Does"}

		if sql != desiredSql {
			t.Error("sql is not the same as result: ", sql)
		}
		if !reflect.DeepEqual(args, desiredArgs) {
			t.Error("args is not the same as result: ", args)
		}
	})

	t.Run("should be able to generate upsert query for sqlite", func(t *testing.T) {
		sql, args, err := bob.
			Upsert("users", bob.SQLite).
			Columns("name", "email").
			Values("John Doe", "john@doe.com").
			Key("email").
			Replace("name", "John Does").
			PlaceholderFormat(bob.Question).
			ToSql()
		if err != nil {
			t.Error(err)
		}

		desiredSql := "INSERT INTO \"users\" (\"name\", \"email\") VALUES (?, ?) ON CONFLICT (\"email\") DO UPDATE SET \"name\" = ?;"
		desiredArgs := []interface{}{"John Doe", "john@doe.com", "John Does"}

		if sql != desiredSql {
			t.Error("sql is not the same as result: ", sql)
		}
		if !reflect.DeepEqual(args, desiredArgs) {
			t.Error("args is not the same as result: ", args)
		}
	})

	t.Run("should be able to generate upsert query for mssql", func(t *testing.T) {
		sql, args, err := bob.
			Upsert("users", bob.MSSQL).
			Columns("name", "email").
			Values("John Doe", "john@doe.com").
			Key("email", "john@doe.com").
			Replace("name", "John Does").
			PlaceholderFormat(bob.AtP).
			ToSql()
		if err != nil {
			t.Error(err)
		}

		desiredSql := "IF NOT EXISTS (SELECT * FROM \"users\" WHERE \"email\" = @p1) INSERT INTO \"users\" (\"name\", \"email\") VALUES (@p2, @p3) ELSE UPDATE \"users\" SET \"name\" = @p4 WHERE \"email\" = @p5;"
		desiredArgs := []interface{}{"john@doe.com", "John Doe", "john@doe.com", "John Does", "john@doe.com"}

		if sql != desiredSql {
			t.Error("sql is not the same as result: ", sql)
		}
		if !reflect.DeepEqual(args, desiredArgs) {
			t.Error("args is not the same as result: ", args)
		}
	})

	t.Run("should emit error without table name", func(t *testing.T) {
		_, _, err := bob.Upsert("", bob.MySQL).ToSql()
		if err == nil && err.Error() != "upsert statement must specify a table" {
			t.Error(err)
		}
	})

	t.Run("should emit error without columns", func(t *testing.T) {
		_, _, err := bob.Upsert("users", bob.PostgreSQL).ToSql()
		if err.Error() != "upsert statement must have at least one column" {
			t.Error(err)
		}
	})

	t.Run("should emit error without values", func(t *testing.T) {
		_, _, err := bob.Upsert("users", bob.PostgreSQL).Columns("name", "email").ToSql()
		if err.Error() != "upsert statements must have at least one set of values" {
			t.Error(err)
		}
	})

	t.Run("should emit error without replaces", func(t *testing.T) {
		_, _, err := bob.Upsert("users", bob.PostgreSQL).Columns("name", "email").Values("James", "james@mail.com").ToSql()
		if err.Error() != "upsert statement must have at least one key value pair to be replaced" {
			t.Error(err)
		}
	})

	t.Run("should emit error without key and value for mssql", func(t *testing.T) {
		_, _, err := bob.Upsert("users", bob.MSSQL).Columns("name", "email").Values("James", "james@mail.com").Replace("name", "Thomas").ToSql()
		if err.Error() != "unique key and value must be provided for MS SQL" {
			t.Error(err)
		}
	})

	t.Run("should emit error without key and value for mssql", func(t *testing.T) {
		_, _, err := bob.Upsert("users", bob.SQLite).Columns("name", "email").Values("James", "james@mail.com").Replace("name", "Thomas").ToSql()
		if err.Error() != "unique key must be provided for PostgreSQL and SQLite" {
			t.Log(err.Error())
			t.Error(err)
		}
	})

	t.Run("should emit error if dialect not supported", func(t *testing.T) {
		_, _, err := bob.Upsert("users", 100).Columns("name", "email").Values("James", "james@mail.com").Replace("name", "Thomas").ToSql()
		if err.Error() != "provided database dialect is not supported" {
			t.Log(err.Error())
			t.Error(err)
		}
	})
}