// Copyright (C) Stichting Deltares and State of the Netherlands 2025. All rights reserved. // // This file is part of Riskeer. // // Riskeer is free software: you can redistribute it and/or modify // it under the terms of the GNU Lesser 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 Lesser General Public License for more details. // // You should have received a copy of the GNU Lesser 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; using System.IO; using System.Linq; using Core.Common.Base.Geometry; using Core.Common.TestUtil; using Core.Components.Gis.Data; using Core.Components.Gis.Features; using Core.Components.Gis.Geometries; using Core.Components.Gis.IO.Writers; using NUnit.Framework; namespace Core.Components.Gis.IO.Test.Writers { [TestFixture] public class PolylineShapeFileWriterTest { [Test] public void DefaultConstructor_ExpectedValues() { // Call using (var writer = new PolylineShapeFileWriter()) { // Assert Assert.IsInstanceOf(writer); } } [Test] public void CopyToFeature_FeatureContainsNoGeometries_ThrowsArgumentException() { // Setup var feature = new MapFeature(new MapGeometry[0]); var mapData = new MapLineData("test") { Features = new[] { feature } }; using (var writer = new PolylineShapeFileWriter()) { // Call TestDelegate call = () => writer.CopyToFeature(mapData); // Assert TestHelper.AssertThrowsArgumentExceptionAndTestMessage(call, "Een feature mag maar één geometrie bevatten."); } } [Test] public void CopyToFeature_FeatureContainsMultipleGeometries_ThrowsArgumentException() { // Setup var feature = new MapFeature(new[] { new MapGeometry(new[] { Enumerable.Empty() }), new MapGeometry(new[] { Enumerable.Empty() }) }); var mapData = new MapLineData("test") { Features = new[] { feature } }; using (var writer = new PolylineShapeFileWriter()) { // Call TestDelegate call = () => writer.CopyToFeature(mapData); // Assert TestHelper.AssertThrowsArgumentExceptionAndTestMessage(call, "Een feature mag maar één geometrie bevatten."); } } [Test] public void CopyToFeature_InconsistentMetaDataBetweenMapLineDatas_ThrowArgumentException() { // Setup MapFeature[] features1 = CreateFeatures(0.0); var mapLineData1 = new MapLineData("test data 1") { Features = features1 }; mapLineData1.Features.First().MetaData["firstKey"] = 123; mapLineData1.Features.First().MetaData["secondKey"] = "aValue"; MapFeature[] features2 = CreateFeatures(10.0); var mapLineData2 = new MapLineData("test data 2") { Features = features2 }; mapLineData2.Features.First().MetaData["firstKey"] = 123; mapLineData2.Features.First().MetaData["anotherKey"] = "anotherValue"; using (var writer = new PolylineShapeFileWriter()) { writer.CopyToFeature(mapLineData1); // Call TestDelegate call = () => writer.CopyToFeature(mapLineData2); // Assert const string message = "Column 'anotherKey' does not belong to table ."; TestHelper.AssertThrowsArgumentExceptionAndTestMessage(call, message); } } [Test] public void SaveAs_ValidMapLineData_WritesShapeFile() { // Setup string directoryPath = TestHelper.GetScratchPadPath(nameof(SaveAs_ValidMapLineData_WritesShapeFile)); string filePath = Path.Combine(directoryPath, "test.shp"); const string baseName = "test"; MapFeature[] features = CreateFeatures(0.0); var mapLineData = new MapLineData("test data") { Features = features }; mapLineData.Features.First().MetaData[""] = 123; using (new DirectoryDisposeHelper(TestHelper.GetScratchPadPath(), nameof(SaveAs_ValidMapLineData_WritesShapeFile))) using (var writer = new PolylineShapeFileWriter()) { writer.CopyToFeature(mapLineData); // Call writer.SaveAs(filePath); // Assert string pathName = Path.Combine(directoryPath, baseName); Assert.IsTrue(File.Exists(pathName + ".shp")); Assert.IsTrue(File.Exists(pathName + ".shx")); Assert.IsTrue(File.Exists(pathName + ".dbf")); } } private static MapFeature[] CreateFeatures(double seed) { return new[] { new MapFeature(new[] { new MapGeometry(new[] { new[] { new Point2D(seed, seed), new Point2D(seed + 1, seed + 1), new Point2D(seed + 2, seed + 2) } }) }) }; } } }