-
Notifications
You must be signed in to change notification settings - Fork 1
/
payload_test.go
67 lines (59 loc) · 1.31 KB
/
payload_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
package apns
import (
"encoding/json"
"testing"
)
func TestBuildPayload(t *testing.T) {
type Payload struct {
APS json.RawMessage
Custom json.RawMessage
}
var p Payload
n := &Notification{
Payload: BuildPayload(nil, map[string]interface{}{
"aps": map[string]interface{}{
"alert": "Hello, Custom!",
},
"custom": map[string]interface{}{
"hello": "Go",
},
}),
}
bytes, err := json.Marshal(n)
if err != nil {
t.Fatal(err)
}
err = json.Unmarshal(bytes, &p)
if err != nil {
t.Fatal(err)
}
if string(p.APS) != `{"alert":"Hello, Custom!"}` {
t.Errorf("APS: %v; want: %v", string(p.APS), `{"alert":"Hello, Custom!"}`)
}
if string(p.Custom) != `{"hello":"Go"}` {
t.Errorf("Custom: %v; want: %v", string(p.Custom), `{"hello":"Go"}`)
}
n = &Notification{
Payload: BuildPayload(&APS{
Alert: "Hello, APS!",
}, map[string]interface{}{
"custom": map[string]interface{}{
"hello": "Go",
},
}),
}
bytes, err = json.Marshal(n)
if err != nil {
t.Fatal(err)
}
err = json.Unmarshal(bytes, &p)
if err != nil {
t.Fatal(err)
}
if string(p.APS) != `{"alert":"Hello, APS!"}` {
t.Errorf("APS: %v; want: %v", string(p.APS), `{"alert":"Hello, APS!"}`)
}
if string(p.Custom) != `{"hello":"Go"}` {
t.Errorf("Custom: %v; want: %v", string(p.Custom), `{"hello":"Go"}`)
}
}