108 lines
4.4 KiB
C#
108 lines
4.4 KiB
C#
using System.Linq;
|
|
using System.Threading.Tasks;
|
|
using Content.Client.Alerts;
|
|
using Content.Client.Alerts.UI;
|
|
using Content.Client.UserInterface;
|
|
using Content.Server.Alert;
|
|
using Content.Shared.Alert;
|
|
using NUnit.Framework;
|
|
using Robust.Client.UserInterface;
|
|
using IPlayerManager = Robust.Server.Player.IPlayerManager;
|
|
|
|
namespace Content.IntegrationTests.Tests.GameObjects.Components.Mobs
|
|
{
|
|
[TestFixture]
|
|
[TestOf(typeof(ClientAlertsComponent))]
|
|
[TestOf(typeof(ServerAlertsComponent))]
|
|
public class AlertsComponentTests : ContentIntegrationTest
|
|
{
|
|
[Test]
|
|
public async Task AlertsTest()
|
|
{
|
|
var (client, server) = await StartConnectedServerClientPair();
|
|
|
|
await server.WaitIdleAsync();
|
|
await client.WaitIdleAsync();
|
|
|
|
var serverPlayerManager = server.ResolveDependency<IPlayerManager>();
|
|
|
|
await server.WaitAssertion(() =>
|
|
{
|
|
var player = serverPlayerManager.GetAllPlayers().Single();
|
|
var playerEnt = player.AttachedEntity;
|
|
Assert.NotNull(playerEnt);
|
|
var alertsComponent = playerEnt.GetComponent<ServerAlertsComponent>();
|
|
Assert.NotNull(alertsComponent);
|
|
|
|
// show 2 alerts
|
|
alertsComponent.ShowAlert(AlertType.Debug1);
|
|
alertsComponent.ShowAlert(AlertType.Debug2);
|
|
});
|
|
|
|
await server.WaitRunTicks(5);
|
|
await client.WaitRunTicks(5);
|
|
|
|
var clientPlayerMgr = client.ResolveDependency<Robust.Client.Player.IPlayerManager>();
|
|
var clientUIMgr = client.ResolveDependency<IUserInterfaceManager>();
|
|
await client.WaitAssertion(() =>
|
|
{
|
|
|
|
var local = clientPlayerMgr.LocalPlayer;
|
|
Assert.NotNull(local);
|
|
var controlled = local.ControlledEntity;
|
|
Assert.NotNull(controlled);
|
|
var alertsComponent = controlled.GetComponent<ClientAlertsComponent>();
|
|
Assert.NotNull(alertsComponent);
|
|
|
|
// find the alertsui
|
|
var alertsUI =
|
|
clientUIMgr.StateRoot.Children.FirstOrDefault(c => c is AlertsUI) as AlertsUI;
|
|
Assert.NotNull(alertsUI);
|
|
|
|
// we should be seeing 3 alerts - our health, and the 2 debug alerts, in a specific order.
|
|
Assert.That(alertsUI.AlertContainer.ChildCount, Is.GreaterThanOrEqualTo(3));
|
|
var alertControls = alertsUI.AlertContainer.Children.Select(c => (AlertControl) c);
|
|
var alertIDs = alertControls.Select(ac => ac.Alert.AlertType).ToArray();
|
|
var expectedIDs = new [] {AlertType.HumanHealth, AlertType.Debug1, AlertType.Debug2};
|
|
Assert.That(alertIDs, Is.SupersetOf(expectedIDs));
|
|
});
|
|
|
|
await server.WaitAssertion(() =>
|
|
{
|
|
var player = serverPlayerManager.GetAllPlayers().Single();
|
|
var playerEnt = player.AttachedEntity;
|
|
Assert.NotNull(playerEnt);
|
|
var alertsComponent = playerEnt.GetComponent<ServerAlertsComponent>();
|
|
Assert.NotNull(alertsComponent);
|
|
|
|
alertsComponent.ClearAlert(AlertType.Debug1);
|
|
});
|
|
await server.WaitRunTicks(5);
|
|
await client.WaitRunTicks(5);
|
|
|
|
await client.WaitAssertion(() =>
|
|
{
|
|
|
|
var local = clientPlayerMgr.LocalPlayer;
|
|
Assert.NotNull(local);
|
|
var controlled = local.ControlledEntity;
|
|
Assert.NotNull(controlled);
|
|
var alertsComponent = controlled.GetComponent<ClientAlertsComponent>();
|
|
Assert.NotNull(alertsComponent);
|
|
|
|
// find the alertsui
|
|
var alertsUI =
|
|
clientUIMgr.StateRoot.Children.FirstOrDefault(c => c is AlertsUI) as AlertsUI;
|
|
Assert.NotNull(alertsUI);
|
|
|
|
// we should be seeing 2 alerts now because one was cleared
|
|
Assert.That(alertsUI.AlertContainer.ChildCount, Is.GreaterThanOrEqualTo(2));
|
|
var alertControls = alertsUI.AlertContainer.Children.Select(c => (AlertControl) c);
|
|
var alertIDs = alertControls.Select(ac => ac.Alert.AlertType).ToArray();
|
|
var expectedIDs = new [] {AlertType.HumanHealth, AlertType.Debug2};
|
|
Assert.That(alertIDs, Is.SupersetOf(expectedIDs));
|
|
});
|
|
}
|
|
}
|
|
}
|