book_persistence_get_test.go

func TestGet(t *testing.T) {
	mockDB, mock, err := sqlmock.New()
	require.Nil(t, err)
	defer mockDB.Close()

	db := sqlx.NewDb(mockDB, "sqlmock")
	bookRepository := NewBookRepository(db)

	rows := sqlmock.NewRows([]string{"id, title, author"}).
		AddRow("testID", "testTitle", "testAuthor")

	mock.ExpectQuery("SELECT * FROM books WHERE id=?").
		WithArgs("testID").
		WillReturnRows(rows)

	newBook, err := bookRepository.Get("testID")
	require.Nil(t, err)
	require.Equal(t, newBook.ID, "testID")
}