From 69104a6e14e8772868ea04051313ac9ef6b18767 Mon Sep 17 00:00:00 2001
From: Simon Gockner <simon.gockner@gmail.com>
Date: Tue, 15 Oct 2019 14:27:32 +0200
Subject: [PATCH] close #22: return the registered `IRegistrationBase`

---
 .../Interfaces/IIocContainer.cs               | 27 ++++---
 LightweightIocContainer/IocContainer.cs       | 72 ++++++++++++++-----
 .../LightweightIocContainer.xml               | 18 +++++
 3 files changed, 90 insertions(+), 27 deletions(-)

diff --git a/LightweightIocContainer/Interfaces/IIocContainer.cs b/LightweightIocContainer/Interfaces/IIocContainer.cs
index d680ec4..be16027 100644
--- a/LightweightIocContainer/Interfaces/IIocContainer.cs
+++ b/LightweightIocContainer/Interfaces/IIocContainer.cs
@@ -27,14 +27,16 @@ namespace LightweightIocContainer.Interfaces
         /// <typeparam name="TInterface">The Interface to register</typeparam>
         /// <typeparam name="TImplementation">The Type that implements the interface</typeparam>
         /// <param name="lifestyle">The <see cref="Lifestyle"/> for this <see cref="IDefaultRegistration{TInterface}"/></param>
-        void Register<TInterface, TImplementation>(Lifestyle lifestyle = Lifestyle.Transient) where TImplementation : TInterface;
+        /// <returns>The created <see cref="IRegistrationBase"/></returns>
+        IDefaultRegistration<TInterface> Register<TInterface, TImplementation>(Lifestyle lifestyle = Lifestyle.Transient) where TImplementation : TInterface;
 
         /// <summary>
         /// Register a <see cref="Type"/> without an interface/>
         /// </summary>
         /// <typeparam name="TImplementation">The <see cref="Type"/> to register</typeparam>
         /// <param name="lifestyle">The <see cref="Lifestyle"/> for this <see cref="IDefaultRegistration{TInterface}"/></param>
-        void Register<TImplementation>(Lifestyle lifestyle = Lifestyle.Transient);
+        /// <returns>The created <see cref="IRegistrationBase"/></returns>
+        IDefaultRegistration<TImplementation> Register<TImplementation>(Lifestyle lifestyle = Lifestyle.Transient);
 
         /// <summary>
         /// Register an Interface with a Type that implements it as a multiton/>
@@ -42,7 +44,8 @@ namespace LightweightIocContainer.Interfaces
         /// <typeparam name="TInterface">The Interface to register</typeparam>
         /// <typeparam name="TImplementation">The Type that implements the interface</typeparam>
         /// <typeparam name="TScope">The Type of the multiton scope</typeparam>
-        void Register<TInterface, TImplementation, TScope>() where TImplementation : TInterface;
+        /// <returns>The created <see cref="IRegistrationBase"/></returns>
+        IMultitonRegistration<TInterface> Register<TInterface, TImplementation, TScope>() where TImplementation : TInterface;
 
         /// <summary>
         /// Register an Interface with a Type that implements it/>
@@ -50,14 +53,16 @@ namespace LightweightIocContainer.Interfaces
         /// <param name="tInterface">The Interface to register</param>
         /// <param name="tImplementation">The Type that implements the interface</param>
         /// <param name="lifestyle">The <see cref="Lifestyle"/> for this <see cref="IDefaultRegistration{TInterface}"/></param>
-        void Register(Type tInterface, Type tImplementation, Lifestyle lifestyle = Lifestyle.Transient);
+        /// <returns>The created <see cref="IRegistrationBase"/></returns>
+        IRegistrationBase Register(Type tInterface, Type tImplementation, Lifestyle lifestyle = Lifestyle.Transient);
 
         /// <summary>
         /// Register a <see cref="Type"/> without an interface/>
         /// </summary>
         /// <param name="tImplementation">The <see cref="Type"/> to register</param>
         /// <param name="lifestyle">The <see cref="Lifestyle"/> for this <see cref="IDefaultRegistration{TInterface}"/></param>
-        void Register(Type tImplementation, Lifestyle lifestyle = Lifestyle.Transient);
+        /// <returns>The created <see cref="IRegistrationBase"/></returns>
+        IRegistrationBase Register(Type tImplementation, Lifestyle lifestyle = Lifestyle.Transient);
 
         /// <summary>
         /// Register an Interface with a Type that implements it as a multiton/>
@@ -65,26 +70,30 @@ namespace LightweightIocContainer.Interfaces
         /// <param name="tInterface">The Interface to register</param>
         /// <param name="tImplementation">The Type that implements the interface</param>
         /// <param name="tScope">The Type of the multiton scope</param>
-        void Register(Type tInterface, Type tImplementation, Type tScope);
+        /// <returns>The created <see cref="IRegistrationBase"/></returns>
+        IRegistrationBase Register(Type tInterface, Type tImplementation, Type tScope);
 
         /// <summary>
         /// Register an Interface as an abstract typed factory/>
         /// </summary>
         /// <typeparam name="TFactory">The abstract typed factory to register</typeparam>
-        void RegisterFactory<TFactory>();
+        /// <returns>The created <see cref="IRegistrationBase"/></returns>
+        ITypedFactoryRegistration<TFactory> RegisterFactory<TFactory>();
 
         /// <summary>
         /// Register an Interface as an abstract typed factory/>
         /// </summary>
         /// <param name="tFactory">The abstract typed factory to register</param>
-        void RegisterFactory(Type tFactory);
+        /// <returns>The created <see cref="IRegistrationBase"/></returns>
+        IRegistrationBase RegisterFactory(Type tFactory);
 
         /// <summary>
         /// Register an Interface with an <see cref="ResolveCallback{T}"/> as a callback that is called when <see cref="Resolve{T}()"/> is called
         /// </summary>
         /// <typeparam name="TInterface">The Interface to register</typeparam>
         /// <param name="unitTestCallback">The <see cref="ResolveCallback{T}"/> for the callback</param>
-        void RegisterUnitTestCallback<TInterface>(ResolveCallback<TInterface> unitTestCallback);
+        /// <returns>The created <see cref="IRegistrationBase"/></returns>
+        IUnitTestCallbackRegistration<TInterface> RegisterUnitTestCallback<TInterface>(ResolveCallback<TInterface> unitTestCallback);
 
         /// <summary>
         /// Gets an instance of the given <see cref="Type"/>
diff --git a/LightweightIocContainer/IocContainer.cs b/LightweightIocContainer/IocContainer.cs
index f07bfbd..cb78a84 100644
--- a/LightweightIocContainer/IocContainer.cs
+++ b/LightweightIocContainer/IocContainer.cs
@@ -58,9 +58,13 @@ namespace LightweightIocContainer
         /// <typeparam name="TInterface">The Interface to register</typeparam>
         /// <typeparam name="TImplementation">The Type that implements the interface</typeparam>
         /// <param name="lifestyle">The <see cref="Lifestyle"/> for this <see cref="IDefaultRegistration{TInterface}"/></param>
-        public void Register<TInterface, TImplementation>(Lifestyle lifestyle = Lifestyle.Transient) where TImplementation : TInterface
+        /// <returns>The created <see cref="IRegistrationBase"/></returns>
+        public IDefaultRegistration<TInterface> Register<TInterface, TImplementation>(Lifestyle lifestyle = Lifestyle.Transient) where TImplementation : TInterface
         {
-            Register(_registrationFactory.Register<TInterface, TImplementation>(lifestyle));
+            IDefaultRegistration<TInterface> registration = _registrationFactory.Register<TInterface, TImplementation>(lifestyle);
+            Register(registration);
+
+            return registration;
         }
 
         /// <summary>
@@ -68,9 +72,13 @@ namespace LightweightIocContainer
         /// </summary>
         /// <typeparam name="TImplementation">The <see cref="Type"/> to register</typeparam>
         /// <param name="lifestyle">The <see cref="Lifestyle"/> for this <see cref="IDefaultRegistration{TInterface}"/></param>
-        public void Register<TImplementation>(Lifestyle lifestyle = Lifestyle.Transient)
+        /// <returns>The created <see cref="IRegistrationBase"/></returns>
+        public IDefaultRegistration<TImplementation> Register<TImplementation>(Lifestyle lifestyle = Lifestyle.Transient)
         {
-            Register(_registrationFactory.Register<TImplementation>(lifestyle));
+            IDefaultRegistration<TImplementation> registration = _registrationFactory.Register<TImplementation>(lifestyle);
+            Register(registration);
+
+            return registration;
         }
 
         /// <summary>
@@ -79,9 +87,13 @@ namespace LightweightIocContainer
         /// <typeparam name="TInterface">The Interface to register</typeparam>
         /// <typeparam name="TImplementation">The Type that implements the interface</typeparam>
         /// <typeparam name="TScope">The Type of the multiton scope</typeparam>
-        public void Register<TInterface, TImplementation, TScope>() where TImplementation : TInterface
+        /// <returns>The created <see cref="IRegistrationBase"/></returns>
+        public IMultitonRegistration<TInterface> Register<TInterface, TImplementation, TScope>() where TImplementation : TInterface
         {
-            Register(_registrationFactory.Register<TInterface, TImplementation, TScope>());
+            IMultitonRegistration<TInterface> registration = _registrationFactory.Register<TInterface, TImplementation, TScope>();
+            Register(registration);
+
+            return registration;
         }
 
         /// <summary>
@@ -90,9 +102,13 @@ namespace LightweightIocContainer
         /// <param name="tInterface">The Interface to register</param>
         /// <param name="tImplementation">The Type that implements the interface</param>
         /// <param name="lifestyle">The <see cref="Lifestyle"/> for this <see cref="IDefaultRegistration{TInterface}"/></param>
-        public void Register(Type tInterface, Type tImplementation, Lifestyle lifestyle = Lifestyle.Transient)
+        /// <returns>The created <see cref="IRegistrationBase"/></returns>
+        public IRegistrationBase Register(Type tInterface, Type tImplementation, Lifestyle lifestyle = Lifestyle.Transient)
         {
-            Register(_registrationFactory.Register(tInterface, tImplementation, lifestyle));
+            IRegistrationBase registration = _registrationFactory.Register(tInterface, tImplementation, lifestyle);
+            Register(registration);
+
+            return registration;
         }
 
         /// <summary>
@@ -100,9 +116,13 @@ namespace LightweightIocContainer
         /// </summary>
         /// <param name="tImplementation">The <see cref="Type"/> to register</param>
         /// <param name="lifestyle">The <see cref="Lifestyle"/> for this <see cref="IDefaultRegistration{TInterface}"/></param>
-        public void Register(Type tImplementation, Lifestyle lifestyle = Lifestyle.Transient)
+        /// <returns>The created <see cref="IRegistrationBase"/></returns>
+        public IRegistrationBase Register(Type tImplementation, Lifestyle lifestyle = Lifestyle.Transient)
         {
-            Register(_registrationFactory.Register(tImplementation, lifestyle));
+            IRegistrationBase registration = _registrationFactory.Register(tImplementation, lifestyle);
+            Register(registration);
+
+            return registration;
         }
 
         /// <summary>
@@ -111,27 +131,39 @@ namespace LightweightIocContainer
         /// <param name="tInterface">The Interface to register</param>
         /// <param name="tImplementation">The Type that implements the interface</param>
         /// <param name="tScope">The Type of the multiton scope</param>
-        public void Register(Type tInterface, Type tImplementation, Type tScope)
+        /// <returns>The created <see cref="IRegistrationBase"/></returns>
+        public IRegistrationBase Register(Type tInterface, Type tImplementation, Type tScope)
         {
-            Register(_registrationFactory.Register(tInterface, tImplementation, tScope));
+            IRegistrationBase registration = _registrationFactory.Register(tInterface, tImplementation, tScope);
+            Register(registration);
+
+            return registration;
         }
 
         /// <summary>
         /// Register an Interface as an abstract typed factory/>
         /// </summary>
         /// <typeparam name="TFactory">The abstract typed factory to register</typeparam>
-        public void RegisterFactory<TFactory>()
+        /// <returns>The created <see cref="IRegistrationBase"/></returns>
+        public ITypedFactoryRegistration<TFactory> RegisterFactory<TFactory>()
         {
-            Register(_registrationFactory.RegisterFactory<TFactory>());
+            ITypedFactoryRegistration<TFactory> registration = _registrationFactory.RegisterFactory<TFactory>();
+            Register(registration);
+
+            return registration;
         }
 
         /// <summary>
         /// Register an Interface as an abstract typed factory/>
         /// </summary>
         /// <param name="tFactory">The abstract typed factory to register</param>
-        public void RegisterFactory(Type tFactory)
+        /// <returns>The created <see cref="IRegistrationBase"/></returns>
+        public IRegistrationBase RegisterFactory(Type tFactory)
         {
-            Register(_registrationFactory.RegisterFactory(tFactory));
+            IRegistrationBase registration = _registrationFactory.RegisterFactory(tFactory);
+            Register(registration);
+
+            return registration;
         }
 
         /// <summary>
@@ -139,9 +171,13 @@ namespace LightweightIocContainer
         /// </summary>
         /// <typeparam name="TInterface">The Interface to register</typeparam>
         /// <param name="unitTestCallback">The <see cref="ResolveCallback{T}"/> for the callback</param>
-        public void RegisterUnitTestCallback<TInterface>(ResolveCallback<TInterface> unitTestCallback)
+        /// <returns>The created <see cref="IRegistrationBase"/></returns>
+        public IUnitTestCallbackRegistration<TInterface> RegisterUnitTestCallback<TInterface>(ResolveCallback<TInterface> unitTestCallback)
         {
-            Register(_registrationFactory.RegisterUnitTestCallback(unitTestCallback));
+            IUnitTestCallbackRegistration<TInterface> registration = _registrationFactory.RegisterUnitTestCallback(unitTestCallback);
+            Register(registration);
+
+            return registration;
         }
 
         /// <summary>
diff --git a/LightweightIocContainer/LightweightIocContainer.xml b/LightweightIocContainer/LightweightIocContainer.xml
index 66412a6..74fab0b 100644
--- a/LightweightIocContainer/LightweightIocContainer.xml
+++ b/LightweightIocContainer/LightweightIocContainer.xml
@@ -224,6 +224,7 @@
             <typeparam name="TInterface">The Interface to register</typeparam>
             <typeparam name="TImplementation">The Type that implements the interface</typeparam>
             <param name="lifestyle">The <see cref="T:LightweightIocContainer.Lifestyle"/> for this <see cref="T:LightweightIocContainer.Interfaces.Registrations.IDefaultRegistration`1"/></param>
+            <returns>The created <see cref="T:LightweightIocContainer.Interfaces.Registrations.IRegistrationBase"/></returns>
         </member>
         <member name="M:LightweightIocContainer.Interfaces.IIocContainer.Register``1(LightweightIocContainer.Lifestyle)">
             <summary>
@@ -231,6 +232,7 @@
             </summary>
             <typeparam name="TImplementation">The <see cref="T:System.Type"/> to register</typeparam>
             <param name="lifestyle">The <see cref="T:LightweightIocContainer.Lifestyle"/> for this <see cref="T:LightweightIocContainer.Interfaces.Registrations.IDefaultRegistration`1"/></param>
+            <returns>The created <see cref="T:LightweightIocContainer.Interfaces.Registrations.IRegistrationBase"/></returns>
         </member>
         <member name="M:LightweightIocContainer.Interfaces.IIocContainer.Register``3">
             <summary>
@@ -239,6 +241,7 @@
             <typeparam name="TInterface">The Interface to register</typeparam>
             <typeparam name="TImplementation">The Type that implements the interface</typeparam>
             <typeparam name="TScope">The Type of the multiton scope</typeparam>
+            <returns>The created <see cref="T:LightweightIocContainer.Interfaces.Registrations.IRegistrationBase"/></returns>
         </member>
         <member name="M:LightweightIocContainer.Interfaces.IIocContainer.Register(System.Type,System.Type,LightweightIocContainer.Lifestyle)">
             <summary>
@@ -247,6 +250,7 @@
             <param name="tInterface">The Interface to register</param>
             <param name="tImplementation">The Type that implements the interface</param>
             <param name="lifestyle">The <see cref="T:LightweightIocContainer.Lifestyle"/> for this <see cref="T:LightweightIocContainer.Interfaces.Registrations.IDefaultRegistration`1"/></param>
+            <returns>The created <see cref="T:LightweightIocContainer.Interfaces.Registrations.IRegistrationBase"/></returns>
         </member>
         <member name="M:LightweightIocContainer.Interfaces.IIocContainer.Register(System.Type,LightweightIocContainer.Lifestyle)">
             <summary>
@@ -254,6 +258,7 @@
             </summary>
             <param name="tImplementation">The <see cref="T:System.Type"/> to register</param>
             <param name="lifestyle">The <see cref="T:LightweightIocContainer.Lifestyle"/> for this <see cref="T:LightweightIocContainer.Interfaces.Registrations.IDefaultRegistration`1"/></param>
+            <returns>The created <see cref="T:LightweightIocContainer.Interfaces.Registrations.IRegistrationBase"/></returns>
         </member>
         <member name="M:LightweightIocContainer.Interfaces.IIocContainer.Register(System.Type,System.Type,System.Type)">
             <summary>
@@ -262,18 +267,21 @@
             <param name="tInterface">The Interface to register</param>
             <param name="tImplementation">The Type that implements the interface</param>
             <param name="tScope">The Type of the multiton scope</param>
+            <returns>The created <see cref="T:LightweightIocContainer.Interfaces.Registrations.IRegistrationBase"/></returns>
         </member>
         <member name="M:LightweightIocContainer.Interfaces.IIocContainer.RegisterFactory``1">
             <summary>
             Register an Interface as an abstract typed factory/>
             </summary>
             <typeparam name="TFactory">The abstract typed factory to register</typeparam>
+            <returns>The created <see cref="T:LightweightIocContainer.Interfaces.Registrations.IRegistrationBase"/></returns>
         </member>
         <member name="M:LightweightIocContainer.Interfaces.IIocContainer.RegisterFactory(System.Type)">
             <summary>
             Register an Interface as an abstract typed factory/>
             </summary>
             <param name="tFactory">The abstract typed factory to register</param>
+            <returns>The created <see cref="T:LightweightIocContainer.Interfaces.Registrations.IRegistrationBase"/></returns>
         </member>
         <member name="M:LightweightIocContainer.Interfaces.IIocContainer.RegisterUnitTestCallback``1(LightweightIocContainer.ResolveCallback{``0})">
             <summary>
@@ -281,6 +289,7 @@
             </summary>
             <typeparam name="TInterface">The Interface to register</typeparam>
             <param name="unitTestCallback">The <see cref="T:LightweightIocContainer.ResolveCallback`1"/> for the callback</param>
+            <returns>The created <see cref="T:LightweightIocContainer.Interfaces.Registrations.IRegistrationBase"/></returns>
         </member>
         <member name="M:LightweightIocContainer.Interfaces.IIocContainer.Resolve``1">
             <summary>
@@ -441,6 +450,7 @@
             <typeparam name="TInterface">The Interface to register</typeparam>
             <typeparam name="TImplementation">The Type that implements the interface</typeparam>
             <param name="lifestyle">The <see cref="T:LightweightIocContainer.Lifestyle"/> for this <see cref="T:LightweightIocContainer.Interfaces.Registrations.IDefaultRegistration`1"/></param>
+            <returns>The created <see cref="T:LightweightIocContainer.Interfaces.Registrations.IRegistrationBase"/></returns>
         </member>
         <member name="M:LightweightIocContainer.IocContainer.Register``1(LightweightIocContainer.Lifestyle)">
             <summary>
@@ -448,6 +458,7 @@
             </summary>
             <typeparam name="TImplementation">The <see cref="T:System.Type"/> to register</typeparam>
             <param name="lifestyle">The <see cref="T:LightweightIocContainer.Lifestyle"/> for this <see cref="T:LightweightIocContainer.Interfaces.Registrations.IDefaultRegistration`1"/></param>
+            <returns>The created <see cref="T:LightweightIocContainer.Interfaces.Registrations.IRegistrationBase"/></returns>
         </member>
         <member name="M:LightweightIocContainer.IocContainer.Register``3">
             <summary>
@@ -456,6 +467,7 @@
             <typeparam name="TInterface">The Interface to register</typeparam>
             <typeparam name="TImplementation">The Type that implements the interface</typeparam>
             <typeparam name="TScope">The Type of the multiton scope</typeparam>
+            <returns>The created <see cref="T:LightweightIocContainer.Interfaces.Registrations.IRegistrationBase"/></returns>
         </member>
         <member name="M:LightweightIocContainer.IocContainer.Register(System.Type,System.Type,LightweightIocContainer.Lifestyle)">
             <summary>
@@ -464,6 +476,7 @@
             <param name="tInterface">The Interface to register</param>
             <param name="tImplementation">The Type that implements the interface</param>
             <param name="lifestyle">The <see cref="T:LightweightIocContainer.Lifestyle"/> for this <see cref="T:LightweightIocContainer.Interfaces.Registrations.IDefaultRegistration`1"/></param>
+            <returns>The created <see cref="T:LightweightIocContainer.Interfaces.Registrations.IRegistrationBase"/></returns>
         </member>
         <member name="M:LightweightIocContainer.IocContainer.Register(System.Type,LightweightIocContainer.Lifestyle)">
             <summary>
@@ -471,6 +484,7 @@
             </summary>
             <param name="tImplementation">The <see cref="T:System.Type"/> to register</param>
             <param name="lifestyle">The <see cref="T:LightweightIocContainer.Lifestyle"/> for this <see cref="T:LightweightIocContainer.Interfaces.Registrations.IDefaultRegistration`1"/></param>
+            <returns>The created <see cref="T:LightweightIocContainer.Interfaces.Registrations.IRegistrationBase"/></returns>
         </member>
         <member name="M:LightweightIocContainer.IocContainer.Register(System.Type,System.Type,System.Type)">
             <summary>
@@ -479,18 +493,21 @@
             <param name="tInterface">The Interface to register</param>
             <param name="tImplementation">The Type that implements the interface</param>
             <param name="tScope">The Type of the multiton scope</param>
+            <returns>The created <see cref="T:LightweightIocContainer.Interfaces.Registrations.IRegistrationBase"/></returns>
         </member>
         <member name="M:LightweightIocContainer.IocContainer.RegisterFactory``1">
             <summary>
             Register an Interface as an abstract typed factory/>
             </summary>
             <typeparam name="TFactory">The abstract typed factory to register</typeparam>
+            <returns>The created <see cref="T:LightweightIocContainer.Interfaces.Registrations.IRegistrationBase"/></returns>
         </member>
         <member name="M:LightweightIocContainer.IocContainer.RegisterFactory(System.Type)">
             <summary>
             Register an Interface as an abstract typed factory/>
             </summary>
             <param name="tFactory">The abstract typed factory to register</param>
+            <returns>The created <see cref="T:LightweightIocContainer.Interfaces.Registrations.IRegistrationBase"/></returns>
         </member>
         <member name="M:LightweightIocContainer.IocContainer.RegisterUnitTestCallback``1(LightweightIocContainer.ResolveCallback{``0})">
             <summary>
@@ -498,6 +515,7 @@
             </summary>
             <typeparam name="TInterface">The Interface to register</typeparam>
             <param name="unitTestCallback">The <see cref="T:LightweightIocContainer.ResolveCallback`1"/> for the callback</param>
+            <returns>The created <see cref="T:LightweightIocContainer.Interfaces.Registrations.IRegistrationBase"/></returns>
         </member>
         <member name="M:LightweightIocContainer.IocContainer.Register(LightweightIocContainer.Interfaces.Registrations.IRegistrationBase)">
             <summary>