-
-
Notifications
You must be signed in to change notification settings - Fork 242
/
model_context_test.go
127 lines (100 loc) · 3.61 KB
/
model_context_test.go
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
124
125
126
127
package pop
import (
"context"
"strings"
"testing"
"time"
"github.com/stretchr/testify/require"
)
type ContextTable struct {
ID string `db:"id"`
Value string `db:"value"`
CreatedAt time.Time `db:"created_at"`
UpdatedAt time.Time `db:"updated_at"`
}
func (t ContextTable) TableName(ctx context.Context) string {
// This is singular on purpose! It will check if the TableName is properly
// Respected in slices as well.
prefix := ctx.Value("prefix").(string)
// PostgreSQL and CockroachDB support schemas which work like a prefix. For those cases, we use
// the schema to ensure that name normalization does not cause query problems.
//
// Since this works only for those two databases, we use underscore for the rest.
//
// While this schema is hardcoded, it would have been too difficult to add this special
// case to the migrations.
switch PDB.Dialect.Name() {
case nameCockroach:
fallthrough
case namePostgreSQL:
prefix = prefix + "." + prefix
}
return "context_prefix_" + prefix + "_table"
}
func Test_ModelContext(t *testing.T) {
if PDB == nil {
t.Skip("skipping integration tests")
}
t.Run("contextless", func(t *testing.T) {
r := require.New(t)
r.Panics(func() {
var c ContextTable
r.NoError(PDB.Create(&c))
}, "panics if context prefix is not set")
})
for _, prefix := range []string{"a", "b"} {
t.Run("prefix="+prefix, func(t *testing.T) {
r := require.New(t)
expected := ContextTable{ID: prefix, Value: prefix}
c := PDB.WithContext(context.WithValue(context.Background(), "prefix", prefix))
r.NoError(c.Create(&expected))
var actual ContextTable
r.NoError(c.Find(&actual, expected.ID))
r.EqualValues(prefix, actual.Value)
r.EqualValues(prefix, actual.ID)
exists, err := c.Where("id = ?", actual.ID).Exists(new(ContextTable))
r.NoError(err)
r.True(exists)
count, err := c.Where("id = ?", actual.ID).Count(new(ContextTable))
r.NoError(err)
r.EqualValues(1, count)
expected.Value += expected.Value
r.NoError(c.Update(&expected))
r.NoError(c.Find(&actual, expected.ID))
r.EqualValues(prefix+prefix, actual.Value)
r.EqualValues(prefix, actual.ID)
var results []ContextTable
require.NoError(t, c.All(&results))
require.NoError(t, c.First(&expected))
require.NoError(t, c.Last(&expected))
r.NoError(c.Destroy(&expected))
})
}
t.Run("prefix=unknown", func(t *testing.T) {
r := require.New(t)
c := PDB.WithContext(context.WithValue(context.Background(), "prefix", "unknown"))
err := c.Create(&ContextTable{ID: "unknown"})
r.Error(err)
if !strings.Contains(err.Error(), "context_prefix_unknown") { // All other databases
t.Fatalf("Expected error to contain indicator that table does not exist but got: %s", err.Error())
}
})
t.Run("cache_busting", func(t *testing.T) {
r := require.New(t)
r.NoError(PDB.WithContext(context.WithValue(context.Background(), "prefix", "a")).Destroy(&ContextTable{ID: "expectedA"}))
r.NoError(PDB.WithContext(context.WithValue(context.Background(), "prefix", "b")).Destroy(&ContextTable{ID: "expectedB"}))
var expectedA, expectedB ContextTable
expectedA.ID = "expectedA"
expectedB.ID = "expectedB"
cA := PDB.WithContext(context.WithValue(context.Background(), "prefix", "a"))
r.NoError(cA.Create(&expectedA))
cB := PDB.WithContext(context.WithValue(context.Background(), "prefix", "b"))
r.NoError(cB.Create(&expectedB))
var actualA, actualB []ContextTable
r.NoError(cA.All(&actualA))
r.NoError(cB.All(&actualB))
r.Len(actualA, 1)
r.Len(actualB, 1)
r.NotEqual(actualA[0].ID, actualB[0].ID, "if these are equal context switching did not work")
})
}