gomapper/auto_test.go at main · Insei/gomapper · 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
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
package gomapper

import (
	"testing"
	"time"

	"github.com/google/uuid"
	"github.com/stretchr/testify/assert"
)

type AutoMappingStructSource struct {
	Age          int
	Name         string
	Time         time.Time
	UUID         uuid.UUID
	PtrTime      *time.Time
	PtrUUID      *uuid.UUID
	NestedStruct NestedStructSource
}

type AutoMappingStructDest struct {
	UUID         uuid.UUID
	Age          int
	Name         string
	SecondName   string
	Time         time.Time
	PtrTime      *time.Time
	PtrUUID      *uuid.UUID
	NestedStruct NestedStructDest
}

type NestedStructSource struct {
	FirstNestedName  string
	DeepNestedStruct DeepNestedStructSource
}

type NestedStructDest struct {
	FirstNestedName       string
	FirstNestedSecondName string
	DeepNestedStruct      DeepNestedStructDest
}

type DeepNestedStructSource struct {
	SecondNestedName string
}

type DeepNestedStructDest struct {
	SecondNestedName string
}

func TestAutoRoute(t *testing.T) {
	_ = AutoRoute[AutoMappingStructSource, AutoMappingStructDest]()
	_ = AutoRoute[TestingStructSource, TestingStructDest]()
	t.Run("Auto route without options", func(t *testing.T) {
		ptrTime := time.Now()
		ptrUuid := uuid.New()
		source := &AutoMappingStructSource{Name: "Test1", Time: time.Now(), UUID: uuid.New(), PtrTime: &ptrTime, PtrUUID: &ptrUuid}
		dest, err := MapTo[AutoMappingStructDest](source)
		assert.NoError(t, err)
		assert.Equal(t, source.Name, dest.Name)
		assert.Equal(t, source.Time, dest.Time)
		assert.Equal(t, source.UUID, dest.UUID)
		assert.Equal(t, source.PtrUUID, dest.PtrUUID)
		assert.Equal(t, source.PtrTime, dest.PtrTime)
	})
	timeNow := time.Now()
	_ = AutoRoute[AutoMappingStructSource, AutoMappingStructDest](WithFunc(func(source AutoMappingStructSource, dest *AutoMappingStructDest) {
		if source.Name == "Test1" {
			dest.SecondName = "Test2"
		}
		if source.PtrTime == nil {
			dest.PtrTime = &timeNow
		}
		dest.Time = timeNow
	}),
		WithFieldSkip(func(dest *AutoMappingStructSource) any {
			return &dest.UUID
		}),
		WithFieldSkip(func(dest *AutoMappingStructSource) any {
			return &dest.Age
		}),
	)
	t.Run("Auto route with options", func(t *testing.T) {
		source := &AutoMappingStructSource{UUID: uuid.New(), Name: "Test1", Age: 25}
		dest, err := MapTo[AutoMappingStructDest](source)
		assert.NoError(t, err)
		assert.Equal(t, source.Name, dest.Name)
		assert.Equal(t, "Test2", dest.SecondName)
		assert.Equal(t, timeNow, *dest.PtrTime)
		assert.Equal(t, timeNow, dest.Time)
		assert.Equal(t, 0, dest.Age)
		assert.Equal(t, uuid.Nil, dest.UUID)
	})
	t.Run("Auto mapping struct fields", func(t *testing.T) {
		source := &AutoMappingStructSource{
			NestedStruct: NestedStructSource{
				FirstNestedName: "Test1",
				DeepNestedStruct: DeepNestedStructSource{
					SecondNestedName: "Test2",
				},
			},
		}
		dest, err := MapTo[AutoMappingStructDest](source)
		assert.NoError(t, err)
		assert.Equal(t, source.NestedStruct.FirstNestedName, dest.NestedStruct.FirstNestedName)
		assert.Equal(t, source.NestedStruct.DeepNestedStruct.SecondNestedName, dest.NestedStruct.DeepNestedStruct.SecondNestedName)
	})

	_ = AddRoute[NestedStructSource, NestedStructDest](func(source NestedStructSource, dest *NestedStructDest) error {
		dest.FirstNestedSecondName = source.FirstNestedName
		return nil
	})
	t.Run("Auto mapping using existing route on nested struct", func(t *testing.T) {
		source := &AutoMappingStructSource{
			NestedStruct: NestedStructSource{
				FirstNestedName: "Test1",
			},
		}
		dest, err := MapTo[AutoMappingStructDest](source)
		assert.NoError(t, err)
		assert.Equal(t, source.NestedStruct.FirstNestedName, dest.NestedStruct.FirstNestedName)
	})
}