mirror of https://github.com/VinGarcia/ksql.git
108 lines
2.3 KiB
Go
108 lines
2.3 KiB
Go
package ksql
|
|
|
|
import (
|
|
"context"
|
|
)
|
|
|
|
// mockTxBeginner mocks the ksql.TxBeginner interface
|
|
type mockTxBeginner struct {
|
|
DBAdapter
|
|
BeginTxFn func(ctx context.Context) (Tx, error)
|
|
}
|
|
|
|
func (b mockTxBeginner) BeginTx(ctx context.Context) (Tx, error) {
|
|
return b.BeginTxFn(ctx)
|
|
}
|
|
|
|
// mockDBAdapter mocks the ksql.DBAdapter interface
|
|
type mockDBAdapter struct {
|
|
ExecContextFn func(ctx context.Context, query string, args ...interface{}) (Result, error)
|
|
QueryContextFn func(ctx context.Context, query string, args ...interface{}) (Rows, error)
|
|
}
|
|
|
|
func (m mockDBAdapter) ExecContext(ctx context.Context, query string, args ...interface{}) (Result, error) {
|
|
return m.ExecContextFn(ctx, query, args...)
|
|
}
|
|
|
|
func (m mockDBAdapter) QueryContext(ctx context.Context, query string, args ...interface{}) (Rows, error) {
|
|
return m.QueryContextFn(ctx, query, args...)
|
|
}
|
|
|
|
type mockRows struct {
|
|
ScanFn func(...interface{}) error
|
|
CloseFn func() error
|
|
NextFn func() bool
|
|
ErrFn func() error
|
|
ColumnsFn func() ([]string, error)
|
|
}
|
|
|
|
func (m mockRows) Scan(values ...interface{}) error {
|
|
return m.ScanFn(values...)
|
|
}
|
|
|
|
func (m mockRows) Close() error {
|
|
if m.CloseFn == nil {
|
|
return nil
|
|
}
|
|
return m.CloseFn()
|
|
}
|
|
|
|
func (m mockRows) Next() bool {
|
|
return m.NextFn()
|
|
}
|
|
|
|
func (m mockRows) Err() error {
|
|
if m.ErrFn == nil {
|
|
return nil
|
|
}
|
|
return m.ErrFn()
|
|
}
|
|
|
|
func (m mockRows) Columns() ([]string, error) {
|
|
return m.ColumnsFn()
|
|
}
|
|
|
|
// mockResult mocks the ksql.Result interface
|
|
type mockResult struct {
|
|
LastInsertIdFn func() (int64, error)
|
|
RowsAffectedFn func() (int64, error)
|
|
}
|
|
|
|
func (m mockResult) LastInsertId() (int64, error) {
|
|
if m.LastInsertIdFn != nil {
|
|
return m.LastInsertIdFn()
|
|
}
|
|
return 0, nil
|
|
}
|
|
|
|
func (m mockResult) RowsAffected() (int64, error) {
|
|
if m.RowsAffectedFn != nil {
|
|
return m.RowsAffectedFn()
|
|
}
|
|
return 0, nil
|
|
}
|
|
|
|
// mockTx mocks the ksql.Tx interface
|
|
type mockTx struct {
|
|
DBAdapter
|
|
RollbackFn func(ctx context.Context) error
|
|
CommitFn func(ctx context.Context) error
|
|
}
|
|
|
|
func (m mockTx) Rollback(ctx context.Context) error {
|
|
return m.RollbackFn(ctx)
|
|
}
|
|
|
|
func (m mockTx) Commit(ctx context.Context) error {
|
|
return m.CommitFn(ctx)
|
|
}
|
|
|
|
// mockCloser mocks the io.Closer interface
|
|
type mockCloser struct {
|
|
CloseFn func() error
|
|
}
|
|
|
|
func (m mockCloser) Close() error {
|
|
return m.CloseFn()
|
|
}
|