Skip to content

Commit f2b3600

Browse files
author
Chris Stockton
committed
chore: update tests to use require package
This change is to make the unit tests more consistent with the rest of the code base.
1 parent 26e23f0 commit f2b3600

File tree

6 files changed

+206
-352
lines changed

6 files changed

+206
-352
lines changed

‎internal/e2e/e2e_test.go‎

Lines changed: 7 additions & 22 deletions
Original file line numberDiff line numberDiff line change
@@ -34,12 +34,8 @@ func TestUtils(t *testing.T) {
3434
configPath = "abc"
3535

3636
globalCfg, err := Config()
37-
if err == nil {
38-
t.Fatal("exp non-nil err")
39-
}
40-
if globalCfg != nil {
41-
t.Fatal("exp nil conn")
42-
}
37+
require.Error(t, err)
38+
require.Nil(t, globalCfg)
4339
}
4440
}()
4541

@@ -49,9 +45,7 @@ func TestUtils(t *testing.T) {
4945
{
5046
globalCfg := Must(Config())
5147
conn := Must(Conn(globalCfg))
52-
if conn == nil {
53-
t.Fatal("exp non-nil conn")
54-
}
48+
require.NotNil(t, conn)
5549
}
5650

5751
// negative
@@ -60,12 +54,8 @@ func TestUtils(t *testing.T) {
6054
globalCfg.DB.Driver = ""
6155
globalCfg.DB.URL = "invalid"
6256
conn, err := Conn(globalCfg)
63-
if err == nil {
64-
t.Fatal("exp non-nil err")
65-
}
66-
if conn != nil {
67-
t.Fatal("exp nil conn")
68-
}
57+
require.Error(t, err)
58+
require.Nil(t, conn)
6959
}
7060

7161
}()
@@ -89,10 +79,7 @@ func TestUtils(t *testing.T) {
8979
panic(errors.New("globalCfg != nil"))
9080
}
9181
}()
92-
93-
if err == nil {
94-
t.Fatal("exp non-nil err")
95-
}
82+
require.Error(t, err)
9683
}()
9784

9885
// block init from main()
@@ -113,8 +100,6 @@ func TestUtils(t *testing.T) {
113100
}()
114101

115102
exp := "package e2e may not be used in a main package"
116-
if errStr != exp {
117-
t.Fatalf("exp %v; got %v", exp, errStr)
118-
}
103+
require.Equal(t, exp, errStr)
119104
}()
120105
}

‎internal/e2e/e2ehooks/e2ehooks_test.go‎

Lines changed: 21 additions & 51 deletions
Original file line numberDiff line numberDiff line change
@@ -9,44 +9,32 @@ import (
99
"testing/iotest"
1010
"time"
1111

12+
"github.com/stretchr/testify/require"
1213
"github.com/supabase/auth/internal/conf"
1314
"github.com/supabase/auth/internal/hooks/v0hooks"
1415
)
1516

1617
func TestInstance(t *testing.T) {
1718
{
1819
globalCfg, err := conf.LoadGlobal("../../../hack/test.env")
19-
if err != nil {
20-
t.Fatalf("exp nil err; got %v", err)
21-
}
20+
require.NoError(t, err)
21+
2222
globalCfg.DB.Driver = ""
2323
globalCfg.DB.URL = "invalid"
2424

2525
inst, err := New(globalCfg)
26-
if err == nil {
27-
t.Fatal("exp non-nil err")
28-
}
29-
if inst != nil {
30-
t.Fatal("exp nil *Instance")
31-
}
26+
require.Error(t, err)
27+
require.Nil(t, inst)
3228
}
3329

3430
{
3531
globalCfg, err := conf.LoadGlobal("../../../hack/test.env")
36-
if err != nil {
37-
t.Fatalf("exp nil err; got %v", err)
38-
}
32+
require.NoError(t, err)
3933

4034
inst, err := New(globalCfg)
41-
if err != nil {
42-
t.Fatalf("exp nil err; got %v", err)
43-
}
44-
if inst == nil {
45-
t.Fatal("exp non-nil *Instance")
46-
}
47-
if err := inst.Close(); err != nil {
48-
t.Fatalf("exp nil err from Close; got %v", err)
49-
}
35+
require.NoError(t, err)
36+
require.NotNil(t, inst)
37+
require.NoError(t, inst.Close())
5038
}
5139
}
5240

@@ -58,9 +46,7 @@ func TestHook(t *testing.T) {
5846

5947
{
6048
calls := hook.GetCalls()
61-
if exp, got := 0, len(calls); exp != got {
62-
t.Fatalf("exp %v; got %v", exp, got)
63-
}
49+
require.Equal(t, 0, len(calls))
6450

6551
u := "http://localhost"
6652
rdr := strings.NewReader("12345")
@@ -70,18 +56,13 @@ func TestHook(t *testing.T) {
7056
hook.ServeHTTP(res, req)
7157

7258
calls = hook.GetCalls()
73-
if exp, got := 1, len(calls); exp != got {
74-
t.Fatalf("exp %v; got %v", exp, got)
75-
}
59+
require.Equal(t, 1, len(calls))
7660
call := calls[0]
7761

7862
var got int
79-
if err := call.Unmarshal(&got); err != nil {
80-
t.Fatalf("exp nil err; got %v", err)
81-
}
82-
if exp := 12345; exp != got {
83-
t.Fatalf("exp %v; got %v", exp, got)
84-
}
63+
err := call.Unmarshal(&got)
64+
require.NoError(t, err)
65+
require.Equal(t, 12345, got)
8566
}
8667

8768
{
@@ -138,9 +119,7 @@ func TestHookRecorder(t *testing.T) {
138119

139120
{
140121
calls := test.hook.GetCalls()
141-
if exp, got := 0, len(calls); exp != got {
142-
t.Fatalf("exp %v; got %v", exp, got)
143-
}
122+
require.Equal(t, 0, len(calls))
144123
}
145124

146125
u := "http://localhost/hooks/" + string(test.name)
@@ -151,23 +130,16 @@ func TestHookRecorder(t *testing.T) {
151130

152131
{
153132
calls := test.hook.GetCalls()
154-
if exp, got := 1, len(calls); exp != got {
155-
t.Fatalf("exp %v; got %v", exp, got)
156-
}
133+
require.Equal(t, 1, len(calls))
157134
call := calls[0]
158135

159136
test.hook.ClearCalls()
160-
if exp, got := 0, len(test.hook.GetCalls()); exp != got {
161-
t.Fatalf("exp %v; got %v", exp, got)
162-
}
137+
require.Equal(t, 0, len(test.hook.GetCalls()))
163138

164139
var got int
165-
if err := call.Unmarshal(&got); err != nil {
166-
t.Fatalf("exp nil err; got %v", err)
167-
}
168-
if exp := 12345; exp != got {
169-
t.Fatalf("exp %v; got %v", exp, got)
170-
}
140+
err := call.Unmarshal(&got)
141+
require.NoError(t, err)
142+
require.Equal(t, 12345, got)
171143
}
172144
}
173145

@@ -179,8 +151,6 @@ func TestHookRecorder(t *testing.T) {
179151
res := httptest.NewRecorder()
180152
hookRec.ServeHTTP(res, req)
181153

182-
if exp, got := 404, res.Result().StatusCode; exp != got {
183-
t.Fatalf("exp %v; got %v", exp, got)
184-
}
154+
require.Equal(t, 404, res.Result().StatusCode)
185155
}
186156
}

0 commit comments

Comments
 (0)