diff options
Diffstat (limited to 'ThirdParty/CsvHelper-master/tests/CsvHelper.Tests/CsvReaderDefaultValueTests.cs')
-rw-r--r-- | ThirdParty/CsvHelper-master/tests/CsvHelper.Tests/CsvReaderDefaultValueTests.cs | 106 |
1 files changed, 106 insertions, 0 deletions
diff --git a/ThirdParty/CsvHelper-master/tests/CsvHelper.Tests/CsvReaderDefaultValueTests.cs b/ThirdParty/CsvHelper-master/tests/CsvHelper.Tests/CsvReaderDefaultValueTests.cs new file mode 100644 index 0000000..ba5d832 --- /dev/null +++ b/ThirdParty/CsvHelper-master/tests/CsvHelper.Tests/CsvReaderDefaultValueTests.cs @@ -0,0 +1,106 @@ +// 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.Globalization; +using System.IO; +using System.Linq; +using System.Text; +using CsvHelper.Configuration; +using Xunit; + +namespace CsvHelper.Tests +{ + + public class CsvReaderDefaultValueTests + { + [Fact] + public void DefaultValueTest() + { + using (var stream = new MemoryStream()) + using (var writer = new StreamWriter(stream)) + using (var reader = new StreamReader(stream)) + using (var csvReader = new CsvReader(reader, CultureInfo.InvariantCulture)) + { + writer.WriteLine("Id,Name,Order"); + writer.WriteLine(",,"); + writer.WriteLine("2,two,2"); + writer.WriteLine(",three,"); + writer.Flush(); + stream.Position = 0; + + csvReader.Context.RegisterClassMap<TestMap>(); + + var records = csvReader.GetRecords<Test>().ToList(); + + var record = records[0]; + Assert.Equal(-1, record.Id); + Assert.Null(record.Name); + Assert.Equal(-2, record.Order); + + record = records[1]; + Assert.Equal(2, record.Id); + Assert.Equal("two", record.Name); + } + } + + [Fact] + public void DefaultStringValueTest() + { + using (var stream = new MemoryStream()) + using (var writer = new StreamWriter(stream)) + using (var reader = new StreamReader(stream)) + using (var csvReader = new CsvReader(reader, CultureInfo.InvariantCulture)) + { + writer.WriteLine("Id,Name,Order"); + writer.WriteLine(",,"); + writer.WriteLine("2,two,2"); + writer.WriteLine(",three,"); + writer.Flush(); + stream.Position = 0; + + csvReader.Context.RegisterClassMap<TestStringMap>(); + + var records = csvReader.GetRecords<Test>().ToList(); + + var record = records[0]; + Assert.Equal(-1, record.Id); + Assert.Null(record.Name); + Assert.Equal(-2, record.Order); + + record = records[1]; + Assert.Equal(2, record.Id); + Assert.Equal("two", record.Name); + } + } + + private class Test + { + public int Id { get; set; } + + public string Name { get; set; } + + public int Order { get; set; } + } + + private sealed class TestMap : ClassMap<Test> + { + public TestMap() + { + Map(m => m.Id).Default(-1); + Map(m => m.Name).Default((string)null); + Map(m => m.Order).Default(-2); + } + } + + private sealed class TestStringMap : ClassMap<Test> + { + public TestStringMap() + { + Map(m => m.Id).Default("-1"); + Map(m => m.Name).Default(null); + Map(m => m.Order).Default("-2"); + } + } + } +} |