测试使用go-sqlmock的gorm问题，使用mock.ExpectQuery和regexp.QuoteMeta比较查询。

huangapple
go评论129阅读模式
英文:

testing gorm with go-sqlmock issue comparing queries with mock.ExpectQuery and regexp.QuoteMeta

问题

我遇到了一个问题，无法将gorm的查询预期结果与真实查询结果进行比较，以下是我的代码：

  1. package repository
  3. import (
  4. 	"regexp"
  5. 	"testing"
  7. 	"github.com/DATA-DOG/go-sqlmock"
  8. 	"YOUR_GO_ROOT/pkg/domain"
  9. 	"github.com/stretchr/testify/assert"
  10. 	"gorm.io/driver/mysql"
  11. 	"gorm.io/gorm"
  12. )
  15. var successGetTransaction domain.Transaction = domain.Transaction{
  16. 	ID:          2,
  17. 	BuyerID:     2,
  18. 	SellerID:    5,
  19. 	ItemID:      2,
  20. 	MessageID:   2,
  21. 	ExpiredDate: "2022-09-010 01:01:00",
  22. 	CreatedAt:   "2022-09-08 01:01:00",
  23. }
  25. func TestSuccessGetTransactionByID(t *testing.T) {
  27. 	db, mock, err := sqlmock.New()
  28. 	assert.NoError(t, err)
  29. 	gdb, err := gorm.Open(mysql.New(mysql.Config{
  30. 		Conn:                      db,
  31. 		SkipInitializeWithVersion: true,
  32. 	}), &gorm.Config{})
  33. 	assert.NoError(t, err)
  35. 	rows := sqlmock.NewRows([]string{"id", "buyer_id", "seller_id", "item_id", "message_id", "expired_date", "created_at"}).
  36. 		AddRow(2, 2, 5, 2, 2, "2022-09-010 01:01:00", "2022-09-08 01:01:00")
  37. 	mock.ExpectQuery(regexp.QuoteMeta("SELECT * FROM transaction WHERE id = ?;")).WillReturnRows(rows)
  39. 	repo := DefaultClient(gdb)
  40. 	actualSectionList, _ := repo.GetTransactionByID(2)
  41. 	
  42. 	assert.Equal(t, successGetTransaction, actualSectionList, "ambas listas deberian ser iguales")
  43. 	assert.NoError(t, mock.ExpectationsWereMet())
  44. }

这是domain模块的代码：

  1. package domain
  3. type Transaction struct {
  4. 	ID          int64  `gorm:"primaryKey;column:id"`
  5. 	BuyerID     int64  `gorm:"column:buyer_id"`
  6. 	SellerID    int64  `gorm:"column:seller_id"`
  7. 	ItemID      int    `gorm:"column:item_id"`
  8. 	MessageID   int    `gorm:"column:message_id"`
  9. 	ExpiredDate string `gorm:"column:expired_date"`
  10. 	CreatedAt   string `gorm:"column:created_at"`
  11. }
  13. func (Transaction) TableName() string {
  14. 	return "transaction"
  15. }
  17. type TransactionStatus struct {
  18. 	ID             int64  `gorm:"primaryKey;column:id"`
  19. 	TransactionID  int64  `gorm:"column:transaction_id"`
  20. 	Status         int    `gorm:"column:status"`
  21. 	NotificationID int    `gorm:"column:notification_id"`
  22. 	CreatedAt      string `gorm:"column:created_at"`
  23. }
  25. func (TransactionStatus) TableName() string {
  26. 	return "transaction_status"
  27. }

这是我正在测试的函数：

  1. package repository
  3. import (
  4. 	"fmt"
  6. 	"YOUR_GO_ROOT/pkg/domain"
  7. 	"gorm.io/gorm"
  8. )
  10. type RepositoryClient interface {
  11. 	GetTransactionByID(id int) (domain.Transaction, error)
  12. }
  14. type repositoryClient struct {
  15. 	db *gorm.DB
  16. }
  18. func DefaultClient(db *gorm.DB) RepositoryClient {
  19. 	return &repositoryClient{
  20. 		db: db,
  21. 	}
  22. }
  24. func (rc repositoryClient) GetTransactionByID(id int) (domain.Transaction, error) {
  25. 	trans := domain.Transaction{}
  26. 	status := rc.db.Where("id = ?", id).Find(&trans)
  28. 	if status.Error != nil {
  29. 		return domain.Transaction{}, status.Error
  30. 	}
  31. 	if trans == (domain.Transaction{}) {
  32. 		return domain.Transaction{}, fmt.Errorf("error finding transaction id %v", id)
  33. 	}
  34. 	return trans, nil
  35. }

这是我从控制台收到的错误信息：

  1. Query: could not match actual sql: "SELECT * FROM `transaction` WHERE id = ?" with expected regexp "SELECT \* FROM transaction WHERE id = \?;"
  3. [seccion][1]中存在一个被替换为"SELECT(.*)"的答案，但根据我所读到的，这不是一个真正的解决方案。
  5. [1]: https://stackoverflow.com/questions/68528616/go-unit-test-could-not-match-actual-sql-with-expected-regexp-while-mocking-gor

希望这能帮到你！

英文:

I am having problems comparing the query expected with the true query of gorm, this is my code:

  1. package repository
  3. import (
  4. 	"regexp"
  5. 	"testing"
  7. 	"github.com/DATA-DOG/go-sqlmock"
  8. 	"YOUR_GO_ROOT/pkg/domain"
  9. 	"github.com/stretchr/testify/assert"
  10. 	"gorm.io/driver/mysql"
  11. 	"gorm.io/gorm"
  12. )
  15. var successGetTransaction domain.Transaction = domain.Transaction{
  16. 	ID:          2,
  17. 	BuyerID:     2,
  18. 	SellerID:    5,
  19. 	ItemID:      2,
  20. 	MessageID:   2,
  21. 	ExpiredDate: "2022-09-010 01:01:00",
  22. 	CreatedAt:   "2022-09-08 01:01:00",
  23. }
  25. func TestSuccessGetTransactionByID(t *testing.T) {
  27. 	db, mock, err := sqlmock.New()
  28. 	assert.NoError(t, err)
  29. 	gdb, err := gorm.Open(mysql.New(mysql.Config{
  30. 		Conn:                      db,
  31. 		SkipInitializeWithVersion: true,
  32. 	}), &gorm.Config{})
  33. 	assert.NoError(t, err)
  35. 	rows := sqlmock.NewRows([]string{"id", "buyer_id", "seller_id", "item_id", "message_id", "expired_date", "created_at"}).
  36. 		AddRow(2, 2, 5, 2, 2, "2022-09-010 01:01:00", "2022-09-08 01:01:00")
  37. 	mock.ExpectQuery(regexp.QuoteMeta("SELECT * FROM transaction WHERE id = ?;")).WillReturnRows(rows)
  39. 	repo := DefaultClient(gdb)
  40. 	actualSectionList, _ := repo.GetTransactionByID(2)
  41. 	
  42. 	assert.Equal(t, successGetTransaction, actualSectionList, "ambas listas deberian ser iguales")
  43. 	assert.NoError(t, mock.ExpectationsWereMet())
  44. }

here is the module domain:

  1. package domain
  3. type Transaction struct {
  4. 	ID          int64  `gorm:"primaryKey;column:id"`
  5. 	BuyerID     int64  `gorm:"column:buyer_id"`
  6. 	SellerID    int64  `gorm:"column:seller_id"`
  7. 	ItemID      int    `gorm:"column:item_id"`
  8. 	MessageID   int    `gorm:"column:message_id"`
  9. 	ExpiredDate string `gorm:"column:expired_date"`
  10. 	CreatedAt   string `gorm:"column:created_at"`
  11. }
  13. func (Transaction) TableName() string {
  14. 	return "transaction"
  15. }
  17. type TransactionStatus struct {
  18. 	ID             int64  `gorm:"primaryKey;column:id"`
  19. 	TransactionID  int64  `gorm:"column:transaction_id"`
  20. 	Status         int    `gorm:"column:status"`
  21. 	NotificationID int    `gorm:"column:notification_id"`
  22. 	CreatedAt      string `gorm:"column:created_at"`
  23. }
  25. func (TransactionStatus) TableName() string {
  26. 	return "transaction_status"
  27. }

and here the function that I am testing:

  1. package repository
  3. import (
  4. 	"fmt"
  6. 	"YOUR_GO_ROOT/pkg/domain"
  7. 	"gorm.io/gorm"
  8. )
  10. type RepositoryClient interface {
  11. 	GetTransactionByID(id int) (domain.Transaction, error)
  12. }
  14. type repositoryClient struct {
  15. 	db *gorm.DB
  16. }
  18. func DefaultClient(db *gorm.DB) RepositoryClient {
  19. 	return &repositoryClient{
  20. 		db: db,
  21. 	}
  22. }
  24. func (rc repositoryClient) GetTransactionByID(id int) (domain.Transaction, error) {
  25. 	trans := domain.Transaction{}
  26. 	status := rc.db.Where("id = ?", id).Find(&trans)
  28. 	if status.Error != nil {
  29. 		return domain.Transaction{}, status.Error
  30. 	}
  31. 	if trans == (domain.Transaction{}) {
  32. 		return domain.Transaction{}, fmt.Errorf("error finding transaction id %v", id)
  33. 	}
  34. 	return trans, nil
  35. }

this is the error that I am getting from the console:

  1. Query: could not match actual sql: "SELECT * FROM `transaction` WHERE id = ?" with expected regexp "SELECT \* FROM transaction WHERE id = \?;"[0m[33m[0.218ms] [34;1m[rows:0][0m SELECT * FROM `transaction` WHERE id = 2

in this seccion exist an answer that is substitute with "SELECT(.*)" but according to what i had read that is not a real solution

答案1

得分: 1

让我试着帮你找出问题所在。我已经下载了你的所有文件，domain.go 和 repository.go 看起来都没问题。然而，在 repository_test.go 文件中我发现了一些小问题：

  1. 你编写的 SQL 查询中缺少反引号（backticks）。
  2. 查询末尾多了一个分号（;）。
  3. 没有调用 WithArgs(2) 方法。

如果你修复了这些小问题，你的代码应该像下面这样：

  1. // ... 省略部分代码
  2. func TestSuccessGetTransactionByID(t *testing.T) {
  3. 	db, mock, err := sqlmock.New()
  4. 	assert.NoError(t, err)
  5. 	gdb, err := gorm.Open(mysql.New(mysql.Config{
  6. 		Conn:                      db,
  7. 		SkipInitializeWithVersion: true,
  8. 	}), &gorm.Config{})
  9. 	assert.NoError(t, err)
  11. 	rows := sqlmock.NewRows([]string{"id", "buyer_id", "seller_id", "item_id", "message_id", "expired_date", "created_at"}).AddRow(2, 2, 5, 2, 2, "2022-09-010 01:01:00", "2022-09-08 01:01:00")
  12. 	mock.ExpectQuery(regexp.QuoteMeta("SELECT * FROM `transaction` WHERE id = ?")).WithArgs(2).WillReturnRows(rows)
  14. 	repo := DefaultClient(gdb)
  15. 	actualSectionList, _ := repo.GetTransactionByID(2)
  17. 	assert.Equal(t, successGetTransaction, actualSectionList, "ambas listas deberian ser iguales")
  18. 	assert.NoError(t, mock.ExpectationsWereMet())
  19. }

然后，如果你尝试运行测试，应该可以正常工作。如果这解决了你的问题，请告诉我，谢谢！

英文:

Let me try to help to figure out the issue. I downloaded all of your files and the domain.go and the repository.go look fine to me.
However, I found some small issues in the repository_test.go file:

  1. Backticks missing in the SQL query you wrote
  2. An extra ; at the end of the query
  3. The invocation to the method WithArgs(2) is missing

If you adjusted these small issues you should have a code that looks like this one:

  1. // ... omitted for brevity
  2. func TestSuccessGetTransactionByID(t *testing.T) {
  3. 	db, mock, err := sqlmock.New()
  4. 	assert.NoError(t, err)
  5. 	gdb, err := gorm.Open(mysql.New(mysql.Config{
  6. 		Conn:                      db,
  7. 		SkipInitializeWithVersion: true,
  8. 	}), &gorm.Config{})
  9. 	assert.NoError(t, err)
  11. 	rows := sqlmock.NewRows([]string{"id", "buyer_id", "seller_id", "item_id", "message_id", "expired_date", "created_at"}).AddRow(2, 2, 5, 2, 2, "2022-09-010 01:01:00", "2022-09-08 01:01:00")
  12. 	mock.ExpectQuery(regexp.QuoteMeta("SELECT * FROM `transaction` WHERE id = ?")).WithArgs(2).WillReturnRows(rows)
  14. 	repo := DefaultClient(gdb)
  15. 	actualSectionList, _ := repo.GetTransactionByID(2)
  17. 	assert.Equal(t, successGetTransaction, actualSectionList, "ambas listas deberian ser iguales")
  18. 	assert.NoError(t, mock.ExpectationsWereMet())
  19. }

Then, if you try to run the test it should work.
Let me know if this solves your issue, thanks!

huangapple
  • 本文由 发表于 2022年12月16日 02:12:37
  • 转载请务必保留本文链接：https://go.coder-hub.com/74815846.html
  • go
  • go-gorm
  • go-sqlmock
  • regex
  • testing
匿名

发表评论

匿名网友

:?: :razz: :sad: :evil: :!: :smile: :oops: :grin: :eek: :shock: :???: :cool: :lol: :mad: :twisted: :roll: :wink: :idea: :arrow: :neutral: :cry: :mrgreen:

确定