1 package pgtype_test
2
3 import (
4 "testing"
5 "time"
6
7 "github.com/jackc/pgtype"
8 "github.com/jackc/pgtype/testutil"
9 )
10
11 func TestDaterangeTranscode(t *testing.T) {
12 testutil.TestSuccessfulTranscodeEqFunc(t, "daterange", []interface{}{
13 &pgtype.Daterange{LowerType: pgtype.Empty, UpperType: pgtype.Empty, Status: pgtype.Present},
14 &pgtype.Daterange{
15 Lower: pgtype.Date{Time: time.Date(1990, 12, 31, 0, 0, 0, 0, time.UTC), Status: pgtype.Present},
16 Upper: pgtype.Date{Time: time.Date(2028, 1, 1, 0, 0, 0, 0, time.UTC), Status: pgtype.Present},
17 LowerType: pgtype.Inclusive,
18 UpperType: pgtype.Exclusive,
19 Status: pgtype.Present,
20 },
21 &pgtype.Daterange{
22 Lower: pgtype.Date{Time: time.Date(1800, 12, 31, 0, 0, 0, 0, time.UTC), Status: pgtype.Present},
23 Upper: pgtype.Date{Time: time.Date(2200, 1, 1, 0, 0, 0, 0, time.UTC), Status: pgtype.Present},
24 LowerType: pgtype.Inclusive,
25 UpperType: pgtype.Exclusive,
26 Status: pgtype.Present,
27 },
28 &pgtype.Daterange{Status: pgtype.Null},
29 }, func(aa, bb interface{}) bool {
30 a := aa.(pgtype.Daterange)
31 b := bb.(pgtype.Daterange)
32
33 return a.Status == b.Status &&
34 a.Lower.Time.Equal(b.Lower.Time) &&
35 a.Lower.Status == b.Lower.Status &&
36 a.Lower.InfinityModifier == b.Lower.InfinityModifier &&
37 a.Upper.Time.Equal(b.Upper.Time) &&
38 a.Upper.Status == b.Upper.Status &&
39 a.Upper.InfinityModifier == b.Upper.InfinityModifier
40 })
41 }
42
43 func TestDaterangeNormalize(t *testing.T) {
44 testutil.TestSuccessfulNormalizeEqFunc(t, []testutil.NormalizeTest{
45 {
46 SQL: "select daterange('2010-01-01', '2010-01-11', '(]')",
47 Value: pgtype.Daterange{
48 Lower: pgtype.Date{Time: time.Date(2010, 1, 2, 0, 0, 0, 0, time.UTC), Status: pgtype.Present},
49 Upper: pgtype.Date{Time: time.Date(2010, 1, 12, 0, 0, 0, 0, time.UTC), Status: pgtype.Present},
50 LowerType: pgtype.Inclusive,
51 UpperType: pgtype.Exclusive,
52 Status: pgtype.Present,
53 },
54 },
55 }, func(aa, bb interface{}) bool {
56 a := aa.(pgtype.Daterange)
57 b := bb.(pgtype.Daterange)
58
59 return a.Status == b.Status &&
60 a.Lower.Time.Equal(b.Lower.Time) &&
61 a.Lower.Status == b.Lower.Status &&
62 a.Lower.InfinityModifier == b.Lower.InfinityModifier &&
63 a.Upper.Time.Equal(b.Upper.Time) &&
64 a.Upper.Status == b.Upper.Status &&
65 a.Upper.InfinityModifier == b.Upper.InfinityModifier
66 })
67 }
68
69 func TestDaterangeSet(t *testing.T) {
70 successfulTests := []struct {
71 source interface{}
72 result pgtype.Daterange
73 }{
74 {
75 source: nil,
76 result: pgtype.Daterange{Status: pgtype.Null},
77 },
78 {
79 source: &pgtype.Daterange{
80 Lower: pgtype.Date{Time: time.Date(1990, 12, 31, 0, 0, 0, 0, time.UTC), Status: pgtype.Present},
81 Upper: pgtype.Date{Time: time.Date(2028, 1, 1, 0, 0, 0, 0, time.UTC), Status: pgtype.Present},
82 LowerType: pgtype.Inclusive,
83 UpperType: pgtype.Exclusive,
84 Status: pgtype.Present,
85 },
86 result: pgtype.Daterange{
87 Lower: pgtype.Date{Time: time.Date(1990, 12, 31, 0, 0, 0, 0, time.UTC), Status: pgtype.Present},
88 Upper: pgtype.Date{Time: time.Date(2028, 1, 1, 0, 0, 0, 0, time.UTC), Status: pgtype.Present},
89 LowerType: pgtype.Inclusive,
90 UpperType: pgtype.Exclusive,
91 Status: pgtype.Present,
92 },
93 },
94 {
95 source: pgtype.Daterange{
96 Lower: pgtype.Date{Time: time.Date(1990, 12, 31, 0, 0, 0, 0, time.UTC), Status: pgtype.Present},
97 Upper: pgtype.Date{Time: time.Date(2028, 1, 1, 0, 0, 0, 0, time.UTC), Status: pgtype.Present},
98 LowerType: pgtype.Inclusive,
99 UpperType: pgtype.Exclusive,
100 Status: pgtype.Present,
101 },
102 result: pgtype.Daterange{
103 Lower: pgtype.Date{Time: time.Date(1990, 12, 31, 0, 0, 0, 0, time.UTC), Status: pgtype.Present},
104 Upper: pgtype.Date{Time: time.Date(2028, 1, 1, 0, 0, 0, 0, time.UTC), Status: pgtype.Present},
105 LowerType: pgtype.Inclusive,
106 UpperType: pgtype.Exclusive,
107 Status: pgtype.Present,
108 },
109 },
110 {
111 source: "[1990-12-31,2028-01-01)",
112 result: pgtype.Daterange{
113 Lower: pgtype.Date{Time: time.Date(1990, 12, 31, 0, 0, 0, 0, time.UTC), Status: pgtype.Present},
114 Upper: pgtype.Date{Time: time.Date(2028, 1, 1, 0, 0, 0, 0, time.UTC), Status: pgtype.Present},
115 LowerType: pgtype.Inclusive,
116 UpperType: pgtype.Exclusive,
117 Status: pgtype.Present,
118 },
119 },
120 }
121
122 for i, tt := range successfulTests {
123 var r pgtype.Daterange
124 err := r.Set(tt.source)
125 if err != nil {
126 t.Errorf("%d: %v", i, err)
127 }
128
129 if r != tt.result {
130 t.Errorf("%d: expected %v to convert to %v, but it was %v", i, tt.source, tt.result, r)
131 }
132 }
133 }
134
View as plain text