diff options
Diffstat (limited to 'ThirdParty/CsvHelper-master/tests/CsvHelper.Tests/Defaults/WritingDefaultsTests.cs')
-rw-r--r-- | ThirdParty/CsvHelper-master/tests/CsvHelper.Tests/Defaults/WritingDefaultsTests.cs | 118 |
1 files changed, 118 insertions, 0 deletions
diff --git a/ThirdParty/CsvHelper-master/tests/CsvHelper.Tests/Defaults/WritingDefaultsTests.cs b/ThirdParty/CsvHelper-master/tests/CsvHelper.Tests/Defaults/WritingDefaultsTests.cs new file mode 100644 index 0000000..85ab8ea --- /dev/null +++ b/ThirdParty/CsvHelper-master/tests/CsvHelper.Tests/Defaults/WritingDefaultsTests.cs @@ -0,0 +1,118 @@ +// 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.Collections.Generic; +using System.Globalization; +using System.IO; +using CsvHelper.Configuration; +using Xunit; + +namespace CsvHelper.Tests.Defaults +{ + + public class WritingDefaultsTests + { + [Fact] + public void EmptyFieldsOnNullReferencePropertyTest() + { + var config = new CsvConfiguration(CultureInfo.InvariantCulture) + { + UseNewObjectForNullReferenceMembers = false, + }; + using (var stream = new MemoryStream()) + using (var reader = new StreamReader(stream)) + using (var writer = new StreamWriter(stream)) + using (var csv = new CsvWriter(writer, config)) + { + var records = new List<A> + { + new A + { + AId = 1, + }, + new A + { + AId = 2, + B = new B + { + BId = 3, + }, + }, + }; + + csv.Context.RegisterClassMap<AMap>(); + csv.WriteRecords(records); + + writer.Flush(); + stream.Position = 0; + + var data = reader.ReadToEnd(); + var expected = "AId,BId,CId\r\n" + + "1,,\r\n" + + "2,3,0\r\n"; + Assert.Equal(expected, data); + } + } + + [Fact] + public void DefaultFieldsOnNullReferencePropertyTest() + { + using (var stream = new MemoryStream()) + using (var reader = new StreamReader(stream)) + using (var writer = new StreamWriter(stream)) + using (var csv = new CsvWriter(writer, CultureInfo.InvariantCulture)) + { + var records = new List<A> + { + new A + { + AId = 1, + }, + new A + { + AId = 2, + B = new B + { + BId = 3, + }, + }, + }; + + csv.Context.RegisterClassMap<AMap>(); + csv.WriteRecords(records); + + writer.Flush(); + stream.Position = 0; + + var data = reader.ReadToEnd(); + var expected = "AId,BId,CId\r\n" + + "1,0,0\r\n" + + "2,3,0\r\n"; + Assert.Equal(expected, data); + } + } + + private class A + { + public int AId { get; set; } + + public B B { get; set; } + } + + private sealed class AMap : ClassMap<A> + { + public AMap() + { + AutoMap(CultureInfo.InvariantCulture); + Map(m => m.AId).Default(1); + } + } + + public class B + { + public int BId { get; set; } + public int CId { get; set; } + } + } +} |