add final test and apply pre-commit hooks

This commit is contained in:
northernSage 2021-03-26 15:46:39 -03:00
parent 0977769472
commit be2a64b6f8
1 changed files with 50 additions and 44 deletions

View File

@ -2,9 +2,9 @@ package gomatrix
import ( import (
"encoding/json" "encoding/json"
"testing"
"strings"
"reflect" "reflect"
"strings"
"testing"
) )
// example events from docs // example events from docs
@ -12,10 +12,11 @@ var testEvent = `{
"content": { "content": {
"body": "eventbody123", "body": "eventbody123",
"msgtype": "m.text", "msgtype": "m.text",
"membership": "join",
"format": "org.matrix.custom.html", "format": "org.matrix.custom.html",
"formatted_body": "<b>This is an example text message</b>" "formatted_body": "<b>This is an example text message</b>"
}, },
"type": "m.room.message", "type": "m.room.member",
"event_id": "$143273582443PhrSn:example.org", "event_id": "$143273582443PhrSn:example.org",
"room_id": "!726s6s6q:example.com", "room_id": "!726s6s6q:example.com",
"sender": "@example:example.org", "sender": "@example:example.org",
@ -76,7 +77,6 @@ func TestGetStateEvent (t *testing.T) {
} }
func TestGetMembershipState(t *testing.T) { func TestGetMembershipState(t *testing.T) {
var e Event var e Event
err := json.NewDecoder(strings.NewReader(testEvent)).Decode(&e) err := json.NewDecoder(strings.NewReader(testEvent)).Decode(&e)
if err != nil { if err != nil {
@ -84,6 +84,12 @@ func TestGetMembershipState (t *testing.T) {
} }
r := NewRoom("test_id_001") r := NewRoom("test_id_001")
r.UpdateState(&e) r.UpdateState(&e)
state := r.GetMembershipState("@alice:example.org") 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)
}
} }