CatStation/Content.IntegrationTests/Tests/GameRules/SecretStartsTest.cs

using System.Linq;
using Content.IntegrationTests.Fixtures;
using Content.Server.GameTicking;
using Robust.Shared.GameObjects;
using Robust.Shared.Prototypes;

namespace Content.IntegrationTests.Tests.GameRules;

[TestFixture]
public sealed class SecretStartsTest : GameTest
{
    public override PoolSettings PoolSettings => new PoolSettings { Dirty = true };

    private static readonly EntProtoId SecretGameRule = "Secret";

    /// <summary>
    ///     Tests that when secret is started, all of the game rules it successfully adds are also started.
    /// </summary>
    [Test]
    public async Task TestSecretStarts()
    {
        var pair = Pair;

        var server = pair.Server;
        await server.WaitIdleAsync();
        var gameTicker = server.System<ServerGameTicker>();

        await server.WaitAssertion(() =>
        {
            // this mimics roundflow:
            // rules added, then round starts
            gameTicker.AddGameRule(SecretGameRule);
            gameTicker.StartGamePresetRules();
        });

        // Wait three ticks for any random update loops that might happen
        await server.WaitRunTicks(3);

        await server.WaitAssertion(() =>
        {
            Assert.That(gameTicker.GetAddedGameRules().Count(), Is.GreaterThan(1), $"No additional rules started by secret rule.");

            // End all rules
            gameTicker.ClearGameRules();
        });
    }
}