ECSatize AlertsSystem (#5559)
This commit is contained in:
@@ -1,14 +1,17 @@
|
||||
using System.IO;
|
||||
using Content.Shared.Alert;
|
||||
using NUnit.Framework;
|
||||
using NUnit.Framework.Interfaces;
|
||||
using Robust.Shared.GameObjects;
|
||||
using Robust.Shared.IoC;
|
||||
using Robust.Shared.Prototypes;
|
||||
using Robust.Shared.Reflection;
|
||||
using Robust.Shared.Serialization.Manager;
|
||||
using Robust.Shared.Utility;
|
||||
|
||||
namespace Content.Tests.Shared.Alert
|
||||
{
|
||||
[TestFixture, TestOf(typeof(AlertManager))]
|
||||
[TestFixture, TestOf(typeof(AlertsSystem))]
|
||||
public class AlertManagerTests : ContentUnitTest
|
||||
{
|
||||
const string PROTOTYPES = @"
|
||||
@@ -24,23 +27,26 @@ namespace Content.Tests.Shared.Alert
|
||||
";
|
||||
|
||||
[Test]
|
||||
[Ignore("There is no way to load extra Systems in a unit test, fixing RobustUnitTest is out of scope.")]
|
||||
public void TestAlertManager()
|
||||
{
|
||||
IoCManager.Resolve<ISerializationManager>().Initialize();
|
||||
|
||||
var reflection = IoCManager.Resolve<IReflectionManager>();
|
||||
reflection.LoadAssemblies();
|
||||
|
||||
var prototypeManager = IoCManager.Resolve<IPrototypeManager>();
|
||||
prototypeManager.Initialize();
|
||||
prototypeManager.LoadFromStream(new StringReader(PROTOTYPES));
|
||||
var alertManager = IoCManager.Resolve<AlertManager>();
|
||||
alertManager.Initialize();
|
||||
|
||||
Assert.That(alertManager.TryGet(AlertType.LowPressure, out var lowPressure));
|
||||
Assert.That(EntitySystem.Get<AlertsSystem>().TryGet(AlertType.LowPressure, out var lowPressure));
|
||||
Assert.That(lowPressure.Icon, Is.EqualTo(new SpriteSpecifier.Texture(new ResourcePath("/Textures/Interface/Alerts/Pressure/lowpressure.png"))));
|
||||
Assert.That(alertManager.TryGet(AlertType.HighPressure, out var highPressure));
|
||||
Assert.That(EntitySystem.Get<AlertsSystem>().TryGet(AlertType.HighPressure, out var highPressure));
|
||||
Assert.That(highPressure.Icon, Is.EqualTo(new SpriteSpecifier.Texture(new ResourcePath("/Textures/Interface/Alerts/Pressure/highpressure.png"))));
|
||||
|
||||
Assert.That(alertManager.TryGet(AlertType.LowPressure, out lowPressure));
|
||||
Assert.That(EntitySystem.Get<AlertsSystem>().TryGet(AlertType.LowPressure, out lowPressure));
|
||||
Assert.That(lowPressure.Icon, Is.EqualTo(new SpriteSpecifier.Texture(new ResourcePath("/Textures/Interface/Alerts/Pressure/lowpressure.png"))));
|
||||
Assert.That(alertManager.TryGet(AlertType.HighPressure, out highPressure));
|
||||
Assert.That(EntitySystem.Get<AlertsSystem>().TryGet(AlertType.HighPressure, out highPressure));
|
||||
Assert.That(highPressure.Icon, Is.EqualTo(new SpriteSpecifier.Texture(new ResourcePath("/Textures/Interface/Alerts/Pressure/highpressure.png"))));
|
||||
}
|
||||
}
|
||||
|
||||
72
Content.Tests/Shared/Alert/ServerAlertsComponentTests.cs
Normal file
72
Content.Tests/Shared/Alert/ServerAlertsComponentTests.cs
Normal file
@@ -0,0 +1,72 @@
|
||||
using System;
|
||||
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.Shared.Alert
|
||||
{
|
||||
[TestFixture]
|
||||
[TestOf(typeof(AlertsComponent))]
|
||||
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]
|
||||
[Ignore("There is no way to load extra Systems in a unit test, fixing RobustUnitTest is out of scope.")]
|
||||
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<AlertsComponent>();
|
||||
prototypeManager.LoadFromStream(new StringReader(PROTOTYPES));
|
||||
prototypeManager.Resync();
|
||||
|
||||
var entSys = IoCManager.Resolve<IEntitySystemManager>();
|
||||
entSys.LoadExtraSystemType<ServerAlertsSystem>();
|
||||
|
||||
var alertsComponent = new AlertsComponent();
|
||||
alertsComponent = IoCManager.InjectDependencies(alertsComponent);
|
||||
|
||||
Assert.That(EntitySystem.Get<AlertsSystem>().TryGet(AlertType.LowPressure, out var lowpressure));
|
||||
Assert.That(EntitySystem.Get<AlertsSystem>().TryGet(AlertType.HighPressure, out var highpressure));
|
||||
|
||||
EntitySystem.Get<AlertsSystem>().ShowAlert(alertsComponent.Owner, AlertType.LowPressure, null, null);
|
||||
var alertState = alertsComponent.GetComponentState() as AlertsComponentState;
|
||||
Assert.NotNull(alertState);
|
||||
Assert.That(alertState.Alerts.Count, Is.EqualTo(1));
|
||||
Assert.That(alertState.Alerts.ContainsKey(lowpressure.AlertKey));
|
||||
|
||||
EntitySystem.Get<AlertsSystem>().ShowAlert(alertsComponent.Owner, AlertType.HighPressure, null, null);
|
||||
alertState = alertsComponent.GetComponentState() as AlertsComponentState;
|
||||
Assert.That(alertState.Alerts.Count, Is.EqualTo(1));
|
||||
Assert.That(alertState.Alerts.ContainsKey(highpressure.AlertKey));
|
||||
|
||||
EntitySystem.Get<AlertsSystem>().ClearAlertCategory(alertsComponent.Owner, AlertCategory.Pressure);
|
||||
alertState = alertsComponent.GetComponentState() as AlertsComponentState;
|
||||
Assert.That(alertState.Alerts.Count, Is.EqualTo(0));
|
||||
}
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user