summaryrefslogtreecommitdiff
path: root/ThirdParty/CsvHelper-master/tests/CsvHelper.Tests/Reading/MultipleHeadersTests.cs
blob: 49a0b22161e0f67058a35f94291a4881fbf95200 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
// 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 CsvHelper.Configuration;
using Xunit;

namespace CsvHelper.Tests.Reading
{
	
	public class MultipleHeadersTests
	{
		[Fact]
		public void ReadWithoutMapTest()
		{
			using (var stream = new MemoryStream())
			using (var reader = new StreamReader(stream))
			using (var writer = new StreamWriter(stream))
			using (var csv = new CsvReader(reader, CultureInfo.InvariantCulture))
			{
				writer.WriteLine("A,B");
				writer.WriteLine("1,one");
				writer.WriteLine("Y,Z");
				writer.WriteLine("two,2");
				writer.Flush();
				stream.Position = 0;

				csv.Read();
				csv.ReadHeader();
				csv.Read();

				Assert.Equal(1, csv.GetField<int>("A"));
				Assert.Equal("one", csv.GetField("B"));

				csv.Read();
				csv.ReadHeader();
				csv.Read();

				Assert.Equal("two", csv.GetField("Y"));
				Assert.Equal(2, csv.GetField<int>("Z"));
			}
		}

		[Fact]
		public void ReadWithMapTest()
		{
			using (var stream = new MemoryStream())
			using (var reader = new StreamReader(stream))
			using (var writer = new StreamWriter(stream))
			using (var csv = new CsvReader(reader, CultureInfo.InvariantCulture))
			{
				writer.WriteLine("A,B");
				writer.WriteLine("1,one");
				writer.WriteLine("Y,Z");
				writer.WriteLine("two,2");
				writer.Flush();
				stream.Position = 0;

				csv.Context.RegisterClassMap<AlphaMap>();
				csv.Context.RegisterClassMap<OmegaMap>();

				csv.Read();
				csv.ReadHeader();

				csv.Read();
				var alphaRecord = csv.GetRecord<Alpha>();

				Assert.Equal(1, alphaRecord.A);
				Assert.Equal("one", alphaRecord.B);

				csv.Read();
				csv.ReadHeader();

				csv.Read();
				var omegaRecord = csv.GetRecord<Omega>();

				Assert.Equal("two", omegaRecord.Y);
				Assert.Equal(2, omegaRecord.Z);
			}
		}

		private class Alpha
		{
			public int A { get; set; }
			public string B { get; set; }
		}

		private class Omega
		{
			public string Y { get; set; }
			public int Z { get; set; }
		}

		private sealed class AlphaMap : ClassMap<Alpha>
		{
			public AlphaMap()
			{
				Map(m => m.A);
				Map(m => m.B);
			}
		}

		private sealed class OmegaMap : ClassMap<Omega>
		{
			public OmegaMap()
			{
				Map(m => m.Y);
				Map(m => m.Z);
			}
		}
	}
}