feat(DependencyInjector): add RemoveInjector static method for testing

jonisavo · May 27, 2022 · e514533 · e514533
1 parent 6b90eb2
commit e514533
Show file tree

Hide file tree

Showing 2 changed files with 24 additions and 0 deletions.
diff --git a/Assets/UIComponents.Tests/DependencyInjectorStaticTests.cs b/Assets/UIComponents.Tests/DependencyInjectorStaticTests.cs
@@ -36,5 +36,19 @@ public void TearDown()
                 );
             }
         }
+
+        [TestFixture]
+        public class RemoveInjector
+        {
+            [Test]
+            public void Removes_The_Injector_From_The_Dependency_Injector()
+            {
+                var componentType = typeof(UIComponentWithDependency);
+                var injector = new DependencyInjector();
+                DependencyInjector.InjectorDictionary[componentType] = injector;
+                DependencyInjector.RemoveInjector(componentType);
+                Assert.That(DependencyInjector.InjectorDictionary.ContainsKey(componentType), Is.False);
+            }
+        }
     }
 }
diff --git a/Assets/UIComponents/Core/DependencyInjector.cs b/Assets/UIComponents/Core/DependencyInjector.cs
@@ -83,6 +83,16 @@ public static DependencyInjector GetInjector(Type consumerType)
             return CreateInjector(consumerType);
         }
 
+        /// <summary>
+        /// Removes the injector of the given consumer type.
+        /// Used primarily for testing.
+        /// </summary>
+        /// <param name="consumerType">Consumer type</param>
+        public static void RemoveInjector(Type consumerType)
+        {
+            InjectorDictionary.Remove(consumerType);
+        }
+
         private static DependencyInjector CreateInjector(Type consumerType)
         {
             var injectAttributes = (DependencyAttribute[])