csvreader/reader_test.go at master · zhnxin/csvreader · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
package csvreader_test

import (
	"encoding/json"
	"fmt"
	"testing"

	"github.com/zhnxin/csvreader"
)

type testStruct struct {
	Name     string
	UserName string
	ID       int
	Enable   bool
	Type     CustomeType
}

type CustomeType int

func (c *CustomeType) FromString(str string) error {
	switch str {
	case "tcp":
		*c = 0
	case "udp":
		*c = 1
	default:
		return fmt.Errorf("unknown type:%s", str)
	}
	return nil
}

func TestSnakeName(t *testing.T) {
	bean := []testStruct{}
	if err := csvreader.New().
		WithHeader([]string{"name", "user_name", "id", "enable"}).
		UnMarshalBytes([]byte("zhengxin,zhnxin,0,false\nxinzheng,zhnxin,1,true"),
			&bean); err != nil {
		t.Fatal(err)
	}
	b, _ := json.Marshal(bean)
	t.Log(string(b))
}

func TestLowerName(t *testing.T) {
	bean := []*testStruct{}
	if err := csvreader.New().
		WithHeader([]string{"NAME", "USERNAME", "ID", "ENABLE"}).
		UnMarshalBytes([]byte("zhengxin,zhnxin,0,false\nxinzheng,zhnxin,1,true"),
			&bean); err != nil {
		t.Fatal(err)
	}
	b, _ := json.Marshal(bean)
	t.Log(string(b))
}

func TestCustom(t *testing.T) {
	bean := []*testStruct{}
	if err := csvreader.New().
		WithHeader([]string{"NAME", "USERNAME", "type", "ENABLE"}).
		UnMarshalBytes([]byte("zhengxin,zhnxin,udp,false\nxinzheng,zhnxin,tcp,true"),
			&bean); err != nil {
		t.Fatal(err)
	}
	b, _ := json.Marshal(bean)
	t.Log(string(b))
}