mirror of https://github.com/VinGarcia/ksql.git
Add more tests to the mocks for codecov
Vinícius Garcia
parent
3b7209b115
commit
3613ad769a
230
mocks_test.go
230
mocks_test.go
|
|
@ -3,6 +3,7 @@ package ksql_test
|
|||
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
"testing"
|
||||
|
||||
"github.com/vingarcia/ksql"
|
||||
|
|
@ -142,4 +143,233 @@ func TestMock(t *testing.T) {
|
|||
tt.AssertEqual(t, executed, true)
|
||||
})
|
||||
})
|
||||
|
||||
t.Run("should call the user provided behavior correctly", func(t *testing.T) {
|
||||
t.Run("Insert", func(t *testing.T) {
|
||||
ctx := context.WithValue(context.Background(), "key", "value")
|
||||
var capturedArgs struct {
|
||||
ctx context.Context
|
||||
table ksql.Table
|
||||
record interface{}
|
||||
}
|
||||
mock := ksql.Mock{
|
||||
InsertFn: func(ctx context.Context, table ksql.Table, record interface{}) error {
|
||||
capturedArgs.ctx = ctx
|
||||
capturedArgs.table = table
|
||||
capturedArgs.record = record
|
||||
return fmt.Errorf("fake-error")
|
||||
},
|
||||
}
|
||||
err := mock.Insert(ctx, UsersTable, &User{
|
||||
Name: "fake-name",
|
||||
Age: 42,
|
||||
})
|
||||
|
||||
tt.AssertErrContains(t, err, "fake-error")
|
||||
tt.AssertEqual(t, capturedArgs.ctx, ctx)
|
||||
tt.AssertEqual(t, capturedArgs.table, UsersTable)
|
||||
tt.AssertEqual(t, capturedArgs.record, &User{
|
||||
Name: "fake-name",
|
||||
Age: 42,
|
||||
})
|
||||
})
|
||||
|
||||
t.Run("Update", func(t *testing.T) {
|
||||
ctx := context.Background()
|
||||
var capturedArgs struct {
|
||||
ctx context.Context
|
||||
table ksql.Table
|
||||
record interface{}
|
||||
}
|
||||
mock := ksql.Mock{
|
||||
UpdateFn: func(ctx context.Context, table ksql.Table, record interface{}) error {
|
||||
capturedArgs.ctx = ctx
|
||||
capturedArgs.table = table
|
||||
capturedArgs.record = record
|
||||
return fmt.Errorf("fake-error")
|
||||
},
|
||||
}
|
||||
err := mock.Update(ctx, UsersTable, &User{
|
||||
ID: 4242,
|
||||
Name: "fake-name",
|
||||
Age: 42,
|
||||
})
|
||||
|
||||
tt.AssertErrContains(t, err, "fake-error")
|
||||
tt.AssertEqual(t, capturedArgs.ctx, ctx)
|
||||
tt.AssertEqual(t, capturedArgs.table, UsersTable)
|
||||
tt.AssertEqual(t, capturedArgs.record, &User{
|
||||
ID: 4242,
|
||||
Name: "fake-name",
|
||||
Age: 42,
|
||||
})
|
||||
})
|
||||
|
||||
t.Run("Delete", func(t *testing.T) {
|
||||
ctx := context.Background()
|
||||
var capturedArgs struct {
|
||||
ctx context.Context
|
||||
table ksql.Table
|
||||
record interface{}
|
||||
}
|
||||
mock := ksql.Mock{
|
||||
DeleteFn: func(ctx context.Context, table ksql.Table, record interface{}) error {
|
||||
capturedArgs.ctx = ctx
|
||||
capturedArgs.table = table
|
||||
capturedArgs.record = record
|
||||
return fmt.Errorf("fake-error")
|
||||
},
|
||||
}
|
||||
err := mock.Delete(ctx, UsersTable, &User{
|
||||
ID: 4242,
|
||||
Name: "fake-name",
|
||||
Age: 42,
|
||||
})
|
||||
|
||||
tt.AssertErrContains(t, err, "fake-error")
|
||||
tt.AssertEqual(t, capturedArgs.ctx, ctx)
|
||||
tt.AssertEqual(t, capturedArgs.table, UsersTable)
|
||||
tt.AssertEqual(t, capturedArgs.record, &User{
|
||||
ID: 4242,
|
||||
Name: "fake-name",
|
||||
Age: 42,
|
||||
})
|
||||
})
|
||||
|
||||
t.Run("Query", func(t *testing.T) {
|
||||
ctx := context.Background()
|
||||
var capturedArgs struct {
|
||||
ctx context.Context
|
||||
records interface{}
|
||||
query string
|
||||
params []interface{}
|
||||
}
|
||||
mock := ksql.Mock{
|
||||
QueryFn: func(ctx context.Context, records interface{}, query string, params ...interface{}) error {
|
||||
capturedArgs.ctx = ctx
|
||||
capturedArgs.records = records
|
||||
capturedArgs.query = query
|
||||
capturedArgs.params = params
|
||||
return fmt.Errorf("fake-error")
|
||||
},
|
||||
}
|
||||
var users []User
|
||||
err := mock.Query(ctx, &users, "SELECT * FROM user WHERE age = ?", 42)
|
||||
|
||||
tt.AssertErrContains(t, err, "fake-error")
|
||||
tt.AssertEqual(t, capturedArgs.ctx, ctx)
|
||||
tt.AssertEqual(t, capturedArgs.records, &users)
|
||||
tt.AssertEqual(t, capturedArgs.query, "SELECT * FROM user WHERE age = ?")
|
||||
tt.AssertEqual(t, capturedArgs.params, []interface{}{42})
|
||||
})
|
||||
|
||||
t.Run("QueryOne", func(t *testing.T) {
|
||||
ctx := context.Background()
|
||||
var capturedArgs struct {
|
||||
ctx context.Context
|
||||
record interface{}
|
||||
query string
|
||||
params []interface{}
|
||||
}
|
||||
mock := ksql.Mock{
|
||||
QueryOneFn: func(ctx context.Context, record interface{}, query string, params ...interface{}) error {
|
||||
capturedArgs.ctx = ctx
|
||||
capturedArgs.record = record
|
||||
capturedArgs.query = query
|
||||
capturedArgs.params = params
|
||||
return fmt.Errorf("fake-error")
|
||||
},
|
||||
}
|
||||
|
||||
var user User
|
||||
err := mock.QueryOne(ctx, &user, "SELECT * FROM user WHERE id = ?", 4242)
|
||||
|
||||
tt.AssertErrContains(t, err, "fake-error")
|
||||
tt.AssertEqual(t, capturedArgs.ctx, ctx)
|
||||
tt.AssertEqual(t, capturedArgs.record, &user)
|
||||
tt.AssertEqual(t, capturedArgs.query, "SELECT * FROM user WHERE id = ?")
|
||||
tt.AssertEqual(t, capturedArgs.params, []interface{}{4242})
|
||||
})
|
||||
|
||||
t.Run("QueryChunks", func(t *testing.T) {
|
||||
ctx := context.Background()
|
||||
var capturedArgs struct {
|
||||
ctx context.Context
|
||||
parser ksql.ChunkParser
|
||||
}
|
||||
mock := ksql.Mock{
|
||||
QueryChunksFn: func(ctx context.Context, parser ksql.ChunkParser) error {
|
||||
capturedArgs.ctx = ctx
|
||||
capturedArgs.parser = parser
|
||||
return fmt.Errorf("fake-error")
|
||||
},
|
||||
}
|
||||
|
||||
var users []User
|
||||
chunkParserFunc := func(chunk []User) error {
|
||||
users = append(users, chunk...)
|
||||
return nil
|
||||
}
|
||||
err := mock.QueryChunks(ctx, ksql.ChunkParser{
|
||||
Query: "SELECT * FROM users WHERE age = ?",
|
||||
Params: []interface{}{
|
||||
4242,
|
||||
},
|
||||
ChunkSize: 10,
|
||||
ForEachChunk: chunkParserFunc,
|
||||
})
|
||||
|
||||
tt.AssertErrContains(t, err, "fake-error")
|
||||
tt.AssertEqual(t, capturedArgs.ctx, ctx)
|
||||
tt.AssertEqual(t, capturedArgs.parser.Query, "SELECT * FROM users WHERE age = ?")
|
||||
tt.AssertEqual(t, capturedArgs.parser.Params, []interface{}{
|
||||
4242,
|
||||
})
|
||||
tt.AssertEqual(t, capturedArgs.parser.ChunkSize, 10)
|
||||
})
|
||||
|
||||
t.Run("Exec", func(t *testing.T) {
|
||||
ctx := context.Background()
|
||||
var capturedArgs struct {
|
||||
ctx context.Context
|
||||
query string
|
||||
params []interface{}
|
||||
}
|
||||
mock := ksql.Mock{
|
||||
ExecFn: func(ctx context.Context, query string, params ...interface{}) (rowsAffected int64, _ error) {
|
||||
capturedArgs.ctx = ctx
|
||||
capturedArgs.query = query
|
||||
capturedArgs.params = params
|
||||
return 42, fmt.Errorf("fake-error")
|
||||
},
|
||||
}
|
||||
rowsAffected, err := mock.Exec(ctx, "INSERT INTO users_permissions(user_id, permission_id) VALUES (?, ?)", 4242, 4)
|
||||
|
||||
tt.AssertErrContains(t, err, "fake-error")
|
||||
tt.AssertEqual(t, rowsAffected, int64(42))
|
||||
tt.AssertEqual(t, capturedArgs.ctx, ctx)
|
||||
tt.AssertEqual(t, capturedArgs.query, "INSERT INTO users_permissions(user_id, permission_id) VALUES (?, ?)")
|
||||
tt.AssertEqual(t, capturedArgs.params, []interface{}{4242, 4})
|
||||
})
|
||||
|
||||
t.Run("Transaction", func(t *testing.T) {
|
||||
ctx := context.Background()
|
||||
executingMockedTransaction := false
|
||||
mock := ksql.Mock{
|
||||
TransactionFn: func(ctx context.Context, fn func(db ksql.Provider) error) error {
|
||||
executingMockedTransaction = true
|
||||
return nil
|
||||
},
|
||||
}
|
||||
|
||||
executed := false
|
||||
mock.Transaction(ctx, func(db ksql.Provider) error {
|
||||
executed = true
|
||||
return nil
|
||||
})
|
||||
|
||||
tt.AssertEqual(t, executingMockedTransaction, true)
|
||||
tt.AssertEqual(t, executed, false)
|
||||
})
|
||||
})
|
||||
}
|
||||
|
|
|
|||
Loading…
Reference in New Issue