// Copyright (C) Stichting Deltares 2016. All rights reserved.
//
// This file is part of Ringtoets.
//
// Ringtoets is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
// GNU General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with this program. If not, see .
//
// All names, logos, and references to "Deltares" are registered trademarks of
// Stichting Deltares and remain full property of Stichting Deltares at all times.
// All rights reserved.
using System.Drawing;
using System.Linq;
using System.Windows.Forms;
using Core.Common.Base;
using Core.Common.Controls.DataGrid;
using Core.Common.Controls.TreeView;
using Core.Common.Gui;
using Core.Common.Gui.Commands;
using Core.Common.Gui.ContextMenu;
using Core.Common.Gui.Forms.MainWindow;
using Core.Common.Gui.TestUtil.ContextMenu;
using Core.Common.Gui.TestUtil.Settings;
using Core.Common.TestUtil;
using Core.Components.BruTile.Configurations;
using Core.Components.BruTile.TestUtil;
using Core.Components.Gis.Data;
using Core.Components.Gis.Forms.Views;
using NUnit.Extensions.Forms;
using NUnit.Framework;
using Rhino.Mocks;
using Ringtoets.Common.Data.AssessmentSection;
using Ringtoets.Common.Data.TestUtil;
using Ringtoets.Integration.Forms;
using Ringtoets.Integration.Forms.Properties;
using RingtoetsCommonFormsResources = Ringtoets.Common.Forms.Properties.Resources;
namespace Ringtoets.Integration.Plugin.Test.TreeNodeInfos
{
[TestFixture]
public class BackgroundDataTreeNodeInfoTest : NUnitFormTest
{
private const int selectContextMenuIndex = 0;
[Test]
public void Initialized_Always_ExpectedPropertiesSet()
{
// Setup
using (var plugin = new RingtoetsPlugin())
{
TreeNodeInfo info = GetInfo(plugin);
// Assert
Assert.IsNotNull(info.Text);
Assert.IsNotNull(info.ForeColor);
Assert.IsNotNull(info.Image);
Assert.IsNotNull(info.ContextMenuStrip);
Assert.IsNull(info.EnsureVisibleOnCreate);
Assert.IsNull(info.ExpandOnCreate);
Assert.IsNull(info.ChildNodeObjects);
Assert.IsNull(info.CanRename);
Assert.IsNull(info.OnNodeRenamed);
Assert.IsNull(info.CanRemove);
Assert.IsNull(info.OnNodeRemoved);
Assert.IsNull(info.CanCheck);
Assert.IsNull(info.IsChecked);
Assert.IsNull(info.OnNodeChecked);
Assert.IsNull(info.CanDrag);
Assert.IsNull(info.CanDrop);
Assert.IsNull(info.CanInsert);
Assert.IsNull(info.OnDrop);
}
}
[Test]
public void Text_Always_ReturnsName()
{
// Setup
var backgroundMapData = new BackgroundData();
using (var plugin = new RingtoetsPlugin())
{
TreeNodeInfo info = GetInfo(plugin);
// Call
string text = info.Text(backgroundMapData);
// Assert
Assert.AreEqual("Achtergrondkaart", text);
}
}
[Test]
public void Image_Always_ReturnsSetImage()
{
// Setup
var backgroundMapData = new BackgroundData();
using (var plugin = new RingtoetsPlugin())
{
TreeNodeInfo info = GetInfo(plugin);
// Call
Image image = info.Image(backgroundMapData);
// Assert
TestHelper.AssertImagesAreEqual(Resources.Map, image);
}
}
[Test]
public void ForeColor_ConnectedMapData_ReturnControlText()
{
WmtsMapData mapData = WmtsMapData.CreateDefaultPdokMapData();
BackgroundData backgroundData = BackgroundDataTestDataGenerator.GetWmtsBackgroundMapData(mapData);
using (var plugin = new RingtoetsPlugin())
{
TreeNodeInfo info = GetInfo(plugin);
// Call
Color image = info.ForeColor(backgroundData);
// Assert
Assert.AreEqual(Color.FromKnownColor(KnownColor.ControlText), image);
}
}
[Test]
public void ForeColor_UnconnectedMapData_ReturnGrayText()
{
WmtsMapData mapData = WmtsMapData.CreateUnconnectedMapData();
BackgroundData backgroundData = BackgroundDataTestDataGenerator.GetWmtsBackgroundMapData(mapData);
using (var plugin = new RingtoetsPlugin())
{
TreeNodeInfo info = GetInfo(plugin);
// Call
Color image = info.ForeColor(backgroundData);
// Assert
Assert.AreEqual(Color.FromKnownColor(KnownColor.GrayText), image);
}
}
[Test]
public void ForeColor_MapDataNull_ReturnGrayText()
{
var backgroundMapData = new BackgroundData();
using (var plugin = new RingtoetsPlugin())
{
TreeNodeInfo info = GetInfo(plugin);
// Call
Color image = info.ForeColor(backgroundMapData);
// Assert
Assert.AreEqual(Color.FromKnownColor(KnownColor.GrayText), image);
}
}
[Test]
public void ContextMenuStrip_Always_CallsBuilder()
{
// Setup
var mockRepository = new MockRepository();
var menuBuilderMock = mockRepository.StrictMock();
using (mockRepository.Ordered())
{
menuBuilderMock.Expect(mb => mb.AddCustomItem(null)).IgnoreArguments().Return(menuBuilderMock);
menuBuilderMock.Expect(mb => mb.AddPropertiesItem()).Return(menuBuilderMock);
menuBuilderMock.Expect(mb => mb.Build()).Return(null);
}
using (var treeViewControl = new TreeViewControl())
{
var gui = mockRepository.Stub();
gui.Stub(g => g.Get(null, treeViewControl)).Return(menuBuilderMock);
gui.Stub(g => g.ProjectOpened += null).IgnoreArguments();
gui.Stub(g => g.ProjectOpened -= null).IgnoreArguments();
mockRepository.ReplayAll();
using (var plugin = new RingtoetsPlugin())
{
var info = GetInfo(plugin);
plugin.Gui = gui;
// Call
info.ContextMenuStrip(null, null, treeViewControl);
}
}
// Assert
mockRepository.VerifyAll();
}
[Test]
public void ContextMenuStrip_Always_ContextMenuItemSelectMapLayerEnabled()
{
// Setup
var mockRepository = new MockRepository();
var backgroundMapData = new BackgroundData();
var assessmentSection = mockRepository.Stub();
using (var treeViewControl = new TreeViewControl())
{
var gui = mockRepository.Stub();
gui.Stub(g => g.Get(backgroundMapData, treeViewControl)).Return(new CustomItemsOnlyContextMenuBuilder());
gui.Stub(g => g.ProjectOpened += null).IgnoreArguments();
gui.Stub(g => g.ProjectOpened -= null).IgnoreArguments();
mockRepository.ReplayAll();
using (var plugin = new RingtoetsPlugin())
{
var info = GetInfo(plugin);
plugin.Gui = gui;
// Call
using (ContextMenuStrip contextMenu = info.ContextMenuStrip(backgroundMapData, assessmentSection, treeViewControl))
{
const string expectedItemText = "&Selecteren...";
const string expectedItemTooltip = "Selecteer een achtergrondkaart.";
TestHelper.AssertContextMenuStripContainsItem(contextMenu, selectContextMenuIndex,
expectedItemText, expectedItemTooltip,
RingtoetsCommonFormsResources.MapsIcon);
}
}
}
// Assert
mockRepository.VerifyAll();
}
[Test]
public void GivenNoMapDataSet_WhenSelectingValidWMTSMapDataFromContextMenu_ThenBackgroundDataSetAndNotifiesObserver()
{
// Given
var mockRepository = new MockRepository();
var tileFactory = mockRepository.StrictMock();
WmtsMapData newMapData = new WmtsMapData("Actueel Hoogtebestand Nederland (AHN1)",
"https://geodata.nationaalgeoregister.nl/tiles/service/wmts/ahn1?request=GetCapabilities",
"()", "image/png");
tileFactory.Expect(tf => tf.GetWmtsTileSources(null)).IgnoreArguments().Return(new[]
{
new TestWmtsTileSource(newMapData)
});
var assessmentSectionObserver = mockRepository.StrictMock();
assessmentSectionObserver.Expect(o => o.UpdateObserver());
var backgroundMapDataObserver = mockRepository.StrictMock();
backgroundMapDataObserver.Expect(o => o.UpdateObserver());
WmtsMapData mapData = WmtsMapData.CreateUnconnectedMapData();
BackgroundData backgroundData = BackgroundDataTestDataGenerator.GetWmtsBackgroundMapData(mapData);
using (new UseCustomSettingsHelper(new TestSettingsHelper
{
ApplicationLocalUserSettingsDirectory = TestHelper.GetTestDataPath(TestDataPath.Core.Components.Gis.IO, "twoValidWmtsConnectionInfos")
}))
using (new UseCustomTileSourceFactoryConfig(tileFactory))
using (var treeViewControl = new TreeViewControl())
using (var plugin = new RingtoetsPlugin())
{
var viewCommands = mockRepository.Stub();
var mainWindow = mockRepository.Stub();
var gui = mockRepository.Stub();
gui.Stub(g => g.MainWindow).Return(mainWindow);
gui.Stub(g => g.ProjectOpened += null).IgnoreArguments();
gui.Stub(g => g.ProjectOpened -= null).IgnoreArguments();
gui.Stub(g => g.ViewCommands).Return(viewCommands);
gui.Stub(cmp => cmp.Get(backgroundData, treeViewControl)).Return(new CustomItemsOnlyContextMenuBuilder());
mockRepository.ReplayAll();
var assessmentSection = new ObservableTestAssessmentSectionStub();
assessmentSection.Attach(assessmentSectionObserver);
assessmentSection.BackgroundData.Attach(backgroundMapDataObserver);
DialogBoxHandler = (name, wnd) =>
{
var dialog = (BackgroundMapDataSelectionDialog) new FormTester(name).TheObject;
var button = new ButtonTester("selectButton", dialog);
button.Click();
dialog.Close();
};
TreeNodeInfo info = GetInfo(plugin);
plugin.Gui = gui;
using (ContextMenuStrip contextMenuStrip = info.ContextMenuStrip(backgroundData, assessmentSection, treeViewControl))
{
// When
contextMenuStrip.Items[selectContextMenuIndex].PerformClick();
// Then
AssertBackgroundMapDataProperties(newMapData, assessmentSection.BackgroundData);
}
}
mockRepository.VerifyAll();
}
[Test]
public void GivenNoMapDataSet_WhenSelectingMapDataFromContextMenuCancelled_ThenNoObserversNotified()
{
// Given
var mockRepository = new MockRepository();
var assessmentSectionObserver = mockRepository.StrictMock();
var backgroundMapDataObserver = mockRepository.StrictMock();
var backgroundMapData = new BackgroundData
{
Name = "background map data",
BackgroundMapDataType = BackgroundMapDataType.Wmts
};
using (var treeViewControl = new TreeViewControl())
using (var plugin = new RingtoetsPlugin())
{
var viewCommands = mockRepository.Stub();
var mainWindow = mockRepository.Stub();
var gui = mockRepository.Stub();
gui.Stub(g => g.MainWindow).Return(mainWindow);
gui.Stub(g => g.ProjectOpened += null).IgnoreArguments();
gui.Stub(g => g.ProjectOpened -= null).IgnoreArguments();
gui.Stub(g => g.ViewCommands).Return(viewCommands);
gui.Stub(cmp => cmp.Get(backgroundMapData, treeViewControl)).Return(new CustomItemsOnlyContextMenuBuilder());
mockRepository.ReplayAll();
var assessmentSection = new ObservableTestAssessmentSectionStub();
assessmentSection.Attach(assessmentSectionObserver);
assessmentSection.BackgroundData.Attach(backgroundMapDataObserver);
BackgroundData oldBackgroundData = assessmentSection.BackgroundData;
DialogBoxHandler = (name, wnd) =>
{
var tester = (BackgroundMapDataSelectionDialog) new FormTester(name).TheObject;
tester.DialogResult = DialogResult.Cancel;
tester.Close();
};
TreeNodeInfo info = GetInfo(plugin);
plugin.Gui = gui;
using (ContextMenuStrip contextMenuStrip = info.ContextMenuStrip(backgroundMapData, assessmentSection, treeViewControl))
{
// When
contextMenuStrip.Items[selectContextMenuIndex].PerformClick();
// Then
AssertBackgroundMapDataProperties(oldBackgroundData, assessmentSection.BackgroundData);
}
}
mockRepository.VerifyAll();
}
[Test]
public void GivenMapDataSet_WhenSelectingValidWmtsMapDataFromContextMenu_ThenBackgroundDataSetAndNotifiesObserver()
{
// Given
var mockRepository = new MockRepository();
var assessmentSectionObserver = mockRepository.StrictMock();
assessmentSectionObserver.Expect(o => o.UpdateObserver());
var backgroundMapDataObserver = mockRepository.StrictMock();
backgroundMapDataObserver.Expect(o => o.UpdateObserver());
WmtsMapData mapData = WmtsMapData.CreateUnconnectedMapData();
WmtsMapData newMapData = WmtsMapData.CreateDefaultPdokMapData();
var newBackgroundMapdata = BackgroundDataTestDataGenerator.GetWmtsBackgroundMapData(newMapData);
var assessmentSection = new ObservableTestAssessmentSectionStub();
using (new UseCustomSettingsHelper(new TestSettingsHelper
{
ApplicationLocalUserSettingsDirectory = TestHelper.GetTestDataPath(TestDataPath.Core.Components.Gis.IO, "noConfig")
}))
using (new UseCustomTileSourceFactoryConfig(newMapData))
using (var treeViewControl = new TreeViewControl())
using (var plugin = new RingtoetsPlugin())
{
var viewCommands = mockRepository.Stub();
var mainWindow = mockRepository.Stub();
var gui = mockRepository.Stub();
gui.Stub(g => g.MainWindow).Return(mainWindow);
gui.Stub(g => g.ProjectOpened += null).IgnoreArguments();
gui.Stub(g => g.ProjectOpened -= null).IgnoreArguments();
gui.Stub(g => g.ViewCommands).Return(viewCommands);
gui.Stub(cmp => cmp.Get(newBackgroundMapdata, treeViewControl)).Return(new CustomItemsOnlyContextMenuBuilder());
mockRepository.ReplayAll();
assessmentSection.Attach(assessmentSectionObserver);
assessmentSection.BackgroundData.Attach(backgroundMapDataObserver);
SetBackgroundData(assessmentSection, mapData);
DialogBoxHandler = (name, wnd) =>
{
var dialog = (BackgroundMapDataSelectionDialog) new FormTester(name).TheObject;
var button = new ButtonTester("selectButton", dialog);
button.Click();
dialog.Close();
};
TreeNodeInfo info = GetInfo(plugin);
plugin.Gui = gui;
using (ContextMenuStrip contextMenuStrip = info.ContextMenuStrip(newBackgroundMapdata, assessmentSection, treeViewControl))
{
// When
contextMenuStrip.Items[selectContextMenuIndex].PerformClick();
// Then
AssertBackgroundMapDataProperties(newMapData, assessmentSection.BackgroundData);
}
}
mockRepository.VerifyAll();
}
[Test]
public void GivenMapDataSet_WhenSelectingValidWellKnownMapDataFromContextMenu_ThenBackgroundDataSetAndNotifiesObserver()
{
// Given
var mockRepository = new MockRepository();
var assessmentSectionObserver = mockRepository.StrictMock();
assessmentSectionObserver.Expect(o => o.UpdateObserver());
var backgroundMapDataObserver = mockRepository.StrictMock();
backgroundMapDataObserver.Expect(o => o.UpdateObserver());
WmtsMapData mapData = WmtsMapData.CreateUnconnectedMapData();
var assessmentSection = new ObservableTestAssessmentSectionStub();
const WellKnownTileSource wellKnownTileSource = WellKnownTileSource.BingAerial;
var newMapData = new WellKnownTileSourceMapData(wellKnownTileSource);
var newBackgroundData = BackgroundDataTestDataGenerator.GetWellKnownBackgroundMapData(wellKnownTileSource);
using (new UseCustomSettingsHelper(new TestSettingsHelper
{
ApplicationLocalUserSettingsDirectory = TestHelper.GetTestDataPath(TestDataPath.Core.Components.Gis.IO, "noConfig")
}))
using (var treeViewControl = new TreeViewControl())
using (var plugin = new RingtoetsPlugin())
{
var viewCommands = mockRepository.Stub();
var mainWindow = mockRepository.Stub();
var gui = mockRepository.Stub();
gui.Stub(g => g.MainWindow).Return(mainWindow);
gui.Stub(g => g.ProjectOpened += null).IgnoreArguments();
gui.Stub(g => g.ProjectOpened -= null).IgnoreArguments();
gui.Stub(g => g.ViewCommands).Return(viewCommands);
gui.Stub(cmp => cmp.Get(newBackgroundData, treeViewControl)).Return(new CustomItemsOnlyContextMenuBuilder());
mockRepository.ReplayAll();
assessmentSection.Attach(assessmentSectionObserver);
assessmentSection.BackgroundData.Attach(backgroundMapDataObserver);
SetBackgroundData(assessmentSection, mapData);
DialogBoxHandler = (name, wnd) =>
{
var dialog = (BackgroundMapDataSelectionDialog) new FormTester(name).TheObject;
var comboBox = (ComboBox) new ComboBoxTester("mapLayerComboBox", dialog).TheObject;
comboBox.SelectedItem = ((IBackgroundMapDataSelectionControl[]) comboBox.DataSource).OfType().First();
var dataGridViewControl = (DataGridViewControl) new ControlTester("dataGridViewControl", dialog).TheObject;
dataGridViewControl.SetCurrentCell(dataGridViewControl.GetCell(0, 0));
var button = new ButtonTester("selectButton", dialog);
button.Click();
dialog.Close();
};
TreeNodeInfo info = GetInfo(plugin);
plugin.Gui = gui;
using (ContextMenuStrip contextMenuStrip = info.ContextMenuStrip(newBackgroundData, assessmentSection, treeViewControl))
{
// When
contextMenuStrip.Items[selectContextMenuIndex].PerformClick();
// Then
AssertBackgroundMapDataProperties(newMapData, assessmentSection.BackgroundData);
}
}
mockRepository.VerifyAll();
}
[Test]
public void GivenMapDataSet_WhenSelectingMapDataFromContextMenuCancelled_ThenNoObserversNotified()
{
// Given
var mockRepository = new MockRepository();
var assessmentSectionObserver = mockRepository.StrictMock();
var backgroundMapDataObserver = mockRepository.StrictMock();
WmtsMapData mapData = WmtsMapData.CreateUnconnectedMapData();
BackgroundData backgroundData = BackgroundDataTestDataGenerator.GetWmtsBackgroundMapData(mapData);
WmtsMapData newMapData = WmtsMapData.CreateDefaultPdokMapData();
BackgroundData newBackgroundData = BackgroundDataTestDataGenerator.GetWmtsBackgroundMapData(newMapData);
using (new UseCustomSettingsHelper(new TestSettingsHelper
{
ApplicationLocalUserSettingsDirectory = TestHelper.GetTestDataPath(TestDataPath.Core.Components.Gis.IO, "noConfig")
}))
using (new UseCustomTileSourceFactoryConfig(newMapData))
using (var treeViewControl = new TreeViewControl())
using (var plugin = new RingtoetsPlugin())
{
var viewCommands = mockRepository.Stub();
var mainWindow = mockRepository.Stub();
var gui = mockRepository.Stub();
gui.Stub(g => g.MainWindow).Return(mainWindow);
gui.Stub(g => g.ProjectOpened += null).IgnoreArguments();
gui.Stub(g => g.ProjectOpened -= null).IgnoreArguments();
gui.Stub(g => g.ViewCommands).Return(viewCommands);
gui.Stub(cmp => cmp.Get(newBackgroundData, treeViewControl)).Return(new CustomItemsOnlyContextMenuBuilder());
mockRepository.ReplayAll();
var assessmentSection = new ObservableTestAssessmentSectionStub();
assessmentSection.Attach(assessmentSectionObserver);
assessmentSection.BackgroundData.Attach(backgroundMapDataObserver);
SetBackgroundData(assessmentSection, mapData);
DialogBoxHandler = (name, wnd) =>
{
var tester = (BackgroundMapDataSelectionDialog) new FormTester(name).TheObject;
tester.DialogResult = DialogResult.Cancel;
tester.Close();
};
TreeNodeInfo info = GetInfo(plugin);
plugin.Gui = gui;
using (ContextMenuStrip contextMenuStrip = info.ContextMenuStrip(newBackgroundData, assessmentSection, treeViewControl))
{
// When
contextMenuStrip.Items[selectContextMenuIndex].PerformClick();
// Then
AssertBackgroundMapDataProperties(backgroundData, assessmentSection.BackgroundData);
}
}
mockRepository.VerifyAll();
}
private static void SetBackgroundData(IAssessmentSection assessmentSection, WmtsMapData mapData)
{
assessmentSection.BackgroundData.Name = mapData.Name;
assessmentSection.BackgroundData.IsVisible = mapData.IsVisible;
assessmentSection.BackgroundData.IsConfigured = mapData.IsConfigured;
assessmentSection.BackgroundData.Transparency = mapData.Transparency;
assessmentSection.BackgroundData.BackgroundMapDataType = BackgroundMapDataType.Wmts;
if (mapData.IsConfigured)
{
assessmentSection.BackgroundData.Parameters[BackgroundDataIdentifiers.SourceCapabilitiesUrl] = mapData.SourceCapabilitiesUrl;
assessmentSection.BackgroundData.Parameters[BackgroundDataIdentifiers.SelectedCapabilityIdentifier] = mapData.SelectedCapabilityIdentifier;
assessmentSection.BackgroundData.Parameters[BackgroundDataIdentifiers.PreferredFormat] = mapData.PreferredFormat;
}
}
private static void AssertBackgroundMapDataProperties(WmtsMapData mapData, BackgroundData backgroundData)
{
Assert.AreEqual(mapData.Name, backgroundData.Name);
Assert.IsTrue(backgroundData.IsVisible);
Assert.AreEqual(mapData.IsConfigured, backgroundData.IsConfigured);
Assert.AreEqual(mapData.Transparency, backgroundData.Transparency);
if (mapData.IsConfigured)
{
Assert.AreEqual(mapData.SourceCapabilitiesUrl, backgroundData.Parameters[BackgroundDataIdentifiers.SourceCapabilitiesUrl]);
Assert.AreEqual(mapData.SelectedCapabilityIdentifier, backgroundData.Parameters[BackgroundDataIdentifiers.SelectedCapabilityIdentifier]);
Assert.AreEqual(mapData.PreferredFormat, backgroundData.Parameters[BackgroundDataIdentifiers.PreferredFormat]);
}
else
{
CollectionAssert.IsEmpty(backgroundData.Parameters);
}
}
private static void AssertBackgroundMapDataProperties(WellKnownTileSourceMapData mapData, BackgroundData backgroundData)
{
Assert.AreEqual(mapData.Name, backgroundData.Name);
Assert.IsTrue(backgroundData.IsVisible);
Assert.AreEqual(mapData.IsConfigured, backgroundData.IsConfigured);
Assert.AreEqual(mapData.Transparency, backgroundData.Transparency);
Assert.AreEqual(((int) mapData.TileSource).ToString(), backgroundData.Parameters[BackgroundDataIdentifiers.WellKnownTileSource]);
}
private static void AssertBackgroundMapDataProperties(BackgroundData expectedBackgroundData, BackgroundData actualBackgroundData)
{
Assert.AreEqual(expectedBackgroundData.Name, actualBackgroundData.Name);
Assert.AreEqual(expectedBackgroundData.IsVisible, actualBackgroundData.IsVisible);
Assert.AreEqual(expectedBackgroundData.IsConfigured, actualBackgroundData.IsConfigured);
Assert.AreEqual(expectedBackgroundData.Transparency, actualBackgroundData.Transparency);
CollectionAssert.AreEquivalent(expectedBackgroundData.Parameters, actualBackgroundData.Parameters);
}
private static TreeNodeInfo GetInfo(RingtoetsPlugin plugin)
{
return plugin.GetTreeNodeInfos().First(tni => tni.TagType == typeof(BackgroundData));
}
}
}