signaller/internal/backends/memory/backend_test.go
2019-08-02 18:17:18 +05:00

139 lines
2.8 KiB
Go

package memory
import (
"testing"
"github.com/stretchr/testify/assert"
"github.com/nxshock/signaller/internal/models/createroom"
)
func TestRegisterUser(t *testing.T) {
backend := NewBackend("localhost")
var (
username = "user1"
password = "password1"
device = "device1"
)
user, token, err := backend.Register(username, password, device)
assert.Nil(t, err)
assert.Equal(t, username, user.Name())
assert.Equal(t, password, user.Password())
assert.NotEmpty(t, token)
}
func TestRegisterUserWithAlreadyTakenName(t *testing.T) {
backend := NewBackend("localhost")
var (
userName = "user1"
)
_, _, err := backend.Register(userName, "", "")
assert.Nil(t, err)
_, _, err = backend.Register(userName, "", "")
assert.NotNil(t, err)
}
func TestLogin(t *testing.T) {
backend := NewBackend("localhost")
var (
userName = "user1"
password = "password1"
)
_, _, err := backend.Register(userName, password, "")
assert.Nil(t, err)
_, token, err := backend.Login(userName, password, "")
assert.Nil(t, err)
assert.NotZero(t, token)
}
func TestLoginWithWrongCredentials(t *testing.T) {
backend := NewBackend("localhost")
var (
userName = "user1"
password = "password1"
)
_, _, err := backend.Register(userName, password, "")
assert.Nil(t, err)
_, _, err = backend.Login(userName, "wrong password", "")
assert.NotNil(t, err)
_, _, err = backend.Login("wrong user name", password, "")
assert.NotNil(t, err)
}
func TestLogout(t *testing.T) {
backend := NewBackend("localhost")
var (
userName = "user1"
password = "password1"
)
user, _, err := backend.Register(userName, password, "")
assert.Nil(t, err)
_, token, err := backend.Login(userName, password, "")
assert.Nil(t, err)
assert.NotZero(t, token)
user.Logout(token)
assert.Nil(t, backend.GetUserByToken(token))
}
func TestGetRoomByID(t *testing.T) {
backend := NewBackend("localhost")
user, token, err := backend.Register("user", "", "")
assert.Nil(t, err)
assert.NotNil(t, user)
assert.NotEmpty(t, token)
request := createroom.Request{
RoomAliasName: "room1",
Name: "room1"}
room, err := user.CreateRoom(request)
assert.Nil(t, err)
assert.NotNil(t, room)
assert.Equal(t, room.ID(), backend.GetRoomByID(room.ID()).ID())
// Get room with wrong id
room = backend.GetRoomByID("worng id")
assert.Nil(t, room)
}
func TestGetUserByName(t *testing.T) {
backend := NewBackend("localhost")
var (
userName = "user"
)
user, token, err := backend.Register(userName, "", "")
assert.Nil(t, err)
assert.NotNil(t, user)
assert.NotEmpty(t, token)
t.Run("Test picking user with username", func(_ *testing.T) {
user2 := backend.GetUserByName(userName)
assert.Equal(t, user, user2)
})
t.Run("Test picking user with wrong username", func(_ *testing.T) {
user2 := backend.GetUserByName("wrong username")
assert.Nil(t, user2)
})
}