Add restart round command test (#1897)

Content.IntegrationTests/Tests/Commands/RestartRoundTest.cs

@@ -0,0 +1,62 @@
+using System.Threading.Tasks;
+using Content.Server.GameTicking;
+using Content.Server.Interfaces.GameTicking;
+using NUnit.Framework;
+using Robust.Shared.Interfaces.Configuration;
+using Robust.Shared.Interfaces.GameObjects;
+using Robust.Shared.Timing;
+
+namespace Content.IntegrationTests.Tests.Commands
+{
+    [TestFixture]
+    [TestOf(typeof(NewRoundCommand))]
+    public class RestartRoundTest : ContentIntegrationTest
+    {
+        [Test]
+        [TestCase(true)]
+        [TestCase(false)]
+        public async Task RestartRoundAfterStart(bool lobbyEnabled)
+        {
+            var server = StartServer();
+
+            await server.WaitIdleAsync();
+
+            var gameTicker = server.ResolveDependency<IGameTicker>();
+            var configManager = server.ResolveDependency<IConfigurationManager>();
+            var entityManager = server.ResolveDependency<IEntityManager>();
+
+            await server.WaitRunTicks(30);
+
+            GameTick tickBeforeRestart = default;
+
+            server.Assert(() =>
+            {
+                configManager.SetCVar("game.lobbyenabled", lobbyEnabled);
+
+                Assert.That(gameTicker.RunLevel, Is.EqualTo(GameRunLevel.InRound));
+
+                tickBeforeRestart = entityManager.CurrentTick;
+
+                var command = new NewRoundCommand();
+                command.Execute(null, null, new string[] { });
+
+                if (lobbyEnabled)
+                {
+                    Assert.That(gameTicker.RunLevel, Is.Not.EqualTo(GameRunLevel.InRound));
+                }
+            });
+
+            await server.WaitIdleAsync();
+            await server.WaitRunTicks(5);
+
+            server.Assert(() =>
+            {
+                var tickAfterRestart = entityManager.CurrentTick;
+
+                Assert.That(tickBeforeRestart < tickAfterRestart);
+            });
+
+            await server.WaitRunTicks(60);
+        }
+    }
+}