Update TestMultiplayerClient to be more in line with server

This commit is contained in:
Dan Balasescu
2021-11-13 01:42:51 +09:00
parent 25202316cc
commit 867403d947

View File

@ -42,6 +42,7 @@ namespace osu.Game.Tests.Visual.Multiplayer
private BeatmapManager beatmaps { get; set; } = null!; private BeatmapManager beatmaps { get; set; } = null!;
private readonly TestMultiplayerRoomManager roomManager; private readonly TestMultiplayerRoomManager roomManager;
private PlaylistItem? currentItem;
public TestMultiplayerClient(TestMultiplayerRoomManager roomManager) public TestMultiplayerClient(TestMultiplayerRoomManager roomManager)
{ {
@ -138,7 +139,7 @@ namespace osu.Game.Tests.Visual.Multiplayer
((IMultiplayerClient)this).ResultsReady(); ((IMultiplayerClient)this).ResultsReady();
finishPlaylistItem().ContinueWith(_ => advanceToNextPlaylistItem()); Task.Run(finishCurrentItem);
} }
break; break;
@ -153,7 +154,7 @@ namespace osu.Game.Tests.Visual.Multiplayer
((IMultiplayerClient)this).UserBeatmapAvailabilityChanged(userId, newBeatmapAvailability); ((IMultiplayerClient)this).UserBeatmapAvailabilityChanged(userId, newBeatmapAvailability);
} }
protected override Task<MultiplayerRoom> JoinRoom(long roomId, string? password = null) protected override async Task<MultiplayerRoom> JoinRoom(long roomId, string? password = null)
{ {
var apiRoom = roomManager.ServerSideRooms.Single(r => r.RoomID.Value == roomId); var apiRoom = roomManager.ServerSideRooms.Single(r => r.RoomID.Value == roomId);
@ -165,13 +166,16 @@ namespace osu.Game.Tests.Visual.Multiplayer
User = api.LocalUser.Value User = api.LocalUser.Value
}; };
await updateCurrentItem(apiRoom: apiRoom).ConfigureAwait(false);
Debug.Assert(currentItem != null);
var room = new MultiplayerRoom(roomId) var room = new MultiplayerRoom(roomId)
{ {
Settings = Settings =
{ {
Name = apiRoom.Name.Value, Name = apiRoom.Name.Value,
MatchType = apiRoom.Type.Value, MatchType = apiRoom.Type.Value,
PlaylistItemId = apiRoom.Playlist.Single().ID, PlaylistItemId = currentItem.ID,
Password = password, Password = password,
QueueMode = apiRoom.QueueMode.Value QueueMode = apiRoom.QueueMode.Value
}, },
@ -182,7 +186,7 @@ namespace osu.Game.Tests.Visual.Multiplayer
RoomSetupAction?.Invoke(room); RoomSetupAction?.Invoke(room);
RoomSetupAction = null; RoomSetupAction = null;
return Task.FromResult(room); return room;
} }
protected override void OnRoomJoined() protected override void OnRoomJoined()
@ -208,11 +212,19 @@ namespace osu.Game.Tests.Visual.Multiplayer
{ {
Debug.Assert(Room != null); Debug.Assert(Room != null);
Debug.Assert(APIRoom != null); Debug.Assert(APIRoom != null);
Debug.Assert(currentItem != null);
// Server is authoritative for the time being. // Server is authoritative for the time being.
settings.PlaylistItemId = Room.Settings.PlaylistItemId; settings.PlaylistItemId = Room.Settings.PlaylistItemId;
await changeSettings(settings).ConfigureAwait(false); await changeQueueMode(settings.QueueMode).ConfigureAwait(false);
await ((IMultiplayerClient)this).SettingsChanged(settings).ConfigureAwait(false);
foreach (var user in Room.Users.Where(u => u.State == MultiplayerUserState.Ready))
ChangeUserState(user.UserID, MultiplayerUserState.Idle);
await changeMatchType(settings.MatchType).ConfigureAwait(false);
} }
public override Task ChangeState(MultiplayerUserState newState) public override Task ChangeState(MultiplayerUserState newState)
@ -282,17 +294,28 @@ namespace osu.Game.Tests.Visual.Multiplayer
{ {
Debug.Assert(Room != null); Debug.Assert(Room != null);
Debug.Assert(APIRoom != null); Debug.Assert(APIRoom != null);
Debug.Assert(currentItem != null);
if (Room.Settings.QueueMode == QueueModes.HostOnly && APIRoom.Playlist.Count > 0) if (Room.Settings.QueueMode == QueueModes.HostOnly && Room.Host?.UserID != LocalUser?.UserID)
{
if (Room.Host?.UserID != LocalUser?.UserID)
throw new InvalidOperationException("Local user is not the room host."); throw new InvalidOperationException("Local user is not the room host.");
item.ID = APIRoom.Playlist.Last().ID; switch (Room.Settings.QueueMode)
{
case QueueModes.HostOnly:
// In host-only mode, the current item is re-used.
item.ID = currentItem.ID;
await ((IMultiplayerClient)this).PlaylistItemChanged(item).ConfigureAwait(false); await ((IMultiplayerClient)this).PlaylistItemChanged(item).ConfigureAwait(false);
// Note: Unlike the server, this is the easiest way to update the current item at this point.
await updateCurrentItem(false).ConfigureAwait(false);
break;
default:
item.ID = APIRoom.Playlist.Last().ID + 1;
await ((IMultiplayerClient)this).PlaylistItemAdded(item).ConfigureAwait(false);
await updateCurrentItem().ConfigureAwait(false);
break;
} }
else
await addPlaylistItem(item).ConfigureAwait(false);
} }
public override Task RemovePlaylistItem(long playlistItemId) public override Task RemovePlaylistItem(long playlistItemId)
@ -325,28 +348,6 @@ namespace osu.Game.Tests.Visual.Multiplayer
return Task.FromResult(apiSet); return Task.FromResult(apiSet);
} }
private async Task changeSettings(MultiplayerRoomSettings settings)
{
Debug.Assert(Room != null);
Debug.Assert(APIRoom != null);
if (settings.QueueMode == QueueModes.HostOnly)
{
foreach (var playlistItem in APIRoom.Playlist.Where(i => !i.Expired && i.ID != Room.Settings.PlaylistItemId).ToArray())
{
APIRoom.Playlist.Remove(playlistItem);
await ((IMultiplayerClient)this).PlaylistItemRemoved(playlistItem.ID).ConfigureAwait(false);
}
}
await ((IMultiplayerClient)this).SettingsChanged(settings).ConfigureAwait(false);
foreach (var user in Room.Users.Where(u => u.State == MultiplayerUserState.Ready))
ChangeUserState(user.UserID, MultiplayerUserState.Idle);
await changeMatchType(settings.MatchType).ConfigureAwait(false);
}
private async Task changeMatchType(MatchType type) private async Task changeMatchType(MatchType type)
{ {
Debug.Assert(Room != null); Debug.Assert(Room != null);
@ -369,47 +370,69 @@ namespace osu.Game.Tests.Visual.Multiplayer
} }
} }
private async Task addPlaylistItem(APIPlaylistItem newItem) private async Task changeQueueMode(QueueModes newMode)
{ {
Debug.Assert(Room != null);
Debug.Assert(APIRoom != null); Debug.Assert(APIRoom != null);
Debug.Assert(currentItem != null);
newItem.ID = (APIRoom.Playlist.LastOrDefault()?.ID ?? 0) + 1; if (newMode == QueueModes.HostOnly)
await ((IMultiplayerClient)this).PlaylistItemAdded(newItem).ConfigureAwait(false); {
foreach (var playlistItem in APIRoom.Playlist.Where(i => !i.Expired && i.ID != currentItem.ID).ToArray())
// A more valid selection can occur as a result of adding a new playlist item (e.g. if all previous items were expired). await ((IMultiplayerClient)this).PlaylistItemRemoved(playlistItem.ID).ConfigureAwait(false);
await advanceToNextPlaylistItem().ConfigureAwait(false);
} }
private async Task finishPlaylistItem() // When changing modes, items could have been added (above) or the queueing order could have changed.
await updateCurrentItem().ConfigureAwait(false);
}
private async Task finishCurrentItem()
{ {
Debug.Assert(Room != null); Debug.Assert(Room != null);
Debug.Assert(APIRoom != null); Debug.Assert(APIRoom != null);
Debug.Assert(currentItem != null);
var currentItem = APIRoom.Playlist.Single(i => i.ID == Room.Settings.PlaylistItemId);
// Expire the current playlist item. // Expire the current playlist item.
await ((IMultiplayerClient)this).PlaylistItemChanged(new APIPlaylistItem(currentItem) { Expired = true }).ConfigureAwait(false); await ((IMultiplayerClient)this).PlaylistItemChanged(new APIPlaylistItem(currentItem) { Expired = true }).ConfigureAwait(false);
// In host-only mode, a duplicate playlist item will be used for the next round. // In host-only mode, a duplicate playlist item will be used for the next round.
if (Room.Settings.QueueMode == QueueModes.HostOnly) if (Room.Settings.QueueMode == QueueModes.HostOnly)
await addPlaylistItem(new APIPlaylistItem(currentItem)).ConfigureAwait(false); await duplicateCurrentItem().ConfigureAwait(false);
await updateCurrentItem().ConfigureAwait(false);
} }
private async Task advanceToNextPlaylistItem() private async Task duplicateCurrentItem()
{ {
Debug.Assert(Room != null); Debug.Assert(Room != null);
Debug.Assert(APIRoom != null); Debug.Assert(APIRoom != null);
Debug.Assert(currentItem != null);
long nextId; var newItem = new PlaylistItem
{
ID = APIRoom.Playlist.Last().ID + 1,
BeatmapID = currentItem.BeatmapID,
RulesetID = currentItem.RulesetID,
};
switch (Room.Settings.QueueMode) newItem.AllowedMods.AddRange(currentItem.AllowedMods);
newItem.RequiredMods.AddRange(currentItem.AllowedMods);
await ((IMultiplayerClient)this).PlaylistItemAdded(new APIPlaylistItem(newItem)).ConfigureAwait(false);
}
private async Task updateCurrentItem(bool notify = true, Room? apiRoom = null)
{
if (apiRoom == null)
{
Debug.Assert(APIRoom != null);
apiRoom = APIRoom;
}
switch (apiRoom.QueueMode.Value)
{ {
default: default:
// Pick the single non-expired playlist item. // Pick the single non-expired playlist item.
nextId = APIRoom.Playlist.FirstOrDefault(i => !i.Expired)?.ID currentItem = apiRoom.Playlist.FirstOrDefault(i => !i.Expired) ?? apiRoom.Playlist.Last();
?? APIRoom.Playlist.LastOrDefault()?.ID
?? 0;
break; break;
case QueueModes.FairRotate: case QueueModes.FairRotate:
@ -417,10 +440,13 @@ namespace osu.Game.Tests.Visual.Multiplayer
throw new NotImplementedException(); throw new NotImplementedException();
} }
if (nextId != Room.Settings.PlaylistItemId) if (Room != null)
{ {
Room.Settings.PlaylistItemId = nextId; long lastItem = Room.Settings.PlaylistItemId;
await changeSettings(Room.Settings).ConfigureAwait(false); Room.Settings.PlaylistItemId = currentItem.ID;
if (notify && currentItem.ID != lastItem)
await ((IMultiplayerClient)this).SettingsChanged(Room.Settings).ConfigureAwait(false);
} }
} }
} }