mirror of https://github.com/VinGarcia/ksql.git
75 lines
1.7 KiB
Go
75 lines
1.7 KiB
Go
package modifiers
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"testing"
|
|
|
|
tt "github.com/vingarcia/ksql/internal/testtools"
|
|
)
|
|
|
|
func TestAttrScanWrapper(t *testing.T) {
|
|
ctx := context.Background()
|
|
|
|
var scanArgs map[string]interface{}
|
|
wrapper := AttrScanWrapper{
|
|
Ctx: ctx,
|
|
AttrPtr: "fakeAttrPtr",
|
|
ScanFn: func(ctx context.Context, opInfo OpInfo, attrPtr interface{}, dbValue interface{}) error {
|
|
scanArgs = map[string]interface{}{
|
|
"opInfo": opInfo,
|
|
"attrPtr": attrPtr,
|
|
"dbValue": dbValue,
|
|
}
|
|
return errors.New("fakeScanErrMsg")
|
|
},
|
|
OpInfo: OpInfo{
|
|
Method: "fakeMethod",
|
|
DriverName: "fakeDriverName",
|
|
},
|
|
}
|
|
|
|
err := wrapper.Scan("fakeDbValue")
|
|
tt.AssertErrContains(t, err, "fakeScanErrMsg")
|
|
tt.AssertEqual(t, scanArgs, map[string]interface{}{
|
|
"opInfo": OpInfo{
|
|
Method: "fakeMethod",
|
|
DriverName: "fakeDriverName",
|
|
},
|
|
"attrPtr": "fakeAttrPtr",
|
|
"dbValue": "fakeDbValue",
|
|
})
|
|
}
|
|
|
|
func TestAttrWrapper(t *testing.T) {
|
|
ctx := context.Background()
|
|
|
|
var valueArgs map[string]interface{}
|
|
wrapper := AttrValueWrapper{
|
|
Ctx: ctx,
|
|
Attr: "fakeAttr",
|
|
ValueFn: func(ctx context.Context, opInfo OpInfo, inputValue interface{}) (outputValue interface{}, _ error) {
|
|
valueArgs = map[string]interface{}{
|
|
"opInfo": opInfo,
|
|
"inputValue": inputValue,
|
|
}
|
|
return "fakeOutputValue", errors.New("fakeValueErrMsg")
|
|
},
|
|
OpInfo: OpInfo{
|
|
Method: "fakeMethod",
|
|
DriverName: "fakeDriverName",
|
|
},
|
|
}
|
|
|
|
value, err := wrapper.Value()
|
|
tt.AssertErrContains(t, err, "fakeValueErrMsg")
|
|
tt.AssertEqual(t, valueArgs, map[string]interface{}{
|
|
"opInfo": OpInfo{
|
|
Method: "fakeMethod",
|
|
DriverName: "fakeDriverName",
|
|
},
|
|
"inputValue": "fakeAttr",
|
|
})
|
|
tt.AssertEqual(t, value, "fakeOutputValue")
|
|
}
|