--- /dev/null
+using Content.IntegrationTests.Tests.Movement;
+using Content.Server.NPC.HTN;
+using Content.Shared.Damage;
+using Content.Shared.FixedPoint;
+using Content.Shared.Item.ItemToggle;
+using Content.Shared.Item.ItemToggle.Components;
+using Content.Shared.Mobs;
+using Content.Shared.Mobs.Components;
+using Content.Shared.Mousetrap;
+using Robust.Shared.Maths;
+using Robust.Shared.Prototypes;
+
+namespace Content.IntegrationTests.Tests.Mousetrap;
+
+/// <summary>
+/// Spawns a mouse and a mousetrap.
+/// Makes the mouse cross the inactive mousetrap, then activates the trap and
+/// makes the mouse try to cross back over it.
+/// </summary>
+/// <remarks>
+/// Yep, every time the tests run, a virtual mouse dies. Sorry.
+/// </remarks>
+public sealed class MousetrapMouseMoveOverTest : MovementTest
+{
+ private static readonly EntProtoId MousetrapProtoId = "Mousetrap";
+ private static readonly EntProtoId MouseProtoId = "MobMouse";
+ protected override string PlayerPrototype => MouseProtoId.Id; // use a mouse as the player entity
+
+ [Test]
+ public async Task MouseMoveOverTest()
+ {
+ // Make sure the mouse doesn't have any AI active
+ await Server.WaitPost(() => SEntMan.RemoveComponent<HTNComponent>(SPlayer));
+
+ // Spawn a mouse trap
+ await SpawnTarget(MousetrapProtoId);
+ Assert.That(Delta(), Is.GreaterThan(0.5), "Mouse and mousetrap not in expected positions.");
+
+ Assert.That(HasComp<MousetrapComponent>(),
+ $"{MousetrapProtoId} does not have a MousetrapComponent. If you're refactoring, please update this test!");
+
+ Assert.That(TryComp<ItemToggleComponent>(out var itemToggleComp),
+ $"{MousetrapProtoId} does not have a ItemToggleComponent. If you're refactoring, please update this test!");
+ Assert.That(itemToggleComp.Activated, Is.False, "Mousetrap started active.");
+
+ // The mouse is spawned by the test before the atmosphere is added, so it has some barotrauma damage already
+ // TODO: fix this since it can have an impact on integration tests
+ Assert.That(SEntMan.TryGetComponent<DamageableComponent>(SPlayer, out var damageComp),
+ $"Player does not have a DamageableComponent.");
+ var startingDamage = damageComp.TotalDamage;
+
+ Assert.That(SEntMan.TryGetComponent<MobStateComponent>(SPlayer, out var mouseMobStateComp),
+ $"{MouseProtoId} does not have a MobStateComponent.");
+ Assert.That(mouseMobStateComp.CurrentState, Is.EqualTo(MobState.Alive), "Mouse was not alive when spawned.");
+
+ // Move mouse over the trap
+ await Move(DirectionFlag.East, 1f);
+
+ Assert.That(Delta(), Is.LessThan(0.5), "Mouse did not move over mousetrap.");
+
+ // Walking over an inactive trap does not trigger it
+ Assert.That(damageComp.TotalDamage, Is.LessThanOrEqualTo(startingDamage), "Mouse took damage from inactive trap!");
+ Assert.That(itemToggleComp.Activated, Is.False, "Mousetrap was activated.");
+
+ // Activate the trap
+ var itemToggleSystem = Server.System<ItemToggleSystem>();
+ await Server.WaitAssertion(() =>
+ {
+ Assert.That(itemToggleSystem.TrySetActive(STarget.Value, true), "Could not activate the mouse trap.");
+ });
+
+ await Move(DirectionFlag.West, 1f);
+ Assert.That(Delta(), Is.LessThan(0.1), "Mouse moved past active mousetrap.");
+
+ // Walking over an active trap triggers it
+ Assert.That(damageComp.TotalDamage, Is.GreaterThan(startingDamage), "Mouse did not take damage from active trap!");
+ Assert.That(itemToggleComp.Activated, Is.False, "Mousetrap was not deactivated after triggering.");
+ Assert.That(mouseMobStateComp.CurrentState, Is.EqualTo(MobState.Dead), "Mouse was not killed by trap.");
+ }
+}
+
+/// <summary>
+/// Spawns a mousetrap and makes the player walk over it without shoes.
+/// Gives the player some shoes and makes them walk back over the trap.
+/// </summary>
+public sealed class MousetrapHumanMoveOverTest : MovementTest
+{
+ private static readonly EntProtoId MousetrapProtoId = "Mousetrap";
+ private const string ShoesProtoId = "InteractionTestShoes";
+
+ [TestPrototypes]
+ private static readonly string TestPrototypes = $@"
+ - type: entity
+ parent: ClothingShoesBase
+ id: {ShoesProtoId}
+ components:
+ - type: Sprite
+ sprite: Clothing/Shoes/Boots/workboots.rsi
+ ";
+
+ [Test]
+ public async Task HumanMoveOverTest()
+ {
+ await SpawnTarget(MousetrapProtoId);
+
+ Assert.That(Delta(), Is.GreaterThan(0.5), "Player and mousetrap not in expected positions.");
+
+ Assert.That(HasComp<MousetrapComponent>(),
+ $"{MousetrapProtoId} does not have a MousetrapComponent. If you're refactoring, please update this test!");
+
+ Assert.That(TryComp<ItemToggleComponent>(out var itemToggleComp),
+ $"{MousetrapProtoId} does not have a ItemToggleComponent. If you're refactoring, please update this test!");
+
+ // Activate the trap
+ var itemToggleSystem = Server.System<ItemToggleSystem>();
+ await Server.WaitAssertion(() =>
+ {
+ Assert.That(itemToggleSystem.TrySetActive(STarget.Value, true), "Could not activate the mouse trap.");
+ });
+
+ Assert.That(SEntMan.TryGetComponent<DamageableComponent>(SPlayer, out var damageComp),
+ $"Player does not have a DamageableComponent.");
+ var startingDamage = damageComp.TotalDamage;
+
+ // Move player over the trap
+ await Move(DirectionFlag.East, 0.5f);
+
+ Assert.That(Delta(), Is.LessThan(0.5), "Player did not move over mousetrap.");
+
+ // Walking over the trap without shoes activates it
+ Assert.That(damageComp.TotalDamage, Is.GreaterThan(startingDamage), "Player did not take damage.");
+ Assert.That(itemToggleComp.Activated, Is.False, "Mousetrap was not deactivated after triggering.");
+
+ // Reactivate the trap
+ await Server.WaitAssertion(() =>
+ {
+ Assert.That(itemToggleSystem.TrySetActive(STarget.Value, true), "Could not activate the mouse trap.");
+ });
+ var afterStepDamage = damageComp.TotalDamage;
+
+ // Give the player some shoes
+ await PlaceInHands(ShoesProtoId);
+ // Thanks to quick-equip, using the shoes will wear them
+ await UseInHand();
+
+ // Move back over the trap
+ await Move(DirectionFlag.West, 1f);
+ Assert.That(Delta(), Is.GreaterThan(0.5), "Player did not move back over mousetrap.");
+
+ // Walking over the trap with shoes on does not activate it
+ Assert.That(damageComp.TotalDamage, Is.LessThanOrEqualTo(afterStepDamage), "Player took damage from trap!");
+ Assert.That(itemToggleComp.Activated, "Mousetrap was deactivated despite the player being protected by shoes.");
+ }
+}
projects / space-station-14.git / commitdiff