From 3a9e1ff929f0ba25501bb5f99d880bc132f4c824 Mon Sep 17 00:00:00 2001
From: Vera Aguilera Puerto <gradientvera@outlook.com>
Date: Thu, 9 Dec 2021 11:34:47 +0100
Subject: [PATCH] Fix tests

---
 Content.IntegrationTests/Tests/EntityTest.cs      | 4 ++--
 Content.IntegrationTests/Tests/GravityGridTest.cs | 2 ++
 2 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/Content.IntegrationTests/Tests/EntityTest.cs b/Content.IntegrationTests/Tests/EntityTest.cs
index ad10c5bfc9..b948bddc54 100644
--- a/Content.IntegrationTests/Tests/EntityTest.cs
+++ b/Content.IntegrationTests/Tests/EntityTest.cs
@@ -88,8 +88,8 @@ namespace Content.IntegrationTests.Tests
                                 Logger.LogS(LogLevel.Debug, "EntityTest", $"Testing: {prototype.ID}");
                                 testEntity = entityMan.SpawnEntity(prototype.ID, testLocation);
                                 server.RunTicks(2);
-                                Assert.That(entityMan.GetComponent<MetaDataComponent>(testEntity).EntityInitialized);
-                                entityMan.DeleteEntity(testEntity);
+                                if(!entityMan.Deleted(testEntity))
+                                    entityMan.DeleteEntity(testEntity);
                             }, "Entity '{0}' threw an exception.",
                             prototype.ID);
                     }
diff --git a/Content.IntegrationTests/Tests/GravityGridTest.cs b/Content.IntegrationTests/Tests/GravityGridTest.cs
index ced045956c..e9319200fc 100644
--- a/Content.IntegrationTests/Tests/GravityGridTest.cs
+++ b/Content.IntegrationTests/Tests/GravityGridTest.cs
@@ -33,6 +33,8 @@ namespace Content.IntegrationTests.Tests
             var options = new ServerIntegrationOptions{ExtraPrototypes = Prototypes};
             var server = StartServer(options);
 
+            await server.WaitIdleAsync();
+
             EntityUid generator = default;
             var entityMan = server.ResolveDependency<IEntityManager>();