Problem writing a unit test using gorm and sql-mock

688 Views Asked by At

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

func Test_Create(t *testing.T) {
    db, mock, err := sqlmock.New()
    if err != nil {
        t.Errorf("Failed to open mock sql db, got error: %v", err)
    }
    if db == nil {
        t.Error("db is null")
    }
    if mock == nil {
        t.Error("mock is null")
    }
    defer db.Close()

    pDb, err := gorm.Open(postgres.New(postgres.Config{
        PreferSimpleProtocol: false,
        DriverName:           "postgres",
        Conn:                 db,
    }))
    if err != nil {
        t.Fatalf("gorm postgres fatal: %v", err)
    }
    student := Student{
        ID:   12345,
        Name: "Test user",
    }

    mock.ExpectBegin()
    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))
    mock.ExpectCommit()

    err = pDb.Create(student).Error
    if err != nil {
        t.Error("error:", err)
    }
}

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

1

There are 1 best solutions below

0
On

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

func Test_Create(t *testing.T) {
    db, mock, err := sqlmock.New()
    if err != nil {
        t.Errorf("Failed to open mock sql db, got error: %v", err)
    }
    if db == nil {
        t.Error("db is null")
    }
    if mock == nil {
        t.Error("mock is null")
    }
    defer db.Close()

    dialector := postgres.New(postgres.Config{
        DSN:                  "sqlmock_db_0",
        DriverName:           "postgres",
        Conn:                 db,
        PreferSimpleProtocol: true,
    })

    pDb, err := gorm.Open(dialector, &gorm.Config{})
    if err != nil {
        t.Fatalf("gorm postgres fatal: %v", err)
    }
    student := Student{
        ID:   12345,
        Name: "Test user",
    }

    mock.ExpectBegin()
    mock.ExpectQuery(
        regexp.QuoteMeta(`INSERT INTO "students" ("name","id") VALUES ($1,$2) RETURNING "id"`)).
        WithArgs(student.Name, student.ID).WillReturnRows(sqlmock.NewRows([]string{"id"}).AddRow(student.ID))
    mock.ExpectCommit()

    if err = pDb.Create(&student).Error; err != nil {
        t.Errorf("Failed to insert to gorm db, got error: %v", err)
        t.FailNow()
    }

    err = mock.ExpectationsWereMet()
    if err != nil {
        t.Errorf("Failed to meet expectations, got error: %v", err)
    }
}