Fix WarmupAssemblyViewModels issue.

Author: ChebanovDD

src/UnityMvvmToolkit.Core/BindingContextObjectProvider.cs

@@ -33,7 +33,7 @@ public BindingContextObjectProvider(IValueConverter[] converters,
         public IObjectProvider WarmupAssemblyViewModels()
         {
             var assemblyTypes = Assembly
-                .GetExecutingAssembly()
+                .GetCallingAssembly()
                 .GetTypes()
                 .Where(type => type.IsInterface == false && type.IsAbstract == false &&
                                typeof(IBindingContext).IsAssignableFrom(type));
@@ -53,6 +53,12 @@ public IObjectProvider WarmupViewModel<TBindingContext>() where TBindingContext
 
         public IObjectProvider WarmupViewModel(Type bindingContextType)
         {
+            if (bindingContextType.IsInterface || bindingContextType.IsAbstract ||
+                typeof(IBindingContext).IsAssignableFrom(bindingContextType) == false)
+            {
+                throw new InvalidOperationException($"Can not warmup {bindingContextType.Name}.");
+            }
+
             if (_bindingContextHandler.TryRegisterBindingContext(bindingContextType) == false)
             {
                 throw new InvalidOperationException($"{bindingContextType.Name} already warmed up.");

src/UnityMvvmToolkit.UnityPackage/Assets/Plugins/UnityMvvmToolkit/Runtime/Core/BindingContextObjectProvider.cs

@@ -33,7 +33,7 @@ public BindingContextObjectProvider(IValueConverter[] converters,
         public IObjectProvider WarmupAssemblyViewModels()
         {
             var assemblyTypes = Assembly
-                .GetExecutingAssembly()
+                .GetCallingAssembly()
                 .GetTypes()
                 .Where(type => type.IsInterface == false && type.IsAbstract == false &&
                                typeof(IBindingContext).IsAssignableFrom(type));
@@ -53,6 +53,12 @@ public IObjectProvider WarmupViewModel<TBindingContext>() where TBindingContext
 
         public IObjectProvider WarmupViewModel(Type bindingContextType)
         {
+            if (bindingContextType.IsInterface || bindingContextType.IsAbstract ||
+                typeof(IBindingContext).IsAssignableFrom(bindingContextType) == false)
+            {
+                throw new InvalidOperationException($"Can not warmup {bindingContextType.Name}.");
+            }
+
             if (_bindingContextHandler.TryRegisterBindingContext(bindingContextType) == false)
             {
                 throw new InvalidOperationException($"{bindingContextType.Name} already warmed up.");

tests/UnityMvvmToolkit.Test.Integration/BindingContextObjectProviderTests.cs

@@ -7,13 +7,37 @@
 using UnityMvvmToolkit.Core.Internal.ObjectHandlers;
 using UnityMvvmToolkit.Core.Internal.ObjectWrappers;
 using UnityMvvmToolkit.Test.Integration.TestBindingContext;
+using UnityMvvmToolkit.Test.Unit.TestBindingContext;
 
 namespace UnityMvvmToolkit.Test.Integration;
 
 public class BindingContextObjectProviderTests
 {
     [Fact]
-    public void WarmupViewModel_ShouldWarmupBindingContext_WhenBindingContextIsNotWarmup()
+    public void WarmupAssemblyViewModels_ShouldWarmupAllViewModels()
+    {
+        // Arrange
+        var objectProvider = new BindingContextObjectProvider(Array.Empty<IValueConverter>());
+
+        // Act
+        objectProvider.WarmupAssemblyViewModels();
+
+        // Assert
+        objectProvider
+            .Invoking(sut => sut.WarmupViewModel<MyBindingContext>())
+            .Should()
+            .Throw<InvalidOperationException>()
+            .WithMessage($"{nameof(MyBindingContext)} already warmed up.");
+
+        objectProvider
+            .Invoking(sut => sut.WarmupViewModel<EmptyBindingContext>())
+            .Should()
+            .Throw<InvalidOperationException>()
+            .WithMessage($"{nameof(EmptyBindingContext)} already warmed up.");
+    }
+
+    [Fact]
+    public void WarmupViewModelT_ShouldWarmupBindingContext_WhenBindingContextIsNotWarmup()
     {
         // Arrange
         var objectProvider = new BindingContextObjectProvider(Array.Empty<IValueConverter>());
@@ -29,6 +53,41 @@ public void WarmupViewModel_ShouldWarmupBindingContext_WhenBindingContextIsNotWa
             .WithMessage($"{nameof(MyBindingContext)} already warmed up.");
     }
 
+    [Fact]
+    public void WarmupViewModel_ShouldWarmupBindingContext_WhenBindingContextIsNotWarmup()
+    {
+        // Arrange
+        var bindingContextType = typeof(MyBindingContext);
+        var objectProvider = new BindingContextObjectProvider(Array.Empty<IValueConverter>());
+
+        // Act
+        objectProvider.WarmupViewModel(bindingContextType);
+
+        // Assert
+        objectProvider
+            .Invoking(sut => sut.WarmupViewModel(bindingContextType))
+            .Should()
+            .Throw<InvalidOperationException>()
+            .WithMessage($"{bindingContextType.Name} already warmed up.");
+    }
+
+    [Theory]
+    [InlineData(typeof(NoBindingContext))]
+    [InlineData(typeof(AbstractBindingContext))]
+    [InlineData(typeof(IInterfaceBindingContext))]
+    public void WarmupViewModel_ShouldThrow_WhenBindingContextIsNotSupported(Type bindingContextType)
+    {
+        // Arrange
+        var objectProvider = new BindingContextObjectProvider(Array.Empty<IValueConverter>());
+
+        // Assert
+        objectProvider
+            .Invoking(sut => sut.WarmupViewModel(bindingContextType))
+            .Should()
+            .Throw<InvalidOperationException>()
+            .WithMessage($"Can not warmup {bindingContextType.Name}.");
+    }
+
     [Fact]
     public void WarmupValueConverter_ShouldNotThrow_WhenValueConverterWasNotWarmup()
     {

tests/UnityMvvmToolkit.Test.Integration/TestBindingContext/AbstractBindingContext.cs

@@ -0,0 +1,7 @@
+using UnityMvvmToolkit.Core.Interfaces;
+
+namespace UnityMvvmToolkit.Test.Integration.TestBindingContext;
+
+public abstract class AbstractBindingContext : IBindingContext
+{
+}

tests/UnityMvvmToolkit.Test.Integration/TestBindingContext/EmptyBindingContext.cs

@@ -0,0 +1,7 @@
+using UnityMvvmToolkit.Core.Interfaces;
+
+namespace UnityMvvmToolkit.Test.Integration.TestBindingContext;
+
+public class EmptyBindingContext : IBindingContext
+{
+}

tests/UnityMvvmToolkit.Test.Integration/TestBindingContext/IInterfaceBindingContext.cs

@@ -0,0 +1,7 @@
+using UnityMvvmToolkit.Core.Interfaces;
+
+namespace UnityMvvmToolkit.Test.Integration.TestBindingContext;
+
+public interface IInterfaceBindingContext : IBindingContext
+{
+}

tests/UnityMvvmToolkit.Test.Integration/TestBindingContext/NotBindingContext.cs

@@ -0,0 +1,5 @@
+namespace UnityMvvmToolkit.Test.Integration.TestBindingContext;
+
+public class NotBindingContext
+{
+}