From 9b09361cc9a5b3ef1fb9cecf34e28f6401ae9def Mon Sep 17 00:00:00 2001
From: smoogipoo <smoogipoo@smgi.me>
Date: Wed, 12 May 2021 12:16:41 +0900
Subject: [PATCH] Add testable spectator streaming client

---
 .../Spectator/TestSpectatorStreamingClient.cs | 84 +++++++++++++++++++
 1 file changed, 84 insertions(+)
 create mode 100644 osu.Game/Tests/Visual/Spectator/TestSpectatorStreamingClient.cs

diff --git a/osu.Game/Tests/Visual/Spectator/TestSpectatorStreamingClient.cs b/osu.Game/Tests/Visual/Spectator/TestSpectatorStreamingClient.cs
new file mode 100644
index 0000000000..c177d3f399
--- /dev/null
+++ b/osu.Game/Tests/Visual/Spectator/TestSpectatorStreamingClient.cs
@@ -0,0 +1,84 @@
+// 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.Collections.Generic;
+using osu.Framework.Bindables;
+using osu.Framework.Utils;
+using osu.Game.Online;
+using osu.Game.Online.Spectator;
+using osu.Game.Replays.Legacy;
+using osu.Game.Scoring;
+
+namespace osu.Game.Tests.Visual.Spectator
+{
+    public class TestSpectatorStreamingClient : SpectatorStreamingClient
+    {
+        public new BindableList<int> PlayingUsers => (BindableList<int>)base.PlayingUsers;
+        private readonly HashSet<int> watchingUsers = new HashSet<int>();
+
+        private readonly Dictionary<int, int> userBeatmapDictionary = new Dictionary<int, int>();
+        private readonly Dictionary<int, bool> userSentStateDictionary = new Dictionary<int, bool>();
+
+        public TestSpectatorStreamingClient()
+            : base(new DevelopmentEndpointConfiguration())
+        {
+        }
+
+        public void StartPlay(int userId, int beatmapId)
+        {
+            userBeatmapDictionary[userId] = beatmapId;
+            sendState(userId, beatmapId);
+        }
+
+        public void EndPlay(int userId, int beatmapId)
+        {
+            ((ISpectatorClient)this).UserFinishedPlaying(userId, new SpectatorState
+            {
+                BeatmapID = beatmapId,
+                RulesetID = 0,
+            });
+
+            userSentStateDictionary[userId] = false;
+        }
+
+        public void SendFrames(int userId, int index, int count)
+        {
+            var frames = new List<LegacyReplayFrame>();
+
+            for (int i = index; i < index + count; i++)
+            {
+                var buttonState = i == index + count - 1 ? ReplayButtonState.None : ReplayButtonState.Left1;
+
+                frames.Add(new LegacyReplayFrame(i * 100, RNG.Next(0, 512), RNG.Next(0, 512), buttonState));
+            }
+
+            var bundle = new FrameDataBundle(new ScoreInfo(), frames);
+            ((ISpectatorClient)this).UserSentFrames(userId, bundle);
+
+            if (!userSentStateDictionary[userId])
+                sendState(userId, userBeatmapDictionary[userId]);
+        }
+
+        public override void WatchUser(int userId)
+        {
+            // When newly watching a user, the server sends the playing state immediately.
+            if (!watchingUsers.Contains(userId) && PlayingUsers.Contains(userId))
+                sendState(userId, userBeatmapDictionary[userId]);
+
+            base.WatchUser(userId);
+
+            watchingUsers.Add(userId);
+        }
+
+        private void sendState(int userId, int beatmapId)
+        {
+            ((ISpectatorClient)this).UserBeganPlaying(userId, new SpectatorState
+            {
+                BeatmapID = beatmapId,
+                RulesetID = 0,
+            });
+
+            userSentStateDictionary[userId] = true;
+        }
+    }
+}