diff options
Diffstat (limited to 'ThirdParty/CsvHelper-master/tests/CsvHelper.Tests/CsvWriterFormatTests.cs')
-rw-r--r-- | ThirdParty/CsvHelper-master/tests/CsvHelper.Tests/CsvWriterFormatTests.cs | 201 |
1 files changed, 201 insertions, 0 deletions
diff --git a/ThirdParty/CsvHelper-master/tests/CsvHelper.Tests/CsvWriterFormatTests.cs b/ThirdParty/CsvHelper-master/tests/CsvHelper.Tests/CsvWriterFormatTests.cs new file mode 100644 index 0000000..08a2c43 --- /dev/null +++ b/ThirdParty/CsvHelper-master/tests/CsvHelper.Tests/CsvWriterFormatTests.cs @@ -0,0 +1,201 @@ +// Copyright 2009-2022 Josh Close +// This file is a part of CsvHelper and is dual licensed under MS-PL and Apache 2.0. +// See LICENSE.txt for details or visit http://www.opensource.org/licenses/ms-pl.html for MS-PL and http://opensource.org/licenses/Apache-2.0 for Apache 2.0. +// https://github.com/JoshClose/CsvHelper +using System; +using System.Globalization; +using System.IO; +using CsvHelper.Configuration; +using Xunit; + +namespace CsvHelper.Tests +{ + + public class CsvWriterFormatTests + { + [Fact] + public void WriteFieldTest() + { + var record = new TestRecord + { + IntColumn = 1, + DateColumn = new DateTime(2012, 10, 1, 12, 12, 12), + DecimalColumn = 150.99m, + FirstColumn = "first column", + }; + + var stream = new MemoryStream(); + var writer = new StreamWriter(stream) { AutoFlush = true }; + var csv = new CsvWriter(writer, new CultureInfo("en-US")); + csv.Context.RegisterClassMap<TestRecordMap>(); + + csv.WriteRecord(record); + csv.NextRecord(); + + stream.Position = 0; + var reader = new StreamReader(stream); + var csvFile = reader.ReadToEnd(); + var expected = "first column,0001,10/1/2012,$150.99\r\n"; + + Assert.Equal(expected, csvFile); + } + + [Fact] + public void WriteFieldShouldQuoteNoTest() + { + var stream = new MemoryStream(); + var writer = new StreamWriter(stream) { AutoFlush = true }; + var csv = new CsvWriter(writer, CultureInfo.InvariantCulture); + + csv.WriteField("a \"b\" c", false); + csv.NextRecord(); + + stream.Position = 0; + var reader = new StreamReader(stream); + var csvFile = reader.ReadToEnd(); + var expected = "a \"b\" c\r\n"; + + Assert.Equal(expected, csvFile); + } + + [Fact] + public void WriteFieldShouldQuoteYesTest() + { + var stream = new MemoryStream(); + var writer = new StreamWriter(stream) { AutoFlush = true }; + var csv = new CsvWriter(writer, CultureInfo.InvariantCulture); + + csv.WriteField("a \"b\" c", true); + csv.NextRecord(); + + stream.Position = 0; + var reader = new StreamReader(stream); + var csvFile = reader.ReadToEnd(); + var expected = "\"a \"\"b\"\" c\"\r\n"; + + Assert.Equal(expected, csvFile); + } + + [Fact] + public void WriteRecordWithReferencesTest() + { + var record = new Person + { + FirstName = "First Name", + LastName = "Last Name", + Updated = new DateTime(2012, 10, 1, 12, 12, 12, 123), + HomeAddress = new Address + { + Street = "Home Street", + City = "Home City", + State = "Home State", + Zip = 02201, + }, + WorkAddress = new Address + { + Street = "Work Street", + City = "Work City", + State = "Work State", + Zip = 04100, + } + }; + + var stream = new MemoryStream(); + var writer = new StreamWriter(stream) { AutoFlush = true }; + var csv = new CsvWriter(writer, CultureInfo.InvariantCulture); + csv.Context.RegisterClassMap<PersonMap>(); + + csv.WriteRecord(record); + csv.NextRecord(); + + stream.Position = 0; + var reader = new StreamReader(stream); + var csvFile = reader.ReadToEnd(); + + var expected = "First Name,Last Name,2012-10-01 12:12:12.123,Home Street,Home City,Home State,02201,Work Street,Work City,Work State,04100\r\n"; + + Assert.Equal(expected, csvFile); + } + + private class TestRecord + { + public int IntColumn { get; set; } + + public DateTime DateColumn { get; set; } + + public decimal DecimalColumn { get; set; } + + public string FirstColumn { get; set; } + } + + private sealed class TestRecordMap : ClassMap<TestRecord> + { + public TestRecordMap() + { + Map(m => m.IntColumn).Name("Int Column").Index(1).TypeConverterOption.Format("0000"); + Map(m => m.DateColumn).Index(2).TypeConverterOption.Format("d"); + Map(m => m.DecimalColumn).Index(3).TypeConverterOption.Format("c"); + Map(m => m.FirstColumn).Index(0); + } + } + + private class Person + { + public string FirstName { get; set; } + + public string LastName { get; set; } + + public DateTime Updated { get; set; } + + public Address HomeAddress { get; set; } + + public Address WorkAddress { get; set; } + } + + private class Address + { + public string Street { get; set; } + + public string City { get; set; } + + public string State { get; set; } + + public int Zip { get; set; } + } + + private sealed class PersonMap : ClassMap<Person> + { + public PersonMap() + { + Map(m => m.FirstName); + Map(m => m.LastName); + Map(m => m.Updated).TypeConverterOption.Format("yyyy-MM-dd HH:mm:ss.fff"); + References<HomeAddressMap>(m => m.HomeAddress); + References<WorkAddressMap>(m => m.WorkAddress); + } + } + + private sealed class HomeAddressMap : ClassMap<Address> + { + public HomeAddressMap() + { + Map(m => m.Street).Name("HomeStreet"); + Map(m => m.City).Name("HomeCity"); + Map(m => m.State).Name("HomeState"); + Map(m => m.Zip).Name("HomeZip").TypeConverterOption.Format("00000"); + } + } + + private sealed class WorkAddressMap : ClassMap<Address> + { + public WorkAddressMap() + { + Map(m => m.Street).Name("WorkStreet"); + Map(m => m.City).Name("WorkCity"); + Map(m => m.State).Name("WorkState"); + Map(m => m.Zip).Name("WorkZip").TypeConverterOption.Format("00000"); + } + } + + } +} |