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