osukey/osu.Game/Tests/Visual/Multiplayer/TestMultiplayerRoomManager.cs

// Copyright (c) ppy Pty Ltd <contact@ppy.sh>. Licensed under the MIT Licence.
// See the LICENCE file in the repository root for full licence text.

using System;
using System.Collections.Generic;
using osu.Game.Online.Rooms;
using osu.Game.Screens.OnlinePlay.Components;
using osu.Game.Screens.OnlinePlay.Multiplayer;
using osu.Game.Tests.Visual.OnlinePlay;

namespace osu.Game.Tests.Visual.Multiplayer
{
    /// <summary>
    /// A <see cref="RoomManager"/> for use in multiplayer test scenes.
    /// Should generally not be used by itself outside of a <see cref="MultiplayerTestScene"/>.
    /// </summary>
    public class TestMultiplayerRoomManager : MultiplayerRoomManager
    {
        public bool RoomJoined { get; private set; }

        private readonly TestRoomRequestsHandler requestsHandler;

        public TestMultiplayerRoomManager(TestRoomRequestsHandler requestsHandler)
        {
            this.requestsHandler = requestsHandler;
        }

        public IReadOnlyList<Room> ServerSideRooms => requestsHandler.ServerSideRooms;

        public override void CreateRoom(Room room, Action<Room> onSuccess = null, Action<string> onError = null)
        {
            base.CreateRoom(room, r =>
            {
                onSuccess?.Invoke(r);
                RoomJoined = true;
            }, onError);
        }

        public override void JoinRoom(Room room, string password = null, Action<Room> onSuccess = null, Action<string> onError = null)
        {
            base.JoinRoom(room, password, r =>
            {
                onSuccess?.Invoke(r);
                RoomJoined = true;
            }, onError);
        }

        public override void PartRoom()
        {
            base.PartRoom();
            RoomJoined = false;
        }

        /// <summary>
        /// Adds a room to a local "server-side" list that's returned when a <see cref="GetRoomsRequest"/> is fired.
        /// </summary>
        /// <param name="room">The room.</param>
        public void AddServerSideRoom(Room room) => requestsHandler.AddServerSideRoom(room);
    }
}