demo2

Assets/Demos/Demo2/Tests/PlayMode/Demo2PlayModeTests.cs

@@ -0,0 +1,64 @@
+#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));
+        }
+    }
+}