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 sealed class ServerAlertsComponentTests : ContentUnitTest { const string PROTOTYPES = @" - type: alert id: LowPressure category: Pressure icon: /Textures/Interface/Alerts/Pressure/lowpressure.png - type: alert id: 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().Initialize(); var prototypeManager = IoCManager.Resolve(); prototypeManager.Initialize(); var factory = IoCManager.Resolve(); factory.RegisterClass(); prototypeManager.LoadFromStream(new StringReader(PROTOTYPES)); prototypeManager.ResolveResults(); var entSys = IoCManager.Resolve(); entSys.LoadExtraSystemType(); var alertsComponent = new AlertsComponent(); alertsComponent = IoCManager.InjectDependencies(alertsComponent); Assert.That(EntitySystem.Get().TryGet(AlertType.LowPressure, out var lowpressure)); Assert.That(EntitySystem.Get().TryGet(AlertType.HighPressure, out var highpressure)); EntitySystem.Get().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().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().ClearAlertCategory(alertsComponent.Owner, AlertCategory.Pressure); alertState = alertsComponent.GetComponentState() as AlertsComponentState; Assert.That(alertState.Alerts.Count, Is.EqualTo(0)); } } }