diff --git a/internal/udf/protobuf_test.go b/internal/udf/protobuf_test.go index 7a408e5d..2077360b 100644 --- a/internal/udf/protobuf_test.go +++ b/internal/udf/protobuf_test.go @@ -1,13 +1,14 @@ package udf import ( - "github.com/stretchr/testify/assert" - "github.com/viant/assertly" - "github.com/viant/toolbox" "log" "path" "testing" + "github.com/stretchr/testify/assert" + "github.com/viant/assertly" + "github.com/viant/toolbox" + "io/ioutil" ) @@ -42,7 +43,7 @@ func TestProtoCodec_AsBinary(t *testing.T) { for _, useCase := range useCases { - codec, err := NewProtoCodec(useCase.protoFile, useCase.importPath, useCase.messageType) + codec, err := NewProtoCodec(useCase.protoFile, useCase.importPath, useCase.messageType, true) if !assert.Nil(t, err, useCase.description) { log.Fatal(err) } diff --git a/internal/udf/udf.go b/internal/udf/udf.go index 6a90da1c..6479d56b 100644 --- a/internal/udf/udf.go +++ b/internal/udf/udf.go @@ -4,7 +4,6 @@ import ( "bytes" "encoding/base64" "fmt" - "google.golang.org/protobuf/proto" "io" "io/ioutil" "os" @@ -12,6 +11,8 @@ import ( "path/filepath" "strings" + "google.golang.org/protobuf/proto" + "github.com/pkg/errors" "github.com/viant/afs/file" "github.com/viant/afs/option" @@ -106,7 +107,7 @@ func LoadData(source interface{}, state data.Map) (interface{}, error) { return c, nil } - return nil, fmt.Errorf("udf LoadData arguments must be string: v%", source) + return nil, fmt.Errorf("udf LoadData arguments must be string: %v", source) } // URLPath return path from URL