// Copyright (C) Stichting Deltares 2024. All rights reserved.
//
// This file is part of the application DAM - UI.
//
// DAM - UI 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.Collections.Generic;
using System.IO;
using Deltares.Dam.Data;
using NUnit.Framework;
namespace Deltares.Dam.Tests
{
[TestFixture]
public class CsvExporterTest
{
const string FileName = "test.csv";
[Test]
public void ThrowWhenFileNameIsNull()
{
Assert.That(() => new CsvExporter(null, null), Throws.InstanceOf());
}
[Test]
public void ThrowWhenCalculationSubDirIsNull()
{
var exportData = new CsvExportData("locationName", "scenarioName");
exportData.BaseFileName = "base";
exportData.CalculationSubDir = "";
Assert.That(() => exportData.GetDesignResultFile(), Throws.InstanceOf());
}
[Test]
public void ThrowWhenFileNameIsNullOrEmpty()
{
Assert.That(() => new CsvExporter("", null), Throws.InstanceOf());
}
[Test]
public void ThrowWhenFileNameOnlyContainsWhitespaces()
{
Assert.That(() => new CsvExporter(" ", null), Throws.InstanceOf());
}
[Test]
public void OverwriteExistingFileShouldWorkWithoutException()
{
const string writtenContent = "something";
StreamWriter file = File.CreateText(FileName);
file.WriteLine(writtenContent);
file.Close();
var exportData = new CsvExportData("locationName", "scenarioName");
var creator = new CsvExporter(FileName, new List
{
exportData
});
creator.WriteFile();
using (StreamReader reader = File.OpenText(FileName))
{
string fileContent = reader.ReadToEnd();
Assert.That(fileContent.Contains(writtenContent), Is.False);
}
}
[Test]
public void ThrowsIndexAlreadyExistsWhenDeclaredMultipleTimes()
{
var creator = new CsvExporter(FileName,
new List
{
new DataStubWithWrongMetaData()
});
Assert.That(() => creator.WriteFile(), Throws.InstanceOf());
}
[Test]
public void ColumnHeaderIsWrittenCorrectly()
{
var dataItems = new List
{
new DataStub
{
ID = 1,
Name = "Barry",
Average = 95.4
},
new DataStub
{
ID = 2,
Name = "Tom",
Average = 97.3
},
new DataStub
{
ID = 3,
Name = "Remko",
Average = 92.2
}
};
var creator = new CsvExporter(FileName, dataItems)
{
WriteHeaderInFirstLine = true
};
creator.WriteFile();
using (StreamReader reader = File.OpenText(FileName))
{
string fileContent = reader.ReadToEnd();
Assert.That(fileContent.Contains("ID;Naam"), Is.True);
}
}
[Test]
public void TestRemoveCarriageReturnsFromMessage()
{
var text = "text1.\r\ntext2.\r\n";
CsvExportData.RemoveCarriageReturnsFromMessage(ref text);
Assert.That(text, Is.EqualTo("text1. text2."));
}
[Test]
public void TestRemoveCarriageReturnsFromResultMessage()
{
var csvExportData = new CsvExportData();
csvExportData.ResultMessage = "text1.\r\ntext2.\r\n";
string message = csvExportData.ResultMessage;
Assert.That(message, Is.EqualTo("text1. text2."));
}
class DataStub
{
[CsvExportColumn("ID", 1)] public int ID { get; set; }
[CsvExportColumn("Naam", 2)] public string Name { get; set; }
[CsvExportColumn("Gemiddelde", 3)] public double Average { get; set; }
}
class DataStubWithWrongMetaData
{
[CsvExportColumn("ID", 1)] public int ID { get; set; }
[CsvExportColumn("Naam", 1)] public string Name { get; set; }
}
}
}