diff options
Diffstat (limited to 'ThirdParty/CsvHelper-master/tests/CsvHelper.Tests/CsvWriterReferenceMappingPrefixTests.cs')
-rw-r--r-- | ThirdParty/CsvHelper-master/tests/CsvHelper.Tests/CsvWriterReferenceMappingPrefixTests.cs | 106 |
1 files changed, 106 insertions, 0 deletions
diff --git a/ThirdParty/CsvHelper-master/tests/CsvHelper.Tests/CsvWriterReferenceMappingPrefixTests.cs b/ThirdParty/CsvHelper-master/tests/CsvHelper.Tests/CsvWriterReferenceMappingPrefixTests.cs new file mode 100644 index 0000000..45a5243 --- /dev/null +++ b/ThirdParty/CsvHelper-master/tests/CsvHelper.Tests/CsvWriterReferenceMappingPrefixTests.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.Collections.Generic; +using System.Globalization; +using System.IO; +using System.Text; +using CsvHelper.Configuration; +using Xunit; + +namespace CsvHelper.Tests +{ + + public class CsvWriterReferenceMappingPrefixTests + { + [Fact] + public void ReferencesWithPrefixTest() + { + 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)) + { + csv.Context.RegisterClassMap<AMap>(); + + var list = new List<A>(); + for (var i = 0; i < 4; i++) + { + var row = i + 1; + list.Add(new A + { + Id = "a" + row, + B = new B + { + Id = "b" + row, + C = new C + { + Id = "c" + row + } + } + }); + } + + csv.WriteRecords(list); + writer.Flush(); + stream.Position = 0; + + var data = reader.ReadToEnd(); + + var expected = new TestStringBuilder(csv.Configuration.NewLine); + expected.AppendLine("Id,BPrefix_Id,C.CId"); + expected.AppendLine("a1,b1,c1"); + expected.AppendLine("a2,b2,c2"); + expected.AppendLine("a3,b3,c3"); + expected.AppendLine("a4,b4,c4"); + Assert.Equal(expected.ToString(), data); + } + } + + private class A + { + public string Id { get; set; } + + public B B { get; set; } + } + + private class B + { + public string Id { get; set; } + + public C C { get; set; } + } + + private class C + { + public string Id { get; set; } + } + + private sealed class AMap : ClassMap<A> + { + public AMap() + { + Map(m => m.Id); + References<BMap>(m => m.B).Prefix("BPrefix_", false); + } + } + + private sealed class BMap : ClassMap<B> + { + public BMap() + { + Map(m => m.Id); + References<CMap>(m => m.C).Prefix(); + } + } + + private sealed class CMap : ClassMap<C> + { + public CMap() + { + Map(m => m.Id).Name("CId"); + } + } + } +} |