pgkit/dbtype/timerange_test.go at 89270fd8c712006480f18cbad256cf65d1b396ee · goware/pgkit · 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
package dbtype_test

import (
	"testing"
	"time"

	"github.com/goware/pgkit/v2/dbtype"
	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/require"
)

func TestTimeRange_UUIDv7_BothBounds(t *testing.T) {
	since := time.Date(2026, 1, 1, 0, 0, 0, 0, time.UTC)
	until := time.Date(2026, 3, 27, 0, 0, 0, 0, time.UTC)
	r := dbtype.UUIDv7Range("id", &since, &until)

	sql, args, err := r.ToSql()
	require.NoError(t, err)
	assert.Equal(t, "id BETWEEN ? AND ?", sql)
	assert.Len(t, args, 2)
}

func TestTimeRange_UUIDv7_SinceOnly(t *testing.T) {
	since := time.Date(2026, 1, 1, 0, 0, 0, 0, time.UTC)
	r := dbtype.UUIDv7Range("id", &since, nil)

	sql, args, err := r.ToSql()
	require.NoError(t, err)
	assert.Equal(t, "id >= ?", sql)
	assert.Len(t, args, 1)
}

func TestTimeRange_UUIDv7_UntilOnly(t *testing.T) {
	until := time.Date(2026, 3, 27, 0, 0, 0, 0, time.UTC)
	r := dbtype.UUIDv7Range("id", nil, &until)

	sql, args, err := r.ToSql()
	require.NoError(t, err)
	assert.Equal(t, "id <= ?", sql)
	assert.Len(t, args, 1)
}

func TestTimeRange_UUIDv7_NeitherBound(t *testing.T) {
	r := dbtype.UUIDv7Range("id", nil, nil)

	sql, args, err := r.ToSql()
	require.NoError(t, err)
	assert.Equal(t, "1=1", sql)
	assert.Empty(t, args)
}

func TestTimeRange_Int64ID_BothBounds(t *testing.T) {
	since := time.Date(2026, 1, 1, 0, 0, 0, 0, time.UTC)
	until := time.Date(2026, 3, 27, 0, 0, 0, 0, time.UTC)
	r := dbtype.Int64IDRange("id", &since, &until)

	sql, args, err := r.ToSql()
	require.NoError(t, err)
	assert.Equal(t, "id BETWEEN ? AND ?", sql)
	assert.Len(t, args, 2)
}

func TestTimeRange_FloorCeil_Accessors(t *testing.T) {
	since := time.Date(2026, 1, 1, 0, 0, 0, 0, time.UTC)
	r := dbtype.UUIDv7Range("id", &since, nil)

	_, ok := r.Floor()
	assert.True(t, ok)
	_, ok = r.Ceil()
	assert.False(t, ok)
}

func TestTimeRange_UUIDv7_BoundValues(t *testing.T) {
	since := time.Date(2026, 1, 1, 0, 0, 0, 0, time.UTC)
	until := time.Date(2026, 3, 27, 0, 0, 0, 0, time.UTC)
	r := dbtype.UUIDv7Range("id", &since, &until)

	_, args, err := r.ToSql()
	require.NoError(t, err)

	assert.Equal(t, dbtype.FloorUUIDv7(since), args[0])
	assert.Equal(t, dbtype.CeilUUIDv7(until), args[1])
}

func TestTimeRange_Int64ID_BoundValues(t *testing.T) {
	since := time.Date(2026, 1, 1, 0, 0, 0, 0, time.UTC)
	until := time.Date(2026, 3, 27, 0, 0, 0, 0, time.UTC)
	r := dbtype.Int64IDRange("id", &since, &until)

	_, args, err := r.ToSql()
	require.NoError(t, err)

	assert.Equal(t, dbtype.FloorInt64ID(since), args[0])
	assert.Equal(t, dbtype.CeilInt64ID(until), args[1])
}