使用gorm和sql-mock编写单元测试时遇到的问题

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

Problem writing a unit test using gorm and sql-mock

问题

这是我使用testing、gorm和sql-mock包的函数：

  1. func Test_Create(t *testing.T) {
  2. 	db, mock, err := sqlmock.New()
  3. 	if err != nil {
  4. 		t.Errorf("无法打开模拟的SQL数据库，错误信息：%v", err)
  5. 	}
  6. 	if db == nil {
  7. 		t.Error("db为空")
  8. 	}
  9. 	if mock == nil {
  10. 		t.Error("mock为空")
  11. 	}
  12. 	defer db.Close()
  14. 	pDb, err := gorm.Open(postgres.New(postgres.Config{
  15. 		PreferSimpleProtocol: false,
  16. 		DriverName:           "postgres",
  17. 		Conn:                 db,
  18. 	}))
  19. 	if err != nil {
  20. 		t.Fatalf("gorm打开postgres失败：%v", err)
  21. 	}
  22. 	student := Student{
  23. 		ID:   12345,
  24. 		Name: "测试用户",
  25. 	}
  27. 	mock.ExpectBegin()
  28. 	mock.ExpectExec(regexp.QuoteMeta(`INSERT INTO "students" ("id","name") VALUES ($1,$2) RETURNING "id"`)).WithArgs(student.ID, student.Name).WillReturnResult(sqlmock.NewResult(student.ID, 1))
  29. 	mock.ExpectCommit()
  31. 	err = pDb.Create(student).Error
  32. 	if err != nil {
  33. 		t.Error("错误信息：", err)
  34. 	}
  35. }

我不明白为什么在创建单元测试时会出现错误，就像下面的示例一样，有人可以帮忙吗？

英文:

this is my funckion using package testing, gorm and sql-mock:

  1. func Test_Create(t *testing.T) {
  2. 	db, mock, err := sqlmock.New()
  3. 	if err != nil {
  4. 		t.Errorf("Failed to open mock sql db, got error: %v", err)
  5. 	}
  6. 	if db == nil {
  7. 		t.Error("db is null")
  8. 	}
  9. 	if mock == nil {
  10. 		t.Error("mock is null")
  11. 	}
  12. 	defer db.Close()
  14. 	pDb, err := gorm.Open(postgres.New(postgres.Config{
  15. 		PreferSimpleProtocol: false,
  16. 		DriverName:           "postgres",
  17. 		Conn:                 db,
  18. 	}))
  19. 	if err != nil {
  20. 		t.Fatalf("gorm postgres fatal: %v", err)
  21. 	}
  22. 	student := Student{
  23. 		ID:   12345,
  24. 		Name: "Test user",
  25. 	}
  27. 	mock.ExpectBegin()
  28. 	mock.ExpectExec(regexp.QuoteMeta(`INSERT INTO "students" ("id","name") VALUES ($1,$2) RETURNING "id"`)).WithArgs(student.ID, student.Name).WillReturnResult(sqlmock.NewResult(student.ID, 1))
  29. 	mock.ExpectCommit()
  31. 	err = pDb.Create(student).Error
  32. 	if err != nil {
  33. 		t.Error("error:", err)
  34. 	}
  35. }

I don't understand why I have an error creating a unit test as in the example below, can anyone help?

答案1

得分: 1

好的，我已经解决了这个问题。这是工作的函数：

  1. func Test_Create(t *testing.T) {
  2. 	db, mock, err := sqlmock.New()
  3. 	if err != nil {
  4. 		t.Errorf("无法打开模拟的 SQL 数据库，错误信息：%v", err)
  5. 	}
  6. 	if db == nil {
  7. 		t.Error("db 为空")
  8. 	}
  9. 	if mock == nil {
  10. 		t.Error("mock 为空")
  11. 	}
  12. 	defer db.Close()
  14. 	dialector := postgres.New(postgres.Config{
  15. 		DSN:                  "sqlmock_db_0",
  16. 		DriverName:           "postgres",
  17. 		Conn:                 db,
  18. 		PreferSimpleProtocol: true,
  19. 	})
  21. 	pDb, err := gorm.Open(dialector, &gorm.Config{})
  22. 	if err != nil {
  23. 		t.Fatalf("gorm postgres 致命错误：%v", err)
  24. 	}
  25. 	student := Student{
  26. 		ID:   12345,
  27. 		Name: "测试用户",
  28. 	}
  30. 	mock.ExpectBegin()
  31. 	mock.ExpectQuery(
  32. 		regexp.QuoteMeta(`INSERT INTO "students" ("name","id") VALUES ($1,$2) RETURNING "id"`)).
  33. 		WithArgs(student.Name, student.ID).WillReturnRows(sqlmock.NewRows([]string{"id"}).AddRow(student.ID))
  34. 	mock.ExpectCommit()
  36. 	if err = pDb.Create(&student).Error; err != nil {
  37. 		t.Errorf("无法插入到 gorm 数据库，错误信息：%v", err)
  38. 		t.FailNow()
  39. 	}
  41. 	err = mock.ExpectationsWereMet()
  42. 	if err != nil {
  43. 		t.Errorf("未满足期望，错误信息：%v", err)
  44. 	}
  45. }
英文:

okay, I've already solved the problem. here's the working func:

  1. func Test_Create(t *testing.T) {
  2. 	db, mock, err := sqlmock.New()
  3. 	if err != nil {
  4. 		t.Errorf("Failed to open mock sql db, got error: %v", err)
  5. 	}
  6. 	if db == nil {
  7. 		t.Error("db is null")
  8. 	}
  9. 	if mock == nil {
  10. 		t.Error("mock is null")
  11. 	}
  12. 	defer db.Close()
  14. 	dialector := postgres.New(postgres.Config{
  15. 		DSN:                  "sqlmock_db_0",
  16. 		DriverName:           "postgres",
  17. 		Conn:                 db,
  18. 		PreferSimpleProtocol: true,
  19. 	})
  21. 	pDb, err := gorm.Open(dialector, &gorm.Config{})
  22. 	if err != nil {
  23. 		t.Fatalf("gorm postgres fatal: %v", err)
  24. 	}
  25. 	student := Student{
  26. 		ID:   12345,
  27. 		Name: "Test user",
  28. 	}
  30. 	mock.ExpectBegin()
  31. 	mock.ExpectQuery(
  32. 		regexp.QuoteMeta(`INSERT INTO "students" ("name","id") VALUES ($1,$2) RETURNING "id"`)).
  33. 		WithArgs(student.Name, student.ID).WillReturnRows(sqlmock.NewRows([]string{"id"}).AddRow(student.ID))
  34. 	mock.ExpectCommit()
  36. 	if err = pDb.Create(&student).Error; err != nil {
  37. 		t.Errorf("Failed to insert to gorm db, got error: %v", err)
  38. 		t.FailNow()
  39. 	}
  41. 	err = mock.ExpectationsWereMet()
  42. 	if err != nil {
  43. 		t.Errorf("Failed to meet expectations, got error: %v", err)
  44. 	}
  45. }

huangapple
  • 本文由 发表于 2023年1月21日 19:00:21
  • 转载请务必保留本文链接：https://go.coder-hub.com/75192805.html
  • go
  • go-gorm
  • go-sqlmock
  • unit-testing
匿名

发表评论

匿名网友

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

确定