move to v4

Signed-off-by: Vasiliy Tolstov <v.tolstov@unistack.org>

util/test/sqlmock_test.go

@@ -0,0 +1,25 @@
+package test
+
+import (
+	"fmt"
+	"strings"
+	"testing"
+
+	"github.com/DATA-DOG/go-sqlmock"
+)
+
+func Test_NewSQLRowsFromFile(t *testing.T) {
+	db, c, err := sqlmock.New()
+	if err != nil {
+		t.Fatal(err)
+	}
+	defer db.Close()
+
+	rows, err := NewSQLRowsFromFile(c, "testdata/Call.csv")
+	if err != nil {
+		t.Fatal(err)
+	}
+	if !strings.Contains(fmt.Sprintf("%#+v", rows), `cols:[]string{"DepAgrId", "DepAgrNum", "DepAgrDate", "DepAgrCloseDate", "AccCur", "MainFinaccNum", "MainFinaccName", "MainFinaccId", "MainFinaccOpenDt", "DepAgrStatus", "MainFinaccBal", "DepartCode", "CardAccId"}`) {
+		t.Fatal("invalid cols after import csv")
+	}
+}

util/test/test.go

@@ -0,0 +1,81 @@
+package test
+
+import (
+	"encoding/csv"
+	"errors"
+	"os"
+	"path"
+	"strings"
+
+	"github.com/DATA-DOG/go-sqlmock"
+	"go.unistack.org/micro/v4/client"
+	"go.unistack.org/micro/v4/codec"
+)
+
+var ErrUnknownContentType = errors.New("unknown content type")
+
+type Extension struct {
+	Ext []string
+}
+
+var ExtToTypes = map[string][]string{
+	"json":  {"application/json", "application/grpc+json"},
+	"yaml":  {"application/yaml", "application/yml", "text/yaml", "text/yml"},
+	"yml":   {"application/yaml", "application/yml", "text/yaml", "text/yml"},
+	"proto": {"application/grpc", "application/grpc+proto", "application/proto"},
+}
+
+func NewRequestFromFile(c client.Client, reqfile string) (client.Request, error) {
+	reqbuf, err := os.ReadFile(reqfile)
+	if err != nil {
+		return nil, err
+	}
+
+	endpoint := path.Base(path.Dir(reqfile))
+	ext := path.Ext(reqfile)
+	if len(ext) > 0 && ext[0] == '.' {
+		ext = ext[1:]
+	}
+
+	var ct string
+	if cts, ok := ExtToTypes[ext]; ok {
+		for _, t := range cts {
+			if _, ok = c.Options().Codecs[t]; ok {
+				ct = t
+				break
+			}
+		}
+	}
+
+	if ct == "" {
+		return nil, ErrUnknownContentType
+	}
+
+	req := c.NewRequest("test", endpoint, &codec.Frame{Data: reqbuf}, client.RequestContentType(ct))
+
+	return req, nil
+}
+
+func NewSQLRowsFromFile(c sqlmock.Sqlmock, file string) (*sqlmock.Rows, error) {
+	fp, err := os.Open(file)
+	if err != nil {
+		return nil, err
+	}
+	defer fp.Close()
+
+	r := csv.NewReader(fp)
+	r.Comma = '\t'
+	r.Comment = '#'
+
+	records, err := r.ReadAll()
+	if err != nil {
+		return nil, err
+	}
+
+	rows := c.NewRows(records[0])
+	for idx := 1; idx < len(records); idx++ {
+		rows.FromCSVString(strings.Join(records[idx], ";"))
+	}
+
+	return rows, nil
+}