blob: ab04ef0a5f3656c1c1694a0f3ebf69a3e5d32670 [file] [log] [blame]
csharptest7d396f92010-11-08 20:06:46 -06001#region Copyright notice and license
2// Protocol Buffers - Google's data interchange format
3// Copyright 2008 Google Inc. All rights reserved.
4// http://github.com/jskeet/dotnet-protobufs/
5// Original C++/Java/Python code:
6// http://code.google.com/p/protobuf/
7//
8// Redistribution and use in source and binary forms, with or without
9// modification, are permitted provided that the following conditions are
10// met:
11//
12// * Redistributions of source code must retain the above copyright
13// notice, this list of conditions and the following disclaimer.
14// * Redistributions in binary form must reproduce the above
15// copyright notice, this list of conditions and the following disclaimer
16// in the documentation and/or other materials provided with the
17// distribution.
18// * Neither the name of Google Inc. nor the names of its
19// contributors may be used to endorse or promote products derived from
20// this software without specific prior written permission.
21//
22// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
23// "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
24// LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
25// A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
26// OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
27// SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
28// LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
29// DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
30// THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
31// (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
32// OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
33#endregion
34
35using System;
36using System.Collections.Generic;
37using Google.ProtocolBuffers;
38using Google.ProtocolBuffers.TestProtos;
39using NUnit.Framework;
40
41namespace Google.ProtocolBuffers {
42 [TestFixture]
43 public class ExtendableMessageTest {
44
45
46 [Test]
47 public void ExtensionWriterTest() {
48 TestAllExtensions.Builder builder = TestAllExtensions.CreateBuilder()
49 .SetExtension(UnitTestProtoFile.DefaultBoolExtension, true)
50 .SetExtension(UnitTestProtoFile.DefaultBytesExtension, ByteString.CopyFromUtf8("123"))
51 .SetExtension(UnitTestProtoFile.DefaultCordExtension, "123")
52 .SetExtension(UnitTestProtoFile.DefaultDoubleExtension, 123)
53 .SetExtension(UnitTestProtoFile.DefaultFixed32Extension, 123u)
54 .SetExtension(UnitTestProtoFile.DefaultFixed64Extension, 123u)
55 .SetExtension(UnitTestProtoFile.DefaultFloatExtension, 123)
56 .SetExtension(UnitTestProtoFile.DefaultForeignEnumExtension, ForeignEnum.FOREIGN_BAZ)
57 .SetExtension(UnitTestProtoFile.DefaultImportEnumExtension, ImportEnum.IMPORT_BAZ)
58 .SetExtension(UnitTestProtoFile.DefaultInt32Extension, 123)
59 .SetExtension(UnitTestProtoFile.DefaultInt64Extension, 123)
60 .SetExtension(UnitTestProtoFile.DefaultNestedEnumExtension, TestAllTypes.Types.NestedEnum.FOO)
61 .SetExtension(UnitTestProtoFile.DefaultSfixed32Extension, 123)
62 .SetExtension(UnitTestProtoFile.DefaultSfixed64Extension, 123)
63 .SetExtension(UnitTestProtoFile.DefaultSint32Extension, 123)
64 .SetExtension(UnitTestProtoFile.DefaultSint64Extension, 123)
65 .SetExtension(UnitTestProtoFile.DefaultStringExtension, "123")
66 .SetExtension(UnitTestProtoFile.DefaultStringPieceExtension, "123")
67 .SetExtension(UnitTestProtoFile.DefaultUint32Extension, 123u)
68 .SetExtension(UnitTestProtoFile.DefaultUint64Extension, 123u)
69 //Optional
70 .SetExtension(UnitTestProtoFile.OptionalBoolExtension, true)
71 .SetExtension(UnitTestProtoFile.OptionalBytesExtension, ByteString.CopyFromUtf8("123"))
72 .SetExtension(UnitTestProtoFile.OptionalCordExtension, "123")
73 .SetExtension(UnitTestProtoFile.OptionalDoubleExtension, 123)
74 .SetExtension(UnitTestProtoFile.OptionalFixed32Extension, 123u)
75 .SetExtension(UnitTestProtoFile.OptionalFixed64Extension, 123u)
76 .SetExtension(UnitTestProtoFile.OptionalFloatExtension, 123)
77 .SetExtension(UnitTestProtoFile.OptionalForeignEnumExtension, ForeignEnum.FOREIGN_BAZ)
78 .SetExtension(UnitTestProtoFile.OptionalImportEnumExtension, ImportEnum.IMPORT_BAZ)
79 .SetExtension(UnitTestProtoFile.OptionalInt32Extension, 123)
80 .SetExtension(UnitTestProtoFile.OptionalInt64Extension, 123)
81 .SetExtension(UnitTestProtoFile.OptionalNestedEnumExtension, TestAllTypes.Types.NestedEnum.FOO)
82 .SetExtension(UnitTestProtoFile.OptionalSfixed32Extension, 123)
83 .SetExtension(UnitTestProtoFile.OptionalSfixed64Extension, 123)
84 .SetExtension(UnitTestProtoFile.OptionalSint32Extension, 123)
85 .SetExtension(UnitTestProtoFile.OptionalSint64Extension, 123)
86 .SetExtension(UnitTestProtoFile.OptionalStringExtension, "123")
87 .SetExtension(UnitTestProtoFile.OptionalStringPieceExtension, "123")
88 .SetExtension(UnitTestProtoFile.OptionalUint32Extension, 123u)
89 .SetExtension(UnitTestProtoFile.OptionalUint64Extension, 123u)
90 //Repeated
91 .AddExtension(UnitTestProtoFile.RepeatedBoolExtension, true)
92 .AddExtension(UnitTestProtoFile.RepeatedBytesExtension, ByteString.CopyFromUtf8("123"))
93 .AddExtension(UnitTestProtoFile.RepeatedCordExtension, "123")
94 .AddExtension(UnitTestProtoFile.RepeatedDoubleExtension, 123)
95 .AddExtension(UnitTestProtoFile.RepeatedFixed32Extension, 123u)
96 .AddExtension(UnitTestProtoFile.RepeatedFixed64Extension, 123u)
97 .AddExtension(UnitTestProtoFile.RepeatedFloatExtension, 123)
98 .AddExtension(UnitTestProtoFile.RepeatedForeignEnumExtension, ForeignEnum.FOREIGN_BAZ)
99 .AddExtension(UnitTestProtoFile.RepeatedImportEnumExtension, ImportEnum.IMPORT_BAZ)
100 .AddExtension(UnitTestProtoFile.RepeatedInt32Extension, 123)
101 .AddExtension(UnitTestProtoFile.RepeatedInt64Extension, 123)
102 .AddExtension(UnitTestProtoFile.RepeatedNestedEnumExtension, TestAllTypes.Types.NestedEnum.FOO)
103 .AddExtension(UnitTestProtoFile.RepeatedSfixed32Extension, 123)
104 .AddExtension(UnitTestProtoFile.RepeatedSfixed64Extension, 123)
105 .AddExtension(UnitTestProtoFile.RepeatedSint32Extension, 123)
106 .AddExtension(UnitTestProtoFile.RepeatedSint64Extension, 123)
107 .AddExtension(UnitTestProtoFile.RepeatedStringExtension, "123")
108 .AddExtension(UnitTestProtoFile.RepeatedStringPieceExtension, "123")
109 .AddExtension(UnitTestProtoFile.RepeatedUint32Extension, 123u)
110 .AddExtension(UnitTestProtoFile.RepeatedUint64Extension, 123u)
111 ;
112 TestAllExtensions msg = builder.Build();
113
114 ExtensionRegistry registry = ExtensionRegistry.CreateInstance();
115 UnitTestProtoFile.RegisterAllExtensions(registry);
116
117 TestAllExtensions.Builder copyBuilder = TestAllExtensions.CreateBuilder().MergeFrom(msg.ToByteArray(), registry);
118 TestAllExtensions copy = copyBuilder.Build();
119
120 Assert.AreEqual(msg.ToByteArray(), copy.ToByteArray());
121
122 Assert.AreEqual(true, copy.GetExtension(UnitTestProtoFile.DefaultBoolExtension));
123 Assert.AreEqual(ByteString.CopyFromUtf8("123"), copy.GetExtension(UnitTestProtoFile.DefaultBytesExtension));
124 Assert.AreEqual("123", copy.GetExtension(UnitTestProtoFile.DefaultCordExtension));
125 Assert.AreEqual(123, copy.GetExtension(UnitTestProtoFile.DefaultDoubleExtension));
126 Assert.AreEqual(123u, copy.GetExtension(UnitTestProtoFile.DefaultFixed32Extension));
127 Assert.AreEqual(123u, copy.GetExtension(UnitTestProtoFile.DefaultFixed64Extension));
128 Assert.AreEqual(123, copy.GetExtension(UnitTestProtoFile.DefaultFloatExtension));
129 Assert.AreEqual(ForeignEnum.FOREIGN_BAZ, copy.GetExtension(UnitTestProtoFile.DefaultForeignEnumExtension));
130 Assert.AreEqual(ImportEnum.IMPORT_BAZ, copy.GetExtension(UnitTestProtoFile.DefaultImportEnumExtension));
131 Assert.AreEqual(123, copy.GetExtension(UnitTestProtoFile.DefaultInt32Extension));
132 Assert.AreEqual(123, copy.GetExtension(UnitTestProtoFile.DefaultInt64Extension));
133 Assert.AreEqual(TestAllTypes.Types.NestedEnum.FOO, copy.GetExtension(UnitTestProtoFile.DefaultNestedEnumExtension));
134 Assert.AreEqual(123, copy.GetExtension(UnitTestProtoFile.DefaultSfixed32Extension));
135 Assert.AreEqual(123, copy.GetExtension(UnitTestProtoFile.DefaultSfixed64Extension));
136 Assert.AreEqual(123, copy.GetExtension(UnitTestProtoFile.DefaultSint32Extension));
137 Assert.AreEqual(123, copy.GetExtension(UnitTestProtoFile.DefaultSint64Extension));
138 Assert.AreEqual("123", copy.GetExtension(UnitTestProtoFile.DefaultStringExtension));
139 Assert.AreEqual("123", copy.GetExtension(UnitTestProtoFile.DefaultStringPieceExtension));
140 Assert.AreEqual(123u, copy.GetExtension(UnitTestProtoFile.DefaultUint32Extension));
141 Assert.AreEqual(123u, copy.GetExtension(UnitTestProtoFile.DefaultUint64Extension));
142
143 Assert.AreEqual(true, copy.GetExtension(UnitTestProtoFile.OptionalBoolExtension));
144 Assert.AreEqual(ByteString.CopyFromUtf8("123"), copy.GetExtension(UnitTestProtoFile.OptionalBytesExtension));
145 Assert.AreEqual("123", copy.GetExtension(UnitTestProtoFile.OptionalCordExtension));
146 Assert.AreEqual(123, copy.GetExtension(UnitTestProtoFile.OptionalDoubleExtension));
147 Assert.AreEqual(123u, copy.GetExtension(UnitTestProtoFile.OptionalFixed32Extension));
148 Assert.AreEqual(123u, copy.GetExtension(UnitTestProtoFile.OptionalFixed64Extension));
149 Assert.AreEqual(123, copy.GetExtension(UnitTestProtoFile.OptionalFloatExtension));
150 Assert.AreEqual(ForeignEnum.FOREIGN_BAZ, copy.GetExtension(UnitTestProtoFile.OptionalForeignEnumExtension));
151 Assert.AreEqual(ImportEnum.IMPORT_BAZ, copy.GetExtension(UnitTestProtoFile.OptionalImportEnumExtension));
152 Assert.AreEqual(123, copy.GetExtension(UnitTestProtoFile.OptionalInt32Extension));
153 Assert.AreEqual(123, copy.GetExtension(UnitTestProtoFile.OptionalInt64Extension));
154 Assert.AreEqual(TestAllTypes.Types.NestedEnum.FOO, copy.GetExtension(UnitTestProtoFile.OptionalNestedEnumExtension));
155 Assert.AreEqual(123, copy.GetExtension(UnitTestProtoFile.OptionalSfixed32Extension));
156 Assert.AreEqual(123, copy.GetExtension(UnitTestProtoFile.OptionalSfixed64Extension));
157 Assert.AreEqual(123, copy.GetExtension(UnitTestProtoFile.OptionalSint32Extension));
158 Assert.AreEqual(123, copy.GetExtension(UnitTestProtoFile.OptionalSint64Extension));
159 Assert.AreEqual("123", copy.GetExtension(UnitTestProtoFile.OptionalStringExtension));
160 Assert.AreEqual("123", copy.GetExtension(UnitTestProtoFile.OptionalStringPieceExtension));
161 Assert.AreEqual(123u, copy.GetExtension(UnitTestProtoFile.OptionalUint32Extension));
162 Assert.AreEqual(123u, copy.GetExtension(UnitTestProtoFile.OptionalUint64Extension));
163
164 Assert.AreEqual(true, copy.GetExtension(UnitTestProtoFile.RepeatedBoolExtension, 0));
165 Assert.AreEqual(ByteString.CopyFromUtf8("123"), copy.GetExtension(UnitTestProtoFile.RepeatedBytesExtension, 0));
166 Assert.AreEqual("123", copy.GetExtension(UnitTestProtoFile.RepeatedCordExtension, 0));
167 Assert.AreEqual(123, copy.GetExtension(UnitTestProtoFile.RepeatedDoubleExtension, 0));
168 Assert.AreEqual(123u, copy.GetExtension(UnitTestProtoFile.RepeatedFixed32Extension, 0));
169 Assert.AreEqual(123u, copy.GetExtension(UnitTestProtoFile.RepeatedFixed64Extension, 0));
170 Assert.AreEqual(123, copy.GetExtension(UnitTestProtoFile.RepeatedFloatExtension, 0));
171 Assert.AreEqual(ForeignEnum.FOREIGN_BAZ, copy.GetExtension(UnitTestProtoFile.RepeatedForeignEnumExtension, 0));
172 Assert.AreEqual(ImportEnum.IMPORT_BAZ, copy.GetExtension(UnitTestProtoFile.RepeatedImportEnumExtension, 0));
173 Assert.AreEqual(123, copy.GetExtension(UnitTestProtoFile.RepeatedInt32Extension, 0));
174 Assert.AreEqual(123, copy.GetExtension(UnitTestProtoFile.RepeatedInt64Extension, 0));
175 Assert.AreEqual(TestAllTypes.Types.NestedEnum.FOO, copy.GetExtension(UnitTestProtoFile.RepeatedNestedEnumExtension, 0));
176 Assert.AreEqual(123, copy.GetExtension(UnitTestProtoFile.RepeatedSfixed32Extension, 0));
177 Assert.AreEqual(123, copy.GetExtension(UnitTestProtoFile.RepeatedSfixed64Extension, 0));
178 Assert.AreEqual(123, copy.GetExtension(UnitTestProtoFile.RepeatedSint32Extension, 0));
179 Assert.AreEqual(123, copy.GetExtension(UnitTestProtoFile.RepeatedSint64Extension, 0));
180 Assert.AreEqual("123", copy.GetExtension(UnitTestProtoFile.RepeatedStringExtension, 0));
181 Assert.AreEqual("123", copy.GetExtension(UnitTestProtoFile.RepeatedStringPieceExtension, 0));
182 Assert.AreEqual(123u, copy.GetExtension(UnitTestProtoFile.RepeatedUint32Extension, 0));
183 Assert.AreEqual(123u, copy.GetExtension(UnitTestProtoFile.RepeatedUint64Extension, 0));
184 }
185
186 }
187}