121 lines
4.0 KiB
C#
121 lines
4.0 KiB
C#
using System.Linq;
|
|
using System.Threading.Tasks;
|
|
using Content.Shared.Containers.ItemSlots;
|
|
using Content.Shared.Whitelist;
|
|
using NUnit.Framework;
|
|
using Robust.Shared.GameObjects;
|
|
using Robust.Shared.IoC;
|
|
using Robust.Shared.Map;
|
|
|
|
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: ItemSlots
|
|
slots:
|
|
slotName:
|
|
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 = GetMainMapId(mapManager);
|
|
var mapCoordinates = new MapCoordinates(0, 0, mapId);
|
|
|
|
IEntity tempQualifier = entityManager.SpawnEntity("ValidComponentDummy", mapCoordinates);
|
|
var validComponent = (EntityUid) tempQualifier;
|
|
IEntity tempQualifier1 = entityManager.SpawnEntity("ValidTagDummy", mapCoordinates);
|
|
var validTag = (EntityUid) tempQualifier1;
|
|
|
|
IEntity tempQualifier2 = entityManager.SpawnEntity("InvalidComponentDummy", mapCoordinates);
|
|
var invalidComponent = (EntityUid) tempQualifier2;
|
|
IEntity tempQualifier3 = entityManager.SpawnEntity("InvalidTagDummy", mapCoordinates);
|
|
var invalidTag = (EntityUid) tempQualifier3;
|
|
|
|
// 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 = IoCManager.Resolve<IEntityManager>().GetComponent<ItemSlotsComponent>(dummy).Slots.Values.First().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);
|
|
});
|
|
}
|
|
}
|
|
}
|