mirror of
https://github.com/signaller-matrix/signaller.git
synced 2024-11-22 18:32:20 +00:00
150 lines
3.5 KiB
Go
150 lines
3.5 KiB
Go
package memory
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/signaller-matrix/signaller/internal/models/createroom"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestCreateRoom(t *testing.T) {
|
|
backend := NewBackend("localhost")
|
|
|
|
user, _, err := backend.Register("user1", "", "")
|
|
assert.NoError(t, err)
|
|
|
|
request := createroom.Request{
|
|
RoomAliasName: "room1",
|
|
Name: "room1",
|
|
Topic: "topic",
|
|
Preset: createroom.PublicChat}
|
|
|
|
room, err := user.CreateRoom(request)
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, request.RoomAliasName, room.AliasName())
|
|
assert.Equal(t, request.Name, room.Name())
|
|
assert.Equal(t, request.Topic, room.Topic())
|
|
assert.Equal(t, user.ID(), room.Creator().ID())
|
|
assert.Equal(t, request.Preset, room.State())
|
|
assert.Equal(t, 1, len(backend.rooms))
|
|
assert.Equal(t, "!"+room.(*Room).id+":"+backend.hostname, room.ID())
|
|
}
|
|
|
|
func TestCreateAlreadyExistingRoom(t *testing.T) {
|
|
backend := NewBackend("localhost")
|
|
|
|
user, _, _ := backend.Register("user1", "", "")
|
|
|
|
request := createroom.Request{
|
|
RoomAliasName: "room1",
|
|
Name: "room1",
|
|
Topic: "topic"}
|
|
|
|
_, err := user.CreateRoom(request)
|
|
assert.NoError(t, err)
|
|
|
|
_, err = user.CreateRoom(request)
|
|
assert.NotNil(t, err)
|
|
}
|
|
|
|
func TestSetRoomTopic(t *testing.T) {
|
|
backend := NewBackend("localhost")
|
|
|
|
user, _, _ := backend.Register("user1", "", "")
|
|
|
|
request := createroom.Request{
|
|
RoomAliasName: "room1",
|
|
Name: "room1",
|
|
Topic: "topic"}
|
|
|
|
room, _ := user.CreateRoom(request)
|
|
|
|
var newTopic = "new topic"
|
|
err := user.SetTopic(room, newTopic)
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, newTopic, room.Topic())
|
|
}
|
|
|
|
func TestSetRoomTopicWithnprivelegedUser(t *testing.T) {
|
|
backend := NewBackend("localhost")
|
|
|
|
creator, _, _ := backend.Register("user1", "", "")
|
|
user2, _, _ := backend.Register("user2", "", "")
|
|
|
|
request := createroom.Request{
|
|
RoomAliasName: "room1",
|
|
Name: "room1",
|
|
Topic: "topic"}
|
|
|
|
room, _ := creator.CreateRoom(request)
|
|
|
|
var newTopic = "new topic"
|
|
err := user2.SetTopic(room, newTopic)
|
|
assert.NotNil(t, err)
|
|
}
|
|
|
|
func TestLeaveRoom(t *testing.T) {
|
|
backend := NewBackend("localhost")
|
|
|
|
user, _, _ := backend.Register("user1", "", "")
|
|
|
|
request := createroom.Request{
|
|
RoomAliasName: "room1",
|
|
Name: "room1",
|
|
Topic: "topic"}
|
|
|
|
room, _ := user.CreateRoom(request)
|
|
|
|
assert.Equal(t, 1, len(room.(*Room).joined))
|
|
|
|
err := user.LeaveRoom(room)
|
|
assert.Equal(t, 0, len(room.(*Room).joined))
|
|
|
|
// Try to leave room again must throw error
|
|
err = user.LeaveRoom(room)
|
|
assert.NotNil(t, err)
|
|
}
|
|
|
|
func TestRoomUserCount(t *testing.T) {
|
|
backend := NewBackend("localhost")
|
|
|
|
user1, _, err := backend.Register("user1", "", "")
|
|
assert.NoError(t, err)
|
|
|
|
request := createroom.Request{
|
|
RoomAliasName: "room1",
|
|
Name: "room1",
|
|
Topic: "topic"}
|
|
|
|
room, err := user1.CreateRoom(request)
|
|
assert.NoError(t, err)
|
|
assert.Len(t, room.Users(), 1)
|
|
|
|
// TODO: add join another user test
|
|
}
|
|
|
|
func TestRoomAliases(t *testing.T) {
|
|
backend := NewBackend("localhost")
|
|
|
|
user1, _, err := backend.Register("user1", "", "")
|
|
assert.NoError(t, err)
|
|
|
|
request := createroom.Request{
|
|
RoomAliasName: "room1",
|
|
Name: "room1",
|
|
Topic: "topic"}
|
|
|
|
room, err := user1.CreateRoom(request)
|
|
assert.NoError(t, err)
|
|
assert.NotNil(t, room)
|
|
|
|
var expecterRoomAliases = []string{"alias1", "alias2"}
|
|
for _, alias := range expecterRoomAliases {
|
|
err := user1.AddRoomAlias(room, alias)
|
|
assert.NoError(t, err)
|
|
}
|
|
|
|
assert.Equal(t, expecterRoomAliases, room.Aliases())
|
|
}
|