补充 struct 传参时 time 类型的单元测试。

Author: bunnier

abstract_db_client_test.go

@@ -6,24 +6,6 @@ import (
 	"time"
 )
 
-type StrA struct {
-	A string
-}
-
-type StrB struct {
-	B string
-}
-
-type StrAB struct {
-	A string
-	B string
-}
-
-type StrABC struct {
-	StrAB
-	C string
-}
-
 func Test_preHandleArgs(t *testing.T) {
 	type TypeA struct {
 		A string
@@ -43,6 +25,11 @@ func Test_preHandleArgs(t *testing.T) {
 		C string
 	}
 
+	type TypeTimeA struct {
+		TypeA
+		Time time.Time
+	}
+
 	testTime := time.Date(2021, 7, 3, 0, 0, 0, 0, time.UTC)
 
 	t.Run("empty_args", func(t *testing.T) {
@@ -155,6 +142,22 @@ func Test_preHandleArgs(t *testing.T) {
 		}
 	})
 
+	t.Run("time_convert", func(t *testing.T) {
+		testTime := time.Now()
+		paramTimeA := TypeTimeA{
+			TypeA: TypeA{A: "abc_a"},
+			Time:  testTime,
+		}
+		got, err := preHandleArgs(paramTimeA)
+		if err != nil {
+			t.Errorf("mergeArgs() error = %v, wantErr false", err)
+			return
+		}
+		if !reflect.DeepEqual(got, []any{map[string]any{"A": "abc_a", "Time": testTime}}) {
+			t.Errorf("mergeArgs() = %v, want %v", got, []any{map[string]any{"A": "abc_a", "Time": testTime}})
+		}
+	})
+
 	t.Run("mixed_types", func(t *testing.T) {
 		args := []any{
 			TypeAB{TypeA: TypeA{A: "ab_a"}, TypeB: TypeB{B: "ab_b"}},