strukturag_nextcloud-spreed-signaling/roomsessions_test.go

/**
 * Standalone signaling server for the Nextcloud Spreed app.
 * Copyright (C) 2019 struktur AG
 *
 * @author Joachim Bauch <bauch@struktur.de>
 *
 * @license GNU AGPL version 3 or any later version
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Affero General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU Affero General Public License for more details.
 *
 * You should have received a copy of the GNU Affero General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */
package signaling

import (
	"encoding/json"
	"net/url"
	"testing"
	"time"
)

type DummySession struct {
	publicId string
}

func (s *DummySession) PrivateId() string {
	return ""
}

func (s *DummySession) PublicId() string {
	return s.publicId
}

func (s *DummySession) ClientType() string {
	return ""
}

func (s *DummySession) Data() *SessionIdData {
	return nil
}

func (s *DummySession) UserId() string {
	return ""
}

func (s *DummySession) UserData() *json.RawMessage {
	return nil
}

func (s *DummySession) Backend() *Backend {
	return nil
}

func (s *DummySession) BackendUrl() string {
	return ""
}

func (s *DummySession) ParsedBackendUrl() *url.URL {
	return nil
}

func (s *DummySession) SetRoom(room *Room) {
}

func (s *DummySession) GetRoom() *Room {
	return nil
}

func (s *DummySession) LeaveRoom(notify bool) *Room {
	return nil
}

func (s *DummySession) IsExpired(now time.Time) bool {
	return false
}

func (s *DummySession) Close() {
}

func (s *DummySession) HasPermission(permission Permission) bool {
	return false
}

func checkSession(t *testing.T, sessions RoomSessions, sessionId string, roomSessionId string) Session {
	session := &DummySession{
		publicId: sessionId,
	}
	if err := sessions.SetRoomSession(session, roomSessionId); err != nil {
		t.Fatalf("Expected no error, got %s", err)
	}
	if sid, err := sessions.GetSessionId(roomSessionId); err != nil {
		t.Errorf("Expected session id %s, got error %s", sessionId, err)
	} else if sid != sessionId {
		t.Errorf("Expected session id %s, got %s", sessionId, sid)
	}
	return session
}

func testRoomSessions(t *testing.T, sessions RoomSessions) {
	if sid, err := sessions.GetSessionId("unknown"); err != nil && err != ErrNoSuchRoomSession {
		t.Errorf("Expected error about invalid room session, got %s", err)
	} else if err == nil {
		t.Errorf("Expected error about invalid room session, got session id %s", sid)
	}

	s1 := checkSession(t, sessions, "session1", "room1")
	s2 := checkSession(t, sessions, "session2", "room2")

	if sid, err := sessions.GetSessionId("room1"); err != nil {
		t.Errorf("Expected session id %s, got error %s", s1.PublicId(), err)
	} else if sid != s1.PublicId() {
		t.Errorf("Expected session id %s, got %s", s1.PublicId(), sid)
	}

	sessions.DeleteRoomSession(s1)
	if sid, err := sessions.GetSessionId("room1"); err != nil && err != ErrNoSuchRoomSession {
		t.Errorf("Expected error about invalid room session, got %s", err)
	} else if err == nil {
		t.Errorf("Expected error about invalid room session, got session id %s", sid)
	}
	if sid, err := sessions.GetSessionId("room2"); err != nil {
		t.Errorf("Expected session id %s, got error %s", s2.PublicId(), err)
	} else if sid != s2.PublicId() {
		t.Errorf("Expected session id %s, got %s", s2.PublicId(), sid)
	}

	if err := sessions.SetRoomSession(s1, "room-session"); err != nil {
		t.Error(err)
	}
	if err := sessions.SetRoomSession(s2, "room-session"); err != nil {
		t.Error(err)
	}
	sessions.DeleteRoomSession(s1)
	if sid, err := sessions.GetSessionId("room-session"); err != nil {
		t.Errorf("Expected session id %s, got error %s", s2.PublicId(), err)
	} else if sid != s2.PublicId() {
		t.Errorf("Expected session id %s, got %s", s2.PublicId(), sid)
	}
}