signaller/internal/backends/memory/rooms_test.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
}