mirror of https://github.com/matrix-org/gomatrix
96 lines
2.7 KiB
Go
96 lines
2.7 KiB
Go
package gomatrix
|
|
|
|
import (
|
|
"encoding/json"
|
|
"reflect"
|
|
"strings"
|
|
"testing"
|
|
)
|
|
|
|
// example events from docs
|
|
var testEvent = `{
|
|
"content": {
|
|
"body": "eventbody123",
|
|
"msgtype": "m.text",
|
|
"membership": "join",
|
|
"format": "org.matrix.custom.html",
|
|
"formatted_body": "<b>This is an example text message</b>"
|
|
},
|
|
"type": "m.room.member",
|
|
"event_id": "$143273582443PhrSn:example.org",
|
|
"room_id": "!726s6s6q:example.com",
|
|
"sender": "@example:example.org",
|
|
"state_key": "@alice:example.org",
|
|
"origin_server_ts": 1432735824653,
|
|
"unsigned": {
|
|
"age": 1234
|
|
}
|
|
}`
|
|
|
|
// IsRoom checks if a given variable is of type *Room
|
|
func IsRoom(t interface{}) bool {
|
|
switch t.(type) {
|
|
case *Room:
|
|
return true
|
|
default:
|
|
return false
|
|
}
|
|
}
|
|
|
|
func TestNewRoom(t *testing.T) {
|
|
var testID = "test-id-001"
|
|
r := NewRoom(testID)
|
|
if r.ID != testID {
|
|
t.Fatalf("TestNewRoom: Expected ID '%s', but got '%s'", testID, r.ID)
|
|
}
|
|
if !IsRoom(r) {
|
|
t.Fatalf("TestNewRoom: Wrong data type returned, expected 'Room'")
|
|
}
|
|
}
|
|
|
|
func TestUpdateState(t *testing.T) {
|
|
var e Event
|
|
err := json.NewDecoder(strings.NewReader(testEvent)).Decode(&e)
|
|
if err != nil {
|
|
t.Fatalf("TestUpdateState: Something went wrong while parsing: %s", testEvent)
|
|
}
|
|
r := NewRoom("test_id_001")
|
|
r.UpdateState(&e)
|
|
// check mapping here first
|
|
if r.State[e.Type][*e.StateKey] != &e {
|
|
t.Fatalf("TestUpdateState: Wrong object reference. The reference is not pointing to the expected event object")
|
|
}
|
|
}
|
|
|
|
func TestGetStateEvent(t *testing.T) {
|
|
var e Event
|
|
err := json.NewDecoder(strings.NewReader(testEvent)).Decode(&e)
|
|
if err != nil {
|
|
t.Fatalf("TestUpdateState: Something went wrong while parsing: %s", testEvent)
|
|
}
|
|
r := NewRoom("test_id_001")
|
|
r.UpdateState(&e)
|
|
e2 := *r.GetStateEvent(e.Type, *e.StateKey)
|
|
if !reflect.DeepEqual(e, e2) {
|
|
t.Fatalf("TestGetStateEvent: Wrong event object returned, expected object at address %p, got %p", &e, &e2)
|
|
}
|
|
}
|
|
|
|
func TestGetMembershipState(t *testing.T) {
|
|
var e Event
|
|
err := json.NewDecoder(strings.NewReader(testEvent)).Decode(&e)
|
|
if err != nil {
|
|
t.Fatalf("TestUpdateState: Something went wrong while parsing: %s", testEvent)
|
|
}
|
|
r := NewRoom("test_id_001")
|
|
r.UpdateState(&e)
|
|
membState := r.GetMembershipState("@alice:example.org")
|
|
if membState != e.Content["membership"] {
|
|
t.Fatalf("TestGetMembershipState: Wrong membership state, expected '%s', got '%s'", e.Content["membership"], membState)
|
|
}
|
|
membState = r.GetMembershipState("@unkown:example.org")
|
|
if membState != "leave" {
|
|
t.Fatalf("TestGetMembershipState: Wrong membership state, expected 'leave', got '%s'", membState)
|
|
}
|
|
}
|