From: Leon Friedrich <60421075+ElectroJr@users.noreply.github.com>
Date: Mon, 3 Jun 2024 05:21:46 +0000 (+1200)
Subject: Add PrototypeUploadTest (#28522)
X-Git-Url: https://git.smokeofanarchy.ru/gitweb.cgi?a=commitdiff_plain;h=d36ff569ab65692dffdc8a0ac8c7d8b8655bff46;p=space-station-14.git

Add PrototypeUploadTest (#28522)
---

diff --git a/Content.IntegrationTests/Tests/PrototypeTests/PrototypeUploadTest.cs b/Content.IntegrationTests/Tests/PrototypeTests/PrototypeUploadTest.cs
new file mode 100644
index 0000000000..c641cd9bd1
--- /dev/null
+++ b/Content.IntegrationTests/Tests/PrototypeTests/PrototypeUploadTest.cs
@@ -0,0 +1,85 @@
+using Content.Shared.Tag;
+using Robust.Client.Upload.Commands;
+using Robust.Shared.GameObjects;
+using Robust.Shared.Prototypes;
+using Robust.Shared.Upload;
+
+namespace Content.IntegrationTests.Tests.PrototypeTests;
+
+public sealed class PrototypeUploadTest
+{
+    public const string IdA = "UploadTestPrototype";
+    public const string IdB = $"{IdA}NoParent";
+    public const string IdC = $"{IdA}Abstract";
+    public const string IdD = $"{IdA}UploadedParent";
+
+    private const string File = $@"
+- type: entity
+  parent: BaseStructure # BaseItem can cause AllItemsHaveSpritesTest to fail
+  id: {IdA}
+
+- type: entity
+  id: {IdB}
+
+- type: entity
+  id: {IdC}
+  abstract: true
+  components:
+  - type: Tag
+
+- type: entity
+  id: {IdD}
+  parent: {IdC}
+";
+
+    [Test]
+    [TestOf(typeof(LoadPrototypeCommand))]
+    public async Task TestFileUpload()
+    {
+        await using var pair = await PoolManager.GetServerClient(new PoolSettings {Connected = true});
+        var sCompFact = pair.Server.ResolveDependency<IComponentFactory>();
+        var cCompFact = pair.Client.ResolveDependency<IComponentFactory>();
+
+        Assert.That(!pair.Server.ProtoMan.TryIndex<EntityPrototype>(IdA, out _));
+        Assert.That(!pair.Server.ProtoMan.TryIndex<EntityPrototype>(IdB, out _));
+        Assert.That(!pair.Server.ProtoMan.TryIndex<EntityPrototype>(IdC, out _));
+        Assert.That(!pair.Server.ProtoMan.TryIndex<EntityPrototype>(IdD, out _));
+
+        Assert.That(!pair.Client.ProtoMan.TryIndex<EntityPrototype>(IdA, out _));
+        Assert.That(!pair.Client.ProtoMan.TryIndex<EntityPrototype>(IdB, out _));
+        Assert.That(!pair.Client.ProtoMan.TryIndex<EntityPrototype>(IdC, out _));
+        Assert.That(!pair.Client.ProtoMan.TryIndex<EntityPrototype>(IdD, out _));
+
+        var protoLoad = pair.Client.ResolveDependency<IGamePrototypeLoadManager>();
+        await pair.Client.WaitPost(() => protoLoad.SendGamePrototype(File));
+        await pair.RunTicksSync(10);
+
+        Assert.That(pair.Server.ProtoMan.TryIndex<EntityPrototype>(IdA, out var sProtoA));
+        Assert.That(pair.Server.ProtoMan.TryIndex<EntityPrototype>(IdB, out var sProtoB));
+        Assert.That(!pair.Server.ProtoMan.TryIndex<EntityPrototype>(IdC, out _));
+        Assert.That(pair.Server.ProtoMan.TryIndex<EntityPrototype>(IdD, out var sProtoD));
+
+        Assert.That(pair.Client.ProtoMan.TryIndex<EntityPrototype>(IdA, out var cProtoA));
+        Assert.That(pair.Client.ProtoMan.TryIndex<EntityPrototype>(IdB, out var cProtoB));
+        Assert.That(!pair.Client.ProtoMan.TryIndex<EntityPrototype>(IdC, out _));
+        Assert.That(pair.Client.ProtoMan.TryIndex<EntityPrototype>(IdD, out var cProtoD));
+
+        // Arbitrarily choosing TagComponent to check that inheritance works for uploaded prototypes.
+
+        await pair.Server.WaitPost(() =>
+        {
+            Assert.That(sProtoA!.TryGetComponent<TagComponent>(out _, sCompFact), Is.True);
+            Assert.That(sProtoB!.TryGetComponent<TagComponent>(out _, sCompFact), Is.False);
+            Assert.That(sProtoD!.TryGetComponent<TagComponent>(out _, sCompFact), Is.True);
+        });
+
+        await pair.Client.WaitPost(() =>
+        {
+            Assert.That(cProtoA!.TryGetComponent<TagComponent>(out _, cCompFact), Is.True);
+            Assert.That(cProtoB!.TryGetComponent<TagComponent>(out _, cCompFact), Is.False);
+            Assert.That(cProtoD!.TryGetComponent<TagComponent>(out _, cCompFact), Is.True);
+        });
+
+        await pair.CleanReturnAsync();
+    }
+}