Files
Workspace/Assets/Demos/Demo2/Tests/PlayMode/Demo2PlayModeTests.cs
T
2026-08-18 02:03:34 +08:00

65 lines
2.6 KiB
C#

#nullable enable
using System.Collections;
using System.Reflection;
using Demo2.Domain;
using Demo2.ECS;
using Demo2.Runtime;
using NUnit.Framework;
using Unity.Entities;
using UnityEngine;
using UnityEngine.SceneManagement;
using UnityEngine.TestTools;
using UnityEngine.UIElements;
namespace Demo2.Tests
{
public sealed class Demo2PlayModeTests
{
[UnityTest]
public IEnumerator SceneEntry_OpensTitleAndStartsTheRealFactoryScreen()
{
yield return SceneManager.LoadSceneAsync("Demo2", LoadSceneMode.Single);
yield return null;
var controller = Object.FindObjectOfType<Demo2DemoController>();
Assert.That(controller, Is.Not.Null);
var document = controller!.GetComponentInChildren<UIDocument>();
Assert.That(document, Is.Not.Null);
var solo = document!.rootVisualElement.Q<Button>("solo-button");
Assert.That(solo, Is.Not.Null);
var simulateClick = typeof(Clickable).GetMethod("SimulateSingleClick", BindingFlags.Instance | BindingFlags.NonPublic);
Assert.That(simulateClick, Is.Not.Null);
simulateClick!.Invoke(solo.clickable, new object?[] { null, 0 });
yield return null;
var factory = document.rootVisualElement.Q<VisualElement>("factory-screen");
Assert.That(factory.resolvedStyle.display, Is.EqualTo(DisplayStyle.Flex));
Assert.That(Demo2GameService.Current?.Phase, Is.EqualTo(Demo2Phase.Build));
}
[UnityTest]
public IEnumerator CompleteFiveCycleFlow_ReachesVictory()
{
using var service = new Demo2GameService();
service.StartNewGame(20260817);
for (var cycle = 1; cycle <= Demo2Protocol.CycleCount; cycle++)
{
service.SetReady("test", true);
var world = service.EcsWorld!;
var economy = world.EntityManager.GetComponentData<Demo2EconomyState>(world.Singleton);
economy.Payload = 1_000_000;
world.EntityManager.SetComponentData(world.Singleton, economy);
service.StartWaveImmediately();
for (var frame = 0; frame < 202; frame++) service.AdvanceFrame(0.1f);
Assert.That(service.Phase, Is.EqualTo(Demo2Phase.Reward), "cycle " + cycle);
Assert.That(service.RewardChoices.Count, Is.EqualTo(3));
Assert.That(service.ChooseReward(service.RewardChoices[0]), Is.True);
yield return null;
}
Assert.That(service.Phase, Is.EqualTo(Demo2Phase.Victory));
}
}
}