forked from zabawaba99/firego
-
Notifications
You must be signed in to change notification settings - Fork 0
/
firebase_test.go
145 lines (119 loc) · 3.26 KB
/
firebase_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
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
package firego
import (
"fmt"
"net/http"
"net/http/httptest"
"testing"
"time"
"github.com/stretchr/testify/assert"
"github.com/stretchr/testify/require"
)
const URL = "https://somefirebaseapp.firebaseIO.com"
type TestServer struct {
*httptest.Server
receivedReqs []*http.Request
}
func newTestServer(response string) *TestServer {
ts := &TestServer{}
ts.Server = httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, req *http.Request) {
ts.receivedReqs = append(ts.receivedReqs, req)
fmt.Fprint(w, response)
}))
return ts
}
func TestNew(t *testing.T) {
t.Parallel()
testCases := []struct {
givenURL string
}{
{
URL,
},
{
URL + "/",
},
{
"somefirebaseapp.firebaseIO.com",
},
{
"somefirebaseapp.firebaseIO.com/",
},
}
for _, tt := range testCases {
fb := New(tt.givenURL)
assert.Equal(t, URL, fb.url, "givenURL: %s", tt.givenURL)
}
}
func TestChild(t *testing.T) {
t.Parallel()
var (
parent = New(URL)
childNode = "node"
child = parent.Child(childNode)
)
assert.Equal(t, fmt.Sprintf("%s/%s", parent.url, childNode), child.url)
}
func TestTimeoutDuration_Headers(t *testing.T) {
defer func(dur time.Duration) { TimeoutDuration = dur }(TimeoutDuration)
TimeoutDuration = time.Millisecond
server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, req *http.Request) {
time.Sleep(2 * TimeoutDuration)
}))
defer server.Close()
fb := New(server.URL)
err := fb.Value("")
assert.NotNil(t, err)
assert.IsType(t, ErrTimeout{}, err)
// ResponseHeaderTimeout should be TimeoutDuration less the time it took to dial, and should be positive
require.IsType(t, (*http.Transport)(nil), fb.client.Transport)
tr := fb.client.Transport.(*http.Transport)
assert.True(t, tr.ResponseHeaderTimeout < TimeoutDuration)
assert.True(t, tr.ResponseHeaderTimeout > 0)
}
func TestTimeoutDuration_Dial(t *testing.T) {
defer func(dur time.Duration) { TimeoutDuration = dur }(TimeoutDuration)
TimeoutDuration = time.Microsecond
fb := New("http://dialtimeouterr.or/")
err := fb.Value("")
assert.NotNil(t, err)
assert.IsType(t, ErrTimeout{}, err)
// ResponseHeaderTimeout should be negative since the total duration was consumed when dialing
require.IsType(t, (*http.Transport)(nil), fb.client.Transport)
assert.True(t, fb.client.Transport.(*http.Transport).ResponseHeaderTimeout < 0)
}
func TestShallow(t *testing.T) {
t.Parallel()
var (
server = newTestServer("")
fb = New(server.URL)
)
defer server.Close()
fb.Shallow(true)
fb.Value("")
require.Len(t, server.receivedReqs, 1)
req := server.receivedReqs[0]
assert.Equal(t, shallowParam+"=true", req.URL.Query().Encode())
fb.Shallow(false)
fb.Value("")
require.Len(t, server.receivedReqs, 2)
req = server.receivedReqs[1]
assert.Equal(t, "", req.URL.Query().Encode())
}
func TestIncludePriority(t *testing.T) {
t.Parallel()
var (
server = newTestServer("")
fb = New(server.URL)
)
defer server.Close()
fb.IncludePriority(true)
fb.Value("")
require.Len(t, server.receivedReqs, 1)
req := server.receivedReqs[0]
assert.Equal(t, formatParam+"="+formatVal, req.URL.Query().Encode())
fb.IncludePriority(false)
fb.Value("")
require.Len(t, server.receivedReqs, 2)
req = server.receivedReqs[1]
assert.Equal(t, "", req.URL.Query().Encode())
}