https://github.com/driftprogramming/pgxpoolmock
pgx postgresql pool
https://github.com/driftprogramming/pgxpoolmock
go mock pgx pool postgresql
Last synced: 21 days ago
JSON representation
pgx postgresql pool
- Host: GitHub
- URL: https://github.com/driftprogramming/pgxpoolmock
- Owner: driftprogramming
- License: mit
- Created: 2021-06-04T04:27:55.000Z (over 4 years ago)
- Default Branch: main
- Last Pushed: 2025-08-07T07:39:21.000Z (6 months ago)
- Last Synced: 2025-08-07T09:30:06.019Z (6 months ago)
- Topics: go, mock, pgx, pool, postgresql
- Language: Go
- Homepage:
- Size: 718 KB
- Stars: 38
- Watchers: 1
- Forks: 17
- Open Issues: 6
-
Metadata Files:
- Readme: README.md
Awesome Lists containing this project
README
### PGX POOL MOCK
This repo is mock `pgxpool` in https://github.com/jackc/pgx. so that you can test your data access code locally, no real database is needed.
I also strongly recommend you use `pgxpool` rather than `pgx` only. see https://github.com/jackc/pgx/tree/master/pgxpool
### How to install
```
go get -u github.com/driftprogramming/pgxpoolmock
```
### How to Use
See file `order_dao_example_test.go` to figure out how to use it. Or see the below:
```go
package pgxpoolmock_test
import (
"testing"
"github.com/driftprogramming/pgxpoolmock"
"github.com/driftprogramming/pgxpoolmock/testdata"
"go.uber.org/mock/gomock"
"github.com/stretchr/testify/assert"
)
func TestInsertOrder(t *testing.T) {
ctrl := gomock.NewController(t)
defer ctrl.Finish()
mockPool := pgxpoolmock.NewMockPgxPool(ctrl)
commandTag := pgconn.CommandTag("INSERT 0 1")
mockPool.EXPECT().Exec(gomock.Any(), "insert into order (id, price) values ($1, $2)", 1, 2.3).Return(commandTag, nil)
orderDao := testdata.OrderDAO{
Pool: mockPool,
}
order := &testdata.Order{
ID: 1,
Price: 2.3,
}
err := orderDao.InsertOrder(order)
if err != nil {
t.Fatalf("%v", err)
}
}
func TestGetOneOrderByID(t *testing.T) {
ctrl := gomock.NewController(t)
defer ctrl.Finish()
mockPool := pgxpoolmock.NewMockPgxPool(ctrl)
columns := []string{"id", "price"}
pgxRow := pgxpoolmock.NewRow(columns, 1, 2.3)
mockPool.EXPECT().QueryRow(gomock.Any(), "select id, price from order where id = $1", 1).Return(pgxRow)
orderDao := testdata.OrderDAO{
Pool: mockPool,
}
actualOrder, err := orderDao.GetOneOrderByID(1)
if err != nil {
t.Fatalf("%v", err)
}
assert.NotNil(t, actualOrder)
assert.Equal(t, 1, actualOrder.ID)
assert.Equal(t, 2.3, actualOrder.Price)
}
func TestName(t *testing.T) {
t.Parallel()
ctrl := gomock.NewController(t)
defer ctrl.Finish()
// given
mockPool := pgxpoolmock.NewMockPgxPool(ctrl)
columns := []string{"id", "price"}
pgxRows := pgxpoolmock.NewRows(columns).AddRow(100, 100000.9).ToPgxRows()
mockPool.EXPECT().Query(gomock.Any(), gomock.Any(), gomock.Any()).Return(pgxRows, nil)
orderDao := testdata.OrderDAO{
Pool: mockPool,
}
// when
actualOrder := orderDao.GetOrderByID(1)
// then
assert.NotNil(t, actualOrder)
assert.Equal(t, 100, actualOrder.ID)
assert.Equal(t, 100000.9, actualOrder.Price)
}
```