Increase test coverage

BastianBlokland · Jul 23, 2019 · 4f0e866 · 4f0e866
1 parent bd83ba0
commit 4f0e866
Showing 1 changed file with 29 additions and 0 deletions.
diff --git a/tests/playmode/ComponentExtensionsTests.cs b/tests/playmode/ComponentExtensionsTests.cs
@@ -240,7 +240,36 @@ public IEnumerator RunnerDestroysItselfWhenComponentIsDestroyed()
             Assert.AreEqual(0, go.GetComponents<MonoBehaviour>().Length);
 
             // Cleanup.
+            Object.Destroy(go);
+
+            async Task TestAsync()
+            {
+                await Task.Yield();
+                await Task.Yield();
+            }
+        }
+
+        [UnityTest]
+        public IEnumerator RunnerDestroysItselfWhenDisabledComponentIsDestroyed()
+        {
+            var go = new GameObject("TestGameObject");
+            var comp = go.AddComponent<MockComponent>();
+            comp.StartTask(TestAsync);
+
+            // Disable component.
+            comp.enabled = false;
+
+            // Assert that runner was created.
+            Assert.AreEqual(2, go.GetComponents<MonoBehaviour>().Length);
+
+            // Destroy component.
+            Object.DestroyImmediate(comp);
             yield return null;
+
+            // Assert that runner has destroyed itself.
+            Assert.AreEqual(0, go.GetComponents<MonoBehaviour>().Length);
+
+            // Cleanup.
             Object.Destroy(go);
 
             async Task TestAsync()