From 65f5f58a07f29c5baf4fe80218cc2a83deba5dc6 Mon Sep 17 00:00:00 2001
From: artyom <worldbeater-dev@yandex.ru>
Date: Thu, 23 May 2019 00:36:53 +0300
Subject: [PATCH] ViewModelViewHost type resolution assertion

---
 .../AutoDataTemplateBindingHookTest.cs        | 44 ++++++++++++++++---
 1 file changed, 39 insertions(+), 5 deletions(-)

diff --git a/tests/Avalonia.ReactiveUI.UnitTests/AutoDataTemplateBindingHookTest.cs b/tests/Avalonia.ReactiveUI.UnitTests/AutoDataTemplateBindingHookTest.cs
index 2391daece2..4ff3e1fed6 100644
--- a/tests/Avalonia.ReactiveUI.UnitTests/AutoDataTemplateBindingHookTest.cs
+++ b/tests/Avalonia.ReactiveUI.UnitTests/AutoDataTemplateBindingHookTest.cs
@@ -6,9 +6,12 @@ using ReactiveUI;
 using Avalonia.ReactiveUI;
 using Avalonia.UnitTests;
 using Avalonia.Controls;
+using Avalonia.Controls.Templates;
 using System.Collections.Generic;
 using System.Collections.ObjectModel;
 using System.Linq;
+using Avalonia.VisualTree;
+using Avalonia.Controls.Presenters;
 using Splat;
 
 namespace Avalonia.ReactiveUI.UnitTests
@@ -26,7 +29,7 @@ namespace Avalonia.ReactiveUI.UnitTests
 
         public class ExampleView : ReactiveUserControl<ExampleViewModel>
         {
-            public ListBox List { get; } = new ListBox();
+            public ItemsControl List { get; } = new ItemsControl();
 
             public ExampleView()
             {
@@ -40,17 +43,48 @@ namespace Avalonia.ReactiveUI.UnitTests
         {
             Locator.CurrentMutable.RegisterConstant(new AutoDataTemplateBindingHook(), typeof(IPropertyBindingHook));
             Locator.CurrentMutable.Register(() => new ExampleView(), typeof(IViewFor<ExampleViewModel>));
+            Locator.CurrentMutable.RegisterConstant(new AvaloniaActivationForViewFetcher(), typeof(IActivationForViewFetcher));
         }
 
         [Fact]
         public void Should_Apply_Data_Template_Binding_When_No_Template_Is_Set()
         {
             var view = new ExampleView();
-            var root = new TestRoot 
-            { 
-                Child = view
-            };
             Assert.NotNull(view.List.ItemTemplate);
         }
+
+        [Fact]
+        public void Should_Use_View_Model_View_Host_As_Data_Template()
+        {
+            var view = new ExampleView();
+            view.ViewModel.Items.Add(new NestedViewModel());
+
+            view.List.Template = GetTemplate();
+            view.List.ApplyTemplate();
+            view.List.Presenter.ApplyTemplate();
+
+            var child = view.List.Presenter.Panel.Children[0];
+            var container = (ContentPresenter) child;
+            container.UpdateChild();
+
+            Assert.IsType<ViewModelViewHost>(container.Child);
+        }
+
+        private FuncControlTemplate GetTemplate()
+        {
+            return new FuncControlTemplate<ItemsControl>(parent =>
+            {
+                return new Border
+                {
+                    Background = new Media.SolidColorBrush(0xffffffff),
+                    Child = new ItemsPresenter
+                    {
+                        Name = "PART_ItemsPresenter",
+                        MemberSelector = parent.MemberSelector,
+                        [~ItemsPresenter.ItemsProperty] = parent[~ItemsControl.ItemsProperty],
+                    }
+                };
+            });
+        }
     }
 }
\ No newline at end of file