How to expect a prepared insert ?

[ghost]

func TestLogHistory(t *testing.T) {
	var mock sqlmock.Sqlmock
	var err error
	db, mock, err = sqlmock.New()
	if err != nil {
		t.Fatal(err)
	}

	mock.ExpectPrepare("INSERT INTO").ExpectExec().WithArgs("2")

	stmt, err := db.Prepare("INSERT INTO history VALUES(?)")
	if err != nil {
		panic(err)
	}

	if _, err := stmt.Exec("2"); err != nil {
		t.Fatal(err)
	}

	if err := mock.ExpectationsWereMet(); err != nil {
		t.Fatal(err)
	}
}

--- FAIL: TestLogHistory (0.00s)
	db_test.go:33: exec query 'INSERT INTO history VALUES(?)' with args [2], must return a database/sql/driver.result, but it was not set for expectation *sqlmock.ExpectedExec as ExpectedExec => expecting Exec which:
		  - matches sql: 'INSERT INTO'
		  - is with arguments:
		    0 - 2

I can't figure out how to setup the mock to expect an insert. Any tips ?

[l3pp4rd]

Hi, you can look into the tests of sqlmock for more details. the error message says, that you must mock the result Exec should return

func TestLogHistory(t *testing.T) {
	var mock sqlmock.Sqlmock
	var err error
	db, mock, err = sqlmock.New()
	if err != nil {
		t.Fatal(err)
	}

	mock.ExpectPrepare("INSERT INTO").ExpectExec().WithArgs("2").WillReturnResult(sqlmock.NewResult(1, 1))

	stmt, err := db.Prepare("INSERT INTO history VALUES(?)")
	if err != nil {
		panic(err)
	}

	if _, err := stmt.Exec("2"); err != nil {
		t.Fatal(err)
	}

	if err := mock.ExpectationsWereMet(); err != nil {
		t.Fatal(err)
	}
}

Because it has to produce the result with a number off affected rows and last insert ID if available.

[ghost]

omg wow, I'm such a doofus, thank you so much

[l3pp4rd]

happens for everyone, cheers ;)

[ivaylopivanov]

@l3pp4rd I'm not able to use WillReturnRows with ExpectPrepare and I need that in order to make a select query. Is there a workaround?

[ivaylopivanov]

OK, found it in the tests ExpectQuery