mirror of
https://github.com/thegeeklab/drone-matrix.git
synced 2024-10-31 16:50:41 +00:00
106 lines
3.0 KiB
Go
106 lines
3.0 KiB
Go
package gomatrix
|
|
|
|
import (
|
|
"bytes"
|
|
"fmt"
|
|
"io/ioutil"
|
|
"net/http"
|
|
"testing"
|
|
)
|
|
|
|
func TestClient_LeaveRoom(t *testing.T) {
|
|
cli := mockClient(func(req *http.Request) (*http.Response, error) {
|
|
if req.Method == "POST" && req.URL.Path == "/_matrix/client/r0/rooms/!foo:bar/leave" {
|
|
return &http.Response{
|
|
StatusCode: 200,
|
|
Body: ioutil.NopCloser(bytes.NewBufferString(`{}`)),
|
|
}, nil
|
|
}
|
|
return nil, fmt.Errorf("unhandled URL: %s", req.URL.Path)
|
|
})
|
|
|
|
if _, err := cli.LeaveRoom("!foo:bar"); err != nil {
|
|
t.Fatalf("LeaveRoom: error, got %s", err.Error())
|
|
}
|
|
}
|
|
|
|
func TestClient_GetAvatarUrl(t *testing.T) {
|
|
cli := mockClient(func(req *http.Request) (*http.Response, error) {
|
|
if req.Method == "GET" && req.URL.Path == "/_matrix/client/r0/profile/@user:test.gomatrix.org/avatar_url" {
|
|
return &http.Response{
|
|
StatusCode: 200,
|
|
Body: ioutil.NopCloser(bytes.NewBufferString(`{"avatar_url":"mxc://matrix.org/iJaUjkshgdfsdkjfn"}`)),
|
|
}, nil
|
|
}
|
|
return nil, fmt.Errorf("unhandled URL: %s", req.URL.Path)
|
|
})
|
|
|
|
if response, err := cli.GetAvatarURL(); err != nil {
|
|
t.Fatalf("GetAvatarURL: Got error: %s", err.Error())
|
|
} else if response == "" {
|
|
t.Fatal("GetAvatarURL: Got empty response")
|
|
} else if response != "mxc://matrix.org/iJaUjkshgdfsdkjfn" {
|
|
t.Fatalf("Unexpected response URL: %s", response)
|
|
}
|
|
|
|
}
|
|
|
|
func TestClient_SetAvatarUrl(t *testing.T) {
|
|
cli := mockClient(func(req *http.Request) (*http.Response, error) {
|
|
if req.Method == "PUT" && req.URL.Path == "/_matrix/client/r0/profile/@user:test.gomatrix.org/avatar_url" {
|
|
return &http.Response{
|
|
StatusCode: 200,
|
|
Body: ioutil.NopCloser(bytes.NewBufferString(`{}`)),
|
|
}, nil
|
|
}
|
|
return nil, fmt.Errorf("unhandled URL: %s", req.URL.Path)
|
|
})
|
|
|
|
if err := cli.SetAvatarURL("https://foo.com/bar.png"); err != nil {
|
|
t.Fatalf("GetAvatarURL: Got error: %s", err.Error())
|
|
}
|
|
}
|
|
|
|
func TestClient_StateEvent(t *testing.T) {
|
|
cli := mockClient(func(req *http.Request) (*http.Response, error) {
|
|
if req.Method == "GET" && req.URL.Path == "/_matrix/client/r0/rooms/!foo:bar/state/m.room.name" {
|
|
return &http.Response{
|
|
StatusCode: 200,
|
|
Body: ioutil.NopCloser(bytes.NewBufferString(`{"name":"Room Name Goes Here"}`)),
|
|
}, nil
|
|
}
|
|
return nil, fmt.Errorf("unhandled URL: %s", req.URL.Path)
|
|
})
|
|
|
|
content := struct {
|
|
Name string `json:"name"`
|
|
}{}
|
|
|
|
if err := cli.StateEvent("!foo:bar", "m.room.name", "", &content); err != nil {
|
|
t.Fatalf("StateEvent: error, got %s", err.Error())
|
|
}
|
|
if content.Name != "Room Name Goes Here" {
|
|
t.Fatalf("StateEvent: got %s, want %s", content.Name, "Room Name Goes Here")
|
|
}
|
|
}
|
|
|
|
func mockClient(fn func(*http.Request) (*http.Response, error)) *Client {
|
|
mrt := MockRoundTripper{
|
|
RT: fn,
|
|
}
|
|
|
|
cli, _ := NewClient("https://test.gomatrix.org", "@user:test.gomatrix.org", "abcdef")
|
|
cli.Client = &http.Client{
|
|
Transport: mrt,
|
|
}
|
|
return cli
|
|
}
|
|
|
|
type MockRoundTripper struct {
|
|
RT func(*http.Request) (*http.Response, error)
|
|
}
|
|
|
|
func (t MockRoundTripper) RoundTrip(req *http.Request) (*http.Response, error) {
|
|
return t.RT(req)
|
|
}
|