2020-01-23 18:51:10 +01:00
|
|
|
package types
|
|
|
|
|
2020-07-29 20:00:04 +02:00
|
|
|
import (
|
2020-10-08 11:03:37 +02:00
|
|
|
"encoding/json"
|
2022-11-29 15:46:28 +01:00
|
|
|
"reflect"
|
2020-07-29 20:00:04 +02:00
|
|
|
"testing"
|
2020-10-08 11:03:37 +02:00
|
|
|
|
|
|
|
"github.com/matrix-org/gomatrixserverlib"
|
2020-07-29 20:00:04 +02:00
|
|
|
)
|
|
|
|
|
2020-12-10 19:57:10 +01:00
|
|
|
func TestSyncTokens(t *testing.T) {
|
|
|
|
shouldPass := map[string]string{
|
2022-04-06 13:11:19 +02:00
|
|
|
"s4_0_0_0_0_0_0_0_3": StreamingToken{4, 0, 0, 0, 0, 0, 0, 0, 3}.String(),
|
|
|
|
"s3_1_0_0_0_0_2_0_5": StreamingToken{3, 1, 0, 0, 0, 0, 2, 0, 5}.String(),
|
|
|
|
"s3_1_2_3_5_0_0_0_6": StreamingToken{3, 1, 2, 3, 5, 0, 0, 0, 6}.String(),
|
|
|
|
"t3_1": TopologyToken{3, 1}.String(),
|
2020-12-10 19:57:10 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
for a, b := range shouldPass {
|
|
|
|
if a != b {
|
|
|
|
t.Errorf("expected %q, got %q", a, b)
|
|
|
|
}
|
2020-01-23 18:51:10 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
shouldFail := []string{
|
|
|
|
"",
|
|
|
|
"s_",
|
|
|
|
"a3_4",
|
|
|
|
"b",
|
|
|
|
"b-1",
|
|
|
|
"-4",
|
2020-05-13 13:14:50 +02:00
|
|
|
"2",
|
2020-01-23 18:51:10 +01:00
|
|
|
}
|
|
|
|
|
2020-12-10 19:57:10 +01:00
|
|
|
for _, f := range append(shouldFail, "t1_2") {
|
|
|
|
if _, err := NewStreamTokenFromString(f); err == nil {
|
|
|
|
t.Errorf("NewStreamTokenFromString %q should have failed", f)
|
2020-01-23 18:51:10 +01:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2020-12-10 19:57:10 +01:00
|
|
|
for _, f := range append(shouldFail, "s1_2_3_4") {
|
|
|
|
if _, err := NewTopologyTokenFromString(f); err == nil {
|
|
|
|
t.Errorf("NewTopologyTokenFromString %q should have failed", f)
|
2020-01-23 18:51:10 +01:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
2020-10-08 11:03:37 +02:00
|
|
|
|
|
|
|
func TestNewInviteResponse(t *testing.T) {
|
|
|
|
event := `{"auth_events":["$SbSsh09j26UAXnjd3RZqf2lyA3Kw2sY_VZJVZQAV9yA","$EwL53onrLwQ5gL8Dv3VrOOCvHiueXu2ovLdzqkNi3lo","$l2wGmz9iAwevBDGpHT_xXLUA5O8BhORxWIGU1cGi1ZM","$GsWFJLXgdlF5HpZeyWkP72tzXYWW3uQ9X28HBuTztHE"],"content":{"avatar_url":"","displayname":"neilalexander","membership":"invite"},"depth":9,"hashes":{"sha256":"8p+Ur4f8vLFX6mkIXhxI0kegPG7X3tWy56QmvBkExAg"},"origin":"matrix.org","origin_server_ts":1602087113066,"prev_events":["$1v-O6tNwhOZcA8bvCYY-Dnj1V2ZDE58lLPxtlV97S28"],"prev_state":[],"room_id":"!XbeXirGWSPXbEaGokF:matrix.org","sender":"@neilalexander:matrix.org","signatures":{"dendrite.neilalexander.dev":{"ed25519:BMJi":"05KQ5lPw0cSFsE4A0x1z7vi/3cc8bG4WHUsFWYkhxvk/XkXMGIYAYkpNThIvSeLfdcHlbm/k10AsBSKH8Uq4DA"},"matrix.org":{"ed25519:a_RXGa":"jeovuHr9E/x0sHbFkdfxDDYV/EyoeLi98douZYqZ02iYddtKhfB7R3WLay/a+D3V3V7IW0FUmPh/A404x5sYCw"}},"state_key":"@neilalexander:dendrite.neilalexander.dev","type":"m.room.member","unsigned":{"age":2512,"invite_room_state":[{"content":{"join_rule":"invite"},"sender":"@neilalexander:matrix.org","state_key":"","type":"m.room.join_rules"},{"content":{"avatar_url":"mxc://matrix.org/BpDaozLwgLnlNStxDxvLzhPr","displayname":"neilalexander","membership":"join"},"sender":"@neilalexander:matrix.org","state_key":"@neilalexander:matrix.org","type":"m.room.member"},{"content":{"name":"Test room"},"sender":"@neilalexander:matrix.org","state_key":"","type":"m.room.name"}]},"_room_version":"5"}`
|
|
|
|
expected := `{"invite_state":{"events":[{"content":{"join_rule":"invite"},"sender":"@neilalexander:matrix.org","state_key":"","type":"m.room.join_rules"},{"content":{"avatar_url":"mxc://matrix.org/BpDaozLwgLnlNStxDxvLzhPr","displayname":"neilalexander","membership":"join"},"sender":"@neilalexander:matrix.org","state_key":"@neilalexander:matrix.org","type":"m.room.member"},{"content":{"name":"Test room"},"sender":"@neilalexander:matrix.org","state_key":"","type":"m.room.name"},{"content":{"avatar_url":"","displayname":"neilalexander","membership":"invite"},"event_id":"$GQmw8e8-26CQv1QuFoHBHpKF1hQj61Flg3kvv_v_XWs","origin_server_ts":1602087113066,"sender":"@neilalexander:matrix.org","state_key":"@neilalexander:dendrite.neilalexander.dev","type":"m.room.member"}]}}`
|
|
|
|
|
|
|
|
ev, err := gomatrixserverlib.NewEventFromTrustedJSON([]byte(event), false, gomatrixserverlib.RoomVersionV5)
|
|
|
|
if err != nil {
|
|
|
|
t.Fatal(err)
|
|
|
|
}
|
|
|
|
|
|
|
|
res := NewInviteResponse(ev.Headered(gomatrixserverlib.RoomVersionV5))
|
|
|
|
j, err := json.Marshal(res)
|
|
|
|
if err != nil {
|
|
|
|
t.Fatal(err)
|
|
|
|
}
|
|
|
|
|
|
|
|
if string(j) != expected {
|
|
|
|
t.Fatalf("Invite response didn't contain correct info")
|
|
|
|
}
|
|
|
|
}
|
2022-11-29 15:46:28 +01:00
|
|
|
|
|
|
|
func TestJoinResponse_MarshalJSON(t *testing.T) {
|
|
|
|
type fields struct {
|
|
|
|
Summary *Summary
|
|
|
|
State *ClientEvents
|
|
|
|
Timeline *Timeline
|
|
|
|
Ephemeral *ClientEvents
|
|
|
|
AccountData *ClientEvents
|
|
|
|
UnreadNotifications *UnreadNotifications
|
|
|
|
}
|
|
|
|
tests := []struct {
|
|
|
|
name string
|
|
|
|
fields fields
|
|
|
|
want []byte
|
|
|
|
wantErr bool
|
|
|
|
}{
|
|
|
|
{
|
|
|
|
name: "empty state is removed",
|
|
|
|
fields: fields{
|
|
|
|
State: &ClientEvents{},
|
|
|
|
},
|
|
|
|
want: []byte("{}"),
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "empty accountdata is removed",
|
|
|
|
fields: fields{
|
|
|
|
AccountData: &ClientEvents{},
|
|
|
|
},
|
|
|
|
want: []byte("{}"),
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "empty ephemeral is removed",
|
|
|
|
fields: fields{
|
|
|
|
Ephemeral: &ClientEvents{},
|
|
|
|
},
|
|
|
|
want: []byte("{}"),
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "empty timeline is removed",
|
|
|
|
fields: fields{
|
|
|
|
Timeline: &Timeline{},
|
|
|
|
},
|
|
|
|
want: []byte("{}"),
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "empty summary is removed",
|
|
|
|
fields: fields{
|
|
|
|
Summary: &Summary{},
|
|
|
|
},
|
|
|
|
want: []byte("{}"),
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "unread notifications are removed, if everything else is empty",
|
|
|
|
fields: fields{
|
|
|
|
UnreadNotifications: &UnreadNotifications{},
|
|
|
|
},
|
|
|
|
want: []byte("{}"),
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "unread notifications are NOT removed, if state is set",
|
|
|
|
fields: fields{
|
|
|
|
State: &ClientEvents{Events: []gomatrixserverlib.ClientEvent{{Content: []byte("{}")}}},
|
|
|
|
UnreadNotifications: &UnreadNotifications{NotificationCount: 1},
|
|
|
|
},
|
|
|
|
want: []byte(`{"state":{"events":[{"content":{},"type":""}]},"unread_notifications":{"highlight_count":0,"notification_count":1}}`),
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "roomID is removed from EDUs",
|
|
|
|
fields: fields{
|
|
|
|
Ephemeral: &ClientEvents{
|
|
|
|
Events: []gomatrixserverlib.ClientEvent{
|
|
|
|
{RoomID: "!someRandomRoomID:test", Content: []byte("{}")},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
want: []byte(`{"ephemeral":{"events":[{"content":{},"type":""}]}}`),
|
|
|
|
},
|
|
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
|
|
jr := JoinResponse{
|
|
|
|
Summary: tt.fields.Summary,
|
|
|
|
State: tt.fields.State,
|
|
|
|
Timeline: tt.fields.Timeline,
|
|
|
|
Ephemeral: tt.fields.Ephemeral,
|
|
|
|
AccountData: tt.fields.AccountData,
|
|
|
|
UnreadNotifications: tt.fields.UnreadNotifications,
|
|
|
|
}
|
|
|
|
got, err := jr.MarshalJSON()
|
|
|
|
if (err != nil) != tt.wantErr {
|
|
|
|
t.Errorf("MarshalJSON() error = %v, wantErr %v", err, tt.wantErr)
|
|
|
|
return
|
|
|
|
}
|
|
|
|
if !reflect.DeepEqual(got, tt.want) {
|
|
|
|
t.Errorf("MarshalJSON() got = %v, want %v", string(got), string(tt.want))
|
|
|
|
}
|
|
|
|
})
|
|
|
|
}
|
|
|
|
}
|