How do I test the results after an update query?

[mchen628]

trafficstars

Here is my test function. I updated the mocked data, but it's returning the not updated version. `func TestSample(t *testing.T){ db, mock, err := sqlmock.New() if err != nil { t.Errorf("failed to open sqlmock database: %v", err) } defer db.Close()

testRows := sqlmock.NewRows([]string{"user","password"}).
	AddRow("mike", "m123").
	AddRow("chris", "c123")

mock.ExpectBegin()
mock.ExpectPrepare("UPDATE zebra SET user='mike'").ExpectExec().WillReturnResult(sqlmock.NewResult(0,1))
mock.ExpectCommit()
mock.ExpectQuery("SELECT (.+) FROM zebra *").WillReturnRows(testRows)

tx, err := db.Begin()
if err != nil {
	fmt.Printf("Something went wrong in begin: %v", err)
}
sql := "UPDATE zebra SET user='mike' WHERE user='test'"

stmt, err := tx.Prepare(sql)
if err != nil {
	fmt.Printf("Something went wrong in begin: %v", err)
}
defer stmt.Close()

_, err = stmt.Exec()
if err != nil {
	tx.Rollback()
	fmt.Printf("Something went wrong in begin: %v", err)
}
err = tx.Commit()
if err != nil {
	fmt.Printf("Something went wrong in begin: %v", err)
}

rows, err := db.Query("SELECT * FROM zebra")
if err != nil {
	fmt.Printf("Not exepecting error: %v", err)
}
defer rows.Close()
var (
	user string
	password string
)
for rows.Next() {
	err := rows.Scan(&user, &password)
	if err != nil {
		fmt.Println(err)
	}
	fmt.Println(user, password)
}
err = rows.Err()
if err != nil {
	fmt.Println(err)
}
err = mock.ExpectationsWereMet()
if err != nil {
	fmt.Printf("Unit test failed %v", err)
}

}`

[mchen628]

The select statement doesn't return the updated data after the update statement.

[l3pp4rd]

Sqlmock is for mocking, it is not a database. You have to mock what it needs to return, given your test scenario