astra/go-mastodon
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Adjust tests

Browse source
This commit is contained in:
Yann David 2022-12-06 21:03:23 -05:00
parent 57ec6cf997
commit 7e51077714
2 changed files with 21 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

4
streaming_test.go
View file

@ -22,6 +22,8 @@ data: {content: error}
event: update event: update
data: {"content": "foo"} data: {"content": "foo"}
event: update event: update
data: {"text": "foo"}
event: update
data: {"content": "%s"} data: {"content": "%s"}
event: notification event: notification
data: {"type": "mention"} data: {"type": "mention"}
@ -42,6 +44,8 @@ data: 1234567
case *UpdateEvent: case *UpdateEvent:
if event.Status.Content == "foo" { if event.Status.Content == "foo" {
passUpdate = true passUpdate = true
} else if event.Status.Text == "foo" {
passUpdate = true
} else if event.Status.Content == largeContent { } else if event.Status.Content == largeContent {
passUpdateLarge = true passUpdateLarge = true
} else { } else {

24
streaming_ws_test.go
View file

@ -79,6 +79,13 @@ func wsMock(w http.ResponseWriter, r *http.Request) {
return return
} }
err = conn.WriteMessage(websocket.TextMessage,
[]byte(`{"event":"update","payload":"{\"text\":\"foo\"}"}`))
if err != nil {
http.Error(w, http.StatusText(http.StatusInternalServerError), http.StatusInternalServerError)
return
}
err = conn.WriteMessage(websocket.TextMessage, err = conn.WriteMessage(websocket.TextMessage,
[]byte(`{"event":"notification","payload":"{\"id\":123}"}`)) []byte(`{"event":"notification","payload":"{\"id\":123}"}`))
if err != nil { if err != nil {
@ -111,20 +118,20 @@ func wsTest(t *testing.T, q chan Event, cancel func()) {
for e := range q { for e := range q {
events = append(events, e) events = append(events, e)
} }
if len(events) != 6 { if len(events) != 7 {
t.Fatalf("result should be four: %d", len(events)) t.Fatalf("result should be four: %d", len(events))
} }
if events[0].(*UpdateEvent).Status.Content != "foo" { if events[0].(*UpdateEvent).Status.Content != "foo" {
t.Fatalf("want %q but %q", "foo", events[0].(*UpdateEvent).Status.Content) t.Fatalf("want %q but %q", "foo", events[0].(*UpdateEvent).Status.Content)
} }
if events[1].(*NotificationEvent).Notification.ID != "123" { if events[1].(*UpdateEvent).Status.Text != "foo" {
t.Fatalf("want %q but %q", "123", events[1].(*NotificationEvent).Notification.ID) t.Fatalf("want %q but %q", "foo", events[1].(*UpdateEvent).Status.Content)
} }
if events[2].(*DeleteEvent).ID != "1234567" { if events[2].(*NotificationEvent).Notification.ID != "123" {
t.Fatalf("want %q but %q", "1234567", events[2].(*DeleteEvent).ID) t.Fatalf("want %q but %q", "123", events[2].(*NotificationEvent).Notification.ID)
} }
if errorEvent, ok := events[3].(*ErrorEvent); !ok { if events[3].(*DeleteEvent).ID != "1234567" {
t.Fatalf("should be fail: %v", errorEvent.err) t.Fatalf("want %q but %q", "1234567", events[3].(*DeleteEvent).ID)
} }
if errorEvent, ok := events[4].(*ErrorEvent); !ok { if errorEvent, ok := events[4].(*ErrorEvent); !ok {
t.Fatalf("should be fail: %v", errorEvent.err) t.Fatalf("should be fail: %v", errorEvent.err)
@ -132,6 +139,9 @@ func wsTest(t *testing.T, q chan Event, cancel func()) {
if errorEvent, ok := events[5].(*ErrorEvent); !ok { if errorEvent, ok := events[5].(*ErrorEvent); !ok {
t.Fatalf("should be fail: %v", errorEvent.err) t.Fatalf("should be fail: %v", errorEvent.err)
} }
if errorEvent, ok := events[6].(*ErrorEvent); !ok {
t.Fatalf("should be fail: %v", errorEvent.err)
}
} }
func TestStreamingWS(t *testing.T) { func TestStreamingWS(t *testing.T) {