Issue 10773 add coverage for ToolStripInSituService

src/System.Windows.Forms.Design/tests/UnitTests/System/Windows/Forms/Design/ToolStripInSituServiceTests.cs

@@ -0,0 +1,194 @@
+// Licensed to the .NET Foundation under one or more agreements.
+// The .NET Foundation licenses this file to you under the MIT license.
+
+using System.ComponentModel;
+using System.ComponentModel.Design;
+using Moq;
+
+namespace System.Windows.Forms.Design.Tests;
+
+public sealed class ToolStripInSituServiceTests : IDisposable
+{
+    private readonly Mock<IServiceProvider> _mockServiceProvider;
+    private readonly Mock<IDesignerHost> _mockDesignerHost;
+    private readonly Mock<IComponentChangeService> _mockComponentChangeService;
+    private readonly Mock<ToolStripDesigner> _mockToolStripDesigner;
+    private readonly Mock<ToolStripItemDesigner> _mockToolStripItemDesigner;
+    private readonly Mock<ISelectionService> _mockSelectionService;
+    private readonly Mock<ToolStripKeyboardHandlingService> _mockToolStripKeyboardHandlingService;
+    private readonly ToolStripInSituService _inSituService;
+    private bool _isInSituServiceDisposed;
+
+    public ToolStripInSituServiceTests()
+    {
+        _mockServiceProvider = new();
+        _mockDesignerHost = new();
+        _mockComponentChangeService = new();
+        _mockToolStripDesigner = new();
+        _mockToolStripItemDesigner = new();
+        _mockSelectionService = new();
+        _mockToolStripKeyboardHandlingService = new(_mockServiceProvider.Object);
+
+        _mockServiceProvider.Setup(sp => sp.GetService(typeof(IDesignerHost))).Returns(_mockDesignerHost.Object);
+        _mockDesignerHost.Setup(dh => dh.GetService(typeof(IComponentChangeService))).Returns(_mockComponentChangeService.Object);
+        _mockServiceProvider.Setup(sp => sp.GetService(typeof(ISelectionService))).Returns(_mockSelectionService.Object);
+        _mockServiceProvider.Setup(sp => sp.GetService(typeof(ToolStripKeyboardHandlingService))).Returns(_mockToolStripKeyboardHandlingService.Object);
+
+        _inSituService = new(_mockServiceProvider.Object);
+        _inSituService.TestAccessor().Dynamic._toolDesigner = _mockToolStripDesigner.Object;
+        _inSituService.TestAccessor().Dynamic._toolItemDesigner = _mockToolStripItemDesigner.Object;
+        _inSituService.TestAccessor().Dynamic._componentChangeService = _mockComponentChangeService.Object;
+    }
+
+    public void Dispose()
+    {
+        if (!_isInSituServiceDisposed)
+        {
+            _inSituService.Dispose();
+        }
+    }
+
+    [Fact]
+    public void Dispose_DisposesToolDesigner()
+    {
+        object toolDesignerValue = _inSituService.TestAccessor().Dynamic._toolDesigner;
+        toolDesignerValue.Should().NotBeNull();
+
+        _inSituService.Dispose();
+        _isInSituServiceDisposed = true;
+
+        toolDesignerValue = _inSituService.TestAccessor().Dynamic._toolDesigner;
+        toolDesignerValue.Should().BeNull();
+    }
+
+    [Fact]
+    public void Dispose_DisposesToolItemDesigner()
+    {
+        object toolItemDesignerValue = _inSituService.TestAccessor().Dynamic._toolItemDesigner;
+        toolItemDesignerValue.Should().NotBeNull();
+
+        _inSituService.Dispose();
+        _isInSituServiceDisposed = true;
+
+        toolItemDesignerValue = _inSituService.TestAccessor().Dynamic._toolItemDesigner;
+        toolItemDesignerValue.Should().BeNull();
+    }
+
+    [Fact]
+    public void Dispose_UnsubscribesFromComponentChangeService()
+    {
+        object componentChangeServiceValue = _inSituService.TestAccessor().Dynamic._componentChangeService;
+        componentChangeServiceValue.Should().NotBeNull();
+
+        _inSituService.Dispose();
+        _isInSituServiceDisposed = true;
+
+        componentChangeServiceValue = _inSituService.TestAccessor().Dynamic._componentChangeService;
+        componentChangeServiceValue.Should().BeNull();
+    }
+
+    [Fact]
+    public void ToolStripKeyBoardService_ReturnsServiceInstance()
+    {
+        object toolStripKeyBoardService = _inSituService.TestAccessor().Dynamic.ToolStripKeyBoardService;
+
+        toolStripKeyBoardService.Should().BeAssignableTo<ToolStripKeyboardHandlingService>();
+        toolStripKeyBoardService.Should().Be(_mockToolStripKeyboardHandlingService.Object);
+    }
+
+    [Fact]
+    public void GetEditWindow_ReturnsZero_WhenNoDesignerIsNotNull()
+    {
+        IntPtr result = _inSituService.GetEditWindow();
+        result.Should().Be(IntPtr.Zero);
+    }
+
+    [Fact]
+    public void OnComponentRemoved_RemovesService_WhenNoToolStripPresent()
+    {
+        Mock<IContainer> mockContainer = new();
+        ComponentCollection componentCollection = new(Array.Empty<IComponent>());
+
+        mockContainer.Setup(c => c.Components).Returns(componentCollection);
+        _mockDesignerHost.Setup(dh => dh.Container).Returns(mockContainer.Object);
+        _mockServiceProvider.Setup(sp => sp.GetService(typeof(ISupportInSituService))).Returns(_inSituService);
+
+        using Component component = new();
+        _inSituService.TestAccessor().Dynamic.OnComponentRemoved(null, new ComponentEventArgs(component));
+
+        _mockDesignerHost.Verify(dh => dh.RemoveService(typeof(ISupportInSituService)), Times.Once);
+    }
+
+    [Fact]
+    public void OnComponentRemoved_DoesNotRemoveService_WhenToolStripPresent()
+    {
+        Mock<ToolStrip> mockToolStrip = new();
+        Mock<IContainer> mockContainer = new();
+        ComponentCollection realComponentCollection = new([mockToolStrip.Object]);
+
+        mockContainer.Setup(c => c.Components).Returns(realComponentCollection);
+        _mockDesignerHost.Setup(dh => dh.Container).Returns(mockContainer.Object);
+        _mockServiceProvider.Setup(sp => sp.GetService(typeof(ISupportInSituService))).Returns(_inSituService);
+
+        using Component component = new();
+        _inSituService.TestAccessor().Dynamic.OnComponentRemoved(null, new ComponentEventArgs(component));
+
+        _mockDesignerHost.Verify(dh => dh.RemoveService(typeof(ISupportInSituService)), Times.Never);
+    }
+
+    [Fact]
+    public void IgnoreMessages_ReturnsFalse_WhenSelectionServiceIsNull()
+    {
+        _mockServiceProvider.Setup(sp => sp.GetService(typeof(ISelectionService))).Returns(null!);
+        bool result = _inSituService.IgnoreMessages;
+        result.Should().BeFalse();
+    }
+
+    [Fact]
+    public void IgnoreMessages_ReturnsFalse_WhenDesignerHostIsNull()
+    {
+        _mockServiceProvider.Setup(sp => sp.GetService(typeof(IDesignerHost))).Returns(null!);
+        bool result = _inSituService.IgnoreMessages;
+        result.Should().BeFalse();
+    }
+
+    [Fact]
+    public void IgnoreMessages_WhenPrimarySelectionIsNotIComponentAndSelectedDesignerControlIsNull_ReturnsFalse()
+    {
+        _inSituService.TestAccessor().Dynamic._toolDesigner = _mockToolStripDesigner.Object;
+        bool result = _inSituService.IgnoreMessages;
+        result.Should().BeFalse();
+    }
+
+    [Fact]
+    public void IgnoreMessages_WhenComponentIsMenuStrip_ReturnsTrue()
+    {
+        Mock<MenuStrip> menuStripMock = new();
+        Mock<ToolStrip> toolStripMock = new();
+        _mockSelectionService.Setup(ss => ss.PrimarySelection).Returns(menuStripMock.Object);
+        _mockDesignerHost.Setup(dh => dh.GetDesigner(menuStripMock.Object)).Returns(_mockToolStripDesigner.Object);
+        bool result = _inSituService.IgnoreMessages;
+        result.Should().BeTrue();
+    }
+
+    [Fact]
+    public void IgnoreMessages_WhenComponentIsToolStripMenuItem_ReturnsTrue()
+    {
+        Mock<ToolStripMenuItem> toolStripMenuItemMock = new();
+        Mock<DesignerToolStripControlHost> designerToolStripControlHostMock = new(toolStripMenuItemMock.Object);
+        _mockSelectionService.Setup(ss => ss.PrimarySelection).Returns(toolStripMenuItemMock.Object);
+        _mockDesignerHost.Setup(dh => dh.GetDesigner(toolStripMenuItemMock.Object)).Returns(_mockToolStripItemDesigner.Object);
+        bool result = _inSituService.IgnoreMessages;
+        result.Should().BeTrue();
+    }
+
+    [Fact]
+    public void HandleKeyChar_WhenToolItemDesignerIsNotMenuDesigner_CallsShowEditNode()
+    {
+        _inSituService.TestAccessor().Dynamic._toolDesigner = null;
+
+        _inSituService.HandleKeyChar();
+
+        _mockToolStripItemDesigner.Verify(d => d.ShowEditNode(false), Times.Once);
+    }
+}