Refactor ExtinguisherCabinet->ItemCabinet and actually maps them in, adds EntityWhitelist (#4154)
* i probably shouldnt have done this in one commit * map nonsense * fix example code * unnecessary * test * reviews * little fix for open datafield * add soul
This commit is contained in:
121
Content.IntegrationTests/Tests/Utility/EntityWhitelistTest.cs
Normal file
121
Content.IntegrationTests/Tests/Utility/EntityWhitelistTest.cs
Normal file
@@ -0,0 +1,121 @@
|
||||
using System.IO;
|
||||
using System.Threading.Tasks;
|
||||
using Content.Server.GameObjects.Components;
|
||||
using Content.Server.GameObjects.Components.Items.Storage;
|
||||
using Content.Shared.GameObjects.Components.Tag;
|
||||
using Content.Shared.Prototypes;
|
||||
using Content.Shared.Utility;
|
||||
using NUnit.Framework;
|
||||
using NUnit.Framework.Internal;
|
||||
using Robust.Shared.GameObjects;
|
||||
using Robust.Shared.IoC;
|
||||
using Robust.Shared.Map;
|
||||
using Robust.Shared.Maths;
|
||||
using Robust.Shared.Prototypes;
|
||||
using Robust.Shared.Serialization.Manager;
|
||||
|
||||
namespace Content.IntegrationTests.Tests.Utility
|
||||
{
|
||||
[TestFixture]
|
||||
[TestOf(typeof(EntityWhitelist))]
|
||||
public class EntityWhitelistTest : ContentIntegrationTest
|
||||
{
|
||||
private const string InvalidComponent = "Sprite";
|
||||
private const string ValidComponent = "Physics";
|
||||
|
||||
private static readonly string Prototypes = $@"
|
||||
- type: Tag
|
||||
id: ValidTag
|
||||
- type: Tag
|
||||
id: InvalidTag
|
||||
|
||||
- type: entity
|
||||
id: WhitelistDummy
|
||||
components:
|
||||
- type: ItemCabinet
|
||||
whitelist:
|
||||
prototypes:
|
||||
- ValidPrototypeDummy
|
||||
components:
|
||||
- {ValidComponent}
|
||||
tags:
|
||||
- ValidTag
|
||||
|
||||
- type: entity
|
||||
id: InvalidComponentDummy
|
||||
components:
|
||||
- type: {InvalidComponent}
|
||||
- type: entity
|
||||
id: InvalidTagDummy
|
||||
components:
|
||||
- type: Tag
|
||||
tags:
|
||||
- InvalidTag
|
||||
|
||||
- type: entity
|
||||
id: ValidComponentDummy
|
||||
components:
|
||||
- type: {ValidComponent}
|
||||
- type: entity
|
||||
id: ValidTagDummy
|
||||
components:
|
||||
- type: Tag
|
||||
tags:
|
||||
- ValidTag";
|
||||
|
||||
[Test]
|
||||
public async Task Test()
|
||||
{
|
||||
var serverOptions = new ServerContentIntegrationOption {ExtraPrototypes = Prototypes};
|
||||
var server = StartServer(serverOptions);
|
||||
|
||||
await server.WaitIdleAsync();
|
||||
var mapManager = server.ResolveDependency<IMapManager>();
|
||||
var entityManager = server.ResolveDependency<IEntityManager>();
|
||||
|
||||
await server.WaitAssertion(() =>
|
||||
{
|
||||
var mapId = new MapId(1);
|
||||
var mapCoordinates = new MapCoordinates(0, 0, mapId);
|
||||
|
||||
var validComponent = entityManager.SpawnEntity("ValidComponentDummy", mapCoordinates);
|
||||
var validTag = entityManager.SpawnEntity("ValidTagDummy", mapCoordinates);
|
||||
|
||||
var invalidComponent = entityManager.SpawnEntity("InvalidComponentDummy", mapCoordinates);
|
||||
var invalidTag = entityManager.SpawnEntity("InvalidTagDummy", mapCoordinates);
|
||||
|
||||
// Test instantiated on its own
|
||||
var whitelistInst = new EntityWhitelist
|
||||
{
|
||||
Components = new[] {$"{ValidComponent}"},
|
||||
Tags = new[] {"ValidTag"}
|
||||
};
|
||||
whitelistInst.UpdateRegistrations();
|
||||
Assert.That(whitelistInst, Is.Not.Null);
|
||||
|
||||
Assert.That(whitelistInst.Components, Is.Not.Null);
|
||||
Assert.That(whitelistInst.Tags, Is.Not.Null);
|
||||
|
||||
Assert.That(whitelistInst.IsValid(validComponent), Is.True);
|
||||
Assert.That(whitelistInst.IsValid(validTag), Is.True);
|
||||
|
||||
Assert.That(whitelistInst.IsValid(invalidComponent), Is.False);
|
||||
Assert.That(whitelistInst.IsValid(invalidTag), Is.False);
|
||||
|
||||
// Test from serialized
|
||||
var dummy = entityManager.SpawnEntity("WhitelistDummy", mapCoordinates);
|
||||
var whitelistSer = dummy.GetComponent<ItemCabinetComponent>().Whitelist;
|
||||
Assert.That(whitelistSer, Is.Not.Null);
|
||||
|
||||
Assert.That(whitelistSer.Components, Is.Not.Null);
|
||||
Assert.That(whitelistSer.Tags, Is.Not.Null);
|
||||
|
||||
Assert.That(whitelistSer.IsValid(validComponent), Is.True);
|
||||
Assert.That(whitelistSer.IsValid(validTag), Is.True);
|
||||
|
||||
Assert.That(whitelistSer.IsValid(invalidComponent), Is.False);
|
||||
Assert.That(whitelistSer.IsValid(invalidTag), Is.False);
|
||||
});
|
||||
}
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user