model_query_test.go

package dbx

import (
	"regexp"
	"testing"

	"github.com/DATA-DOG/go-sqlmock"
)

type testUser struct {
	ID    int64  `db:"id,pk"`
	Name  string `db:"name"`
	Email string `db:"email"`
}

func (testUser) TableName() string { return "users" }

func TestModelInsertSQLiteAutoPK(t *testing.T) {
	sqlDB, mock, err := sqlmock.New()
	if err != nil {
		t.Fatalf("sqlmock: %v", err)
	}
	defer sqlDB.Close()

	db := NewFromDB(sqlDB, "sqlite3")
	u := &testUser{Name: "alice", Email: "a@example.com"}

	mock.ExpectExec(regexp.QuoteMeta("INSERT INTO `users` (`email`, `name`) VALUES (?, ?)")).
		WithArgs("a@example.com", "alice").
		WillReturnResult(sqlmock.NewResult(42, 1))

	if err := db.Model(u).Insert(); err != nil {
		t.Fatalf("insert error: %v", err)
	}
	if u.ID != 42 {
		t.Fatalf("expected id=42, got %d", u.ID)
	}
	if err := mock.ExpectationsWereMet(); err != nil {
		t.Fatalf("expectations: %v", err)
	}
}

func TestModelInsertPostgresReturning(t *testing.T) {
	sqlDB, mock, err := sqlmock.New()
	if err != nil {
		t.Fatalf("sqlmock: %v", err)
	}
	defer sqlDB.Close()

	db := NewFromDB(sqlDB, "postgres")
	u := &testUser{Name: "bob", Email: "b@example.com"}

	rows := sqlmock.NewRows([]string{"id"}).AddRow(int64(100))
	mock.ExpectQuery(regexp.QuoteMeta(`INSERT INTO "users" ("email", "name") VALUES ($1, $2) RETURNING "id"`)).
		WithArgs("b@example.com", "bob").
		WillReturnRows(rows)

	if err := db.Model(u).Insert(); err != nil {
		t.Fatalf("insert error: %v", err)
	}
	if u.ID != 100 {
		t.Fatalf("expected id=100, got %d", u.ID)
	}
	if err := mock.ExpectationsWereMet(); err != nil {
		t.Fatalf("expectations: %v", err)
	}
}

func TestModelUpdateDeleteByPK(t *testing.T) {
	sqlDB, mock, err := sqlmock.New()
	if err != nil {
		t.Fatalf("sqlmock: %v", err)
	}
	defer sqlDB.Close()

	db := NewFromDB(sqlDB, "sqlite3")
	u := &testUser{ID: 7, Name: "c", Email: "c@example.com"}

	mock.ExpectExec(regexp.QuoteMeta("UPDATE `users` SET `email`=?, `name`=? WHERE `id`=?")).
		WithArgs("c@example.com", "c", int64(7)).
		WillReturnResult(sqlmock.NewResult(0, 1))

	if err := db.Model(u).Update(); err != nil {
		t.Fatalf("update error: %v", err)
	}

	mock.ExpectExec(regexp.QuoteMeta("DELETE FROM `users` WHERE `id`=?")).
		WithArgs(int64(7)).
		WillReturnResult(sqlmock.NewResult(0, 1))

	if err := db.Model(u).Delete(); err != nil {
		t.Fatalf("delete error: %v", err)
	}

	if err := mock.ExpectationsWereMet(); err != nil {
		t.Fatalf("expectations: %v", err)
	}
}

func TestModelUpdateMissingPK(t *testing.T) {
	type noPK struct {
		Name string
	}

	db := NewFromDB(nil, "sqlite3")
	m := &noPK{Name: "x"}
	err := db.Model(m).Update()
	if err != MissingPKError {
		t.Fatalf("expected MissingPKError, got %v", err)
	}
}