ECSatize AlertsSystem (#5559)
This commit is contained in:
@@ -1,70 +0,0 @@
|
||||
using System.IO;
|
||||
using Content.Server.Alert;
|
||||
using Content.Shared.Alert;
|
||||
using NUnit.Framework;
|
||||
using Robust.Shared.GameObjects;
|
||||
using Robust.Shared.IoC;
|
||||
using Robust.Shared.Prototypes;
|
||||
using Robust.Shared.Serialization.Manager;
|
||||
|
||||
namespace Content.Tests.Server.GameObjects.Components.Mobs
|
||||
{
|
||||
[TestFixture]
|
||||
[TestOf(typeof(ServerAlertsComponent))]
|
||||
public class ServerAlertsComponentTests : ContentUnitTest
|
||||
{
|
||||
const string PROTOTYPES = @"
|
||||
- type: alert
|
||||
name: AlertLowPressure
|
||||
alertType: LowPressure
|
||||
category: Pressure
|
||||
icon: /Textures/Interface/Alerts/Pressure/lowpressure.png
|
||||
|
||||
- type: alert
|
||||
name: AlertHighPressure
|
||||
alertType: HighPressure
|
||||
category: Pressure
|
||||
icon: /Textures/Interface/Alerts/Pressure/highpressure.png
|
||||
";
|
||||
|
||||
[Test]
|
||||
public void ShowAlerts()
|
||||
{
|
||||
// this is kind of unnecessary because there's integration test coverage of Alert components
|
||||
// but wanted to keep it anyway to see what's possible w.r.t. testing components
|
||||
// in a unit test
|
||||
|
||||
IoCManager.Resolve<ISerializationManager>().Initialize();
|
||||
var prototypeManager = IoCManager.Resolve<IPrototypeManager>();
|
||||
prototypeManager.Initialize();
|
||||
var factory = IoCManager.Resolve<IComponentFactory>();
|
||||
factory.RegisterClass<ServerAlertsComponent>();
|
||||
prototypeManager.LoadFromStream(new StringReader(PROTOTYPES));
|
||||
prototypeManager.Resync();
|
||||
var alertManager = IoCManager.Resolve<AlertManager>();
|
||||
alertManager.Initialize();
|
||||
|
||||
|
||||
var alertsComponent = new ServerAlertsComponent();
|
||||
alertsComponent = IoCManager.InjectDependencies(alertsComponent);
|
||||
|
||||
Assert.That(alertManager.TryGet(AlertType.LowPressure, out var lowpressure));
|
||||
Assert.That(alertManager.TryGet(AlertType.HighPressure, out var highpressure));
|
||||
|
||||
alertsComponent.ShowAlert(AlertType.LowPressure);
|
||||
var alertState = alertsComponent.GetComponentState() as AlertsComponentState;
|
||||
Assert.NotNull(alertState);
|
||||
Assert.That(alertState.Alerts.Count, Is.EqualTo(1));
|
||||
Assert.That(alertState.Alerts.ContainsKey(lowpressure.AlertKey));
|
||||
|
||||
alertsComponent.ShowAlert(AlertType.HighPressure);
|
||||
alertState = alertsComponent.GetComponentState() as AlertsComponentState;
|
||||
Assert.That(alertState.Alerts.Count, Is.EqualTo(1));
|
||||
Assert.That(alertState.Alerts.ContainsKey(highpressure.AlertKey));
|
||||
|
||||
alertsComponent.ClearAlertCategory(AlertCategory.Pressure);
|
||||
alertState = alertsComponent.GetComponentState() as AlertsComponentState;
|
||||
Assert.That(alertState.Alerts.Count, Is.EqualTo(0));
|
||||
}
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user