Michael McGuinness
b1f8ac453b
feat!: base template * first bit of templating * codeowners * linuxbased * restart * baseline project * add grpc api and basic integration test * startqueue * queueMsg * splitscripts * migrations * queueintegrationtest * gateway
59 lines
1.5 KiB
Go
59 lines
1.5 KiB
Go
package pgxmock
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
|
|
pgx "github.com/jackc/pgx/v5"
|
|
"github.com/jackc/pgx/v5/pgxpool"
|
|
)
|
|
|
|
type pgxmockConn struct {
|
|
pgxmock
|
|
}
|
|
|
|
// NewConn creates PgxConnIface database connection and a mock to manage expectations.
|
|
// Accepts options, like QueryMatcherOption, to match SQL query strings in more sophisticated ways.
|
|
func NewConn(options ...func(*pgxmock) error) (PgxConnIface, error) {
|
|
smock := &pgxmockConn{}
|
|
smock.ordered = true
|
|
return smock, smock.open(options)
|
|
}
|
|
|
|
func (c *pgxmockConn) Config() *pgx.ConnConfig {
|
|
return &pgx.ConnConfig{}
|
|
}
|
|
|
|
type pgxmockPool struct {
|
|
pgxmock
|
|
}
|
|
|
|
// NewPool creates PgxPoolIface pool of database connections and a mock to manage expectations.
|
|
// Accepts options, like QueryMatcherOption, to match SQL query strings in more sophisticated ways.
|
|
func NewPool(options ...func(*pgxmock) error) (PgxPoolIface, error) {
|
|
smock := &pgxmockPool{}
|
|
smock.ordered = true
|
|
return smock, smock.open(options)
|
|
}
|
|
|
|
func (p *pgxmockPool) Close() {
|
|
p.pgxmock.Close(context.Background())
|
|
}
|
|
|
|
func (p *pgxmockPool) Acquire(context.Context) (*pgxpool.Conn, error) {
|
|
return nil, errors.New("pgpool.Acquire() method is not implemented")
|
|
}
|
|
|
|
func (p *pgxmockPool) Config() *pgxpool.Config {
|
|
return &pgxpool.Config{}
|
|
}
|
|
|
|
// AsConn is similar to Acquire but returns proper mocking interface
|
|
func (p *pgxmockPool) AsConn() PgxConnIface {
|
|
return &pgxmockConn{pgxmock: p.pgxmock}
|
|
}
|
|
|
|
func (p *pgxmockPool) Stat() *pgxpool.Stat {
|
|
return &pgxpool.Stat{}
|
|
}
|