#region Copyright notice and license // Protocol Buffers - Google's data interchange format // Copyright 2008 Google Inc. All rights reserved. // http://github.com/jskeet/dotnet-protobufs/ // Original C++/Java/Python code: // http://code.google.com/p/protobuf/ // // Redistribution and use in source and binary forms, with or without // modification, are permitted provided that the following conditions are // met: // // * Redistributions of source code must retain the above copyright // notice, this list of conditions and the following disclaimer. // * Redistributions in binary form must reproduce the above // copyright notice, this list of conditions and the following disclaimer // in the documentation and/or other materials provided with the // distribution. // * Neither the name of Google Inc. nor the names of its // contributors may be used to endorse or promote products derived from // this software without specific prior written permission. // // THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS // "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT // LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR // A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT // OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, // SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT // LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, // DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY // THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT // (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE // OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. #endregion using System; using System.Collections.Generic; using Google.ProtocolBuffers; using Google.ProtocolBuffers.TestProtos; using Microsoft.VisualStudio.TestTools.UnitTesting; namespace Google.ProtocolBuffers { [TestClass] public class ExtendableMessageTest { [TestMethod, ExpectedException(typeof(ArgumentException))] public void ExtensionWriterInvalidExtension() { TestPackedExtensions.CreateBuilder()[UnitTestProtoFile.OptionalForeignMessageExtension.Descriptor] = ForeignMessage.DefaultInstance; } [TestMethod] public void ExtensionWriterTest() { TestAllExtensions.Builder builder = TestAllExtensions.CreateBuilder() .SetExtension(UnitTestProtoFile.DefaultBoolExtension, true) .SetExtension(UnitTestProtoFile.DefaultBytesExtension, ByteString.CopyFromUtf8("123")) .SetExtension(UnitTestProtoFile.DefaultCordExtension, "123") .SetExtension(UnitTestProtoFile.DefaultDoubleExtension, 123) .SetExtension(UnitTestProtoFile.DefaultFixed32Extension, 123u) .SetExtension(UnitTestProtoFile.DefaultFixed64Extension, 123u) .SetExtension(UnitTestProtoFile.DefaultFloatExtension, 123) .SetExtension(UnitTestProtoFile.DefaultForeignEnumExtension, ForeignEnum.FOREIGN_BAZ) .SetExtension(UnitTestProtoFile.DefaultImportEnumExtension, ImportEnum.IMPORT_BAZ) .SetExtension(UnitTestProtoFile.DefaultInt32Extension, 123) .SetExtension(UnitTestProtoFile.DefaultInt64Extension, 123) .SetExtension(UnitTestProtoFile.DefaultNestedEnumExtension, TestAllTypes.Types.NestedEnum.FOO) .SetExtension(UnitTestProtoFile.DefaultSfixed32Extension, 123) .SetExtension(UnitTestProtoFile.DefaultSfixed64Extension, 123) .SetExtension(UnitTestProtoFile.DefaultSint32Extension, 123) .SetExtension(UnitTestProtoFile.DefaultSint64Extension, 123) .SetExtension(UnitTestProtoFile.DefaultStringExtension, "123") .SetExtension(UnitTestProtoFile.DefaultStringPieceExtension, "123") .SetExtension(UnitTestProtoFile.DefaultUint32Extension, 123u) .SetExtension(UnitTestProtoFile.DefaultUint64Extension, 123u) //Optional .SetExtension(UnitTestProtoFile.OptionalBoolExtension, true) .SetExtension(UnitTestProtoFile.OptionalBytesExtension, ByteString.CopyFromUtf8("123")) .SetExtension(UnitTestProtoFile.OptionalCordExtension, "123") .SetExtension(UnitTestProtoFile.OptionalDoubleExtension, 123) .SetExtension(UnitTestProtoFile.OptionalFixed32Extension, 123u) .SetExtension(UnitTestProtoFile.OptionalFixed64Extension, 123u) .SetExtension(UnitTestProtoFile.OptionalFloatExtension, 123) .SetExtension(UnitTestProtoFile.OptionalForeignEnumExtension, ForeignEnum.FOREIGN_BAZ) .SetExtension(UnitTestProtoFile.OptionalImportEnumExtension, ImportEnum.IMPORT_BAZ) .SetExtension(UnitTestProtoFile.OptionalInt32Extension, 123) .SetExtension(UnitTestProtoFile.OptionalInt64Extension, 123) .SetExtension(UnitTestProtoFile.OptionalNestedEnumExtension, TestAllTypes.Types.NestedEnum.FOO) .SetExtension(UnitTestProtoFile.OptionalSfixed32Extension, 123) .SetExtension(UnitTestProtoFile.OptionalSfixed64Extension, 123) .SetExtension(UnitTestProtoFile.OptionalSint32Extension, 123) .SetExtension(UnitTestProtoFile.OptionalSint64Extension, 123) .SetExtension(UnitTestProtoFile.OptionalStringExtension, "123") .SetExtension(UnitTestProtoFile.OptionalStringPieceExtension, "123") .SetExtension(UnitTestProtoFile.OptionalUint32Extension, 123u) .SetExtension(UnitTestProtoFile.OptionalUint64Extension, 123u) //Repeated .AddExtension(UnitTestProtoFile.RepeatedBoolExtension, true) .AddExtension(UnitTestProtoFile.RepeatedBytesExtension, ByteString.CopyFromUtf8("123")) .AddExtension(UnitTestProtoFile.RepeatedCordExtension, "123") .AddExtension(UnitTestProtoFile.RepeatedDoubleExtension, 123) .AddExtension(UnitTestProtoFile.RepeatedFixed32Extension, 123u) .AddExtension(UnitTestProtoFile.RepeatedFixed64Extension, 123u) .AddExtension(UnitTestProtoFile.RepeatedFloatExtension, 123) .AddExtension(UnitTestProtoFile.RepeatedForeignEnumExtension, ForeignEnum.FOREIGN_BAZ) .AddExtension(UnitTestProtoFile.RepeatedImportEnumExtension, ImportEnum.IMPORT_BAZ) .AddExtension(UnitTestProtoFile.RepeatedInt32Extension, 123) .AddExtension(UnitTestProtoFile.RepeatedInt64Extension, 123) .AddExtension(UnitTestProtoFile.RepeatedNestedEnumExtension, TestAllTypes.Types.NestedEnum.FOO) .AddExtension(UnitTestProtoFile.RepeatedSfixed32Extension, 123) .AddExtension(UnitTestProtoFile.RepeatedSfixed64Extension, 123) .AddExtension(UnitTestProtoFile.RepeatedSint32Extension, 123) .AddExtension(UnitTestProtoFile.RepeatedSint64Extension, 123) .AddExtension(UnitTestProtoFile.RepeatedStringExtension, "123") .AddExtension(UnitTestProtoFile.RepeatedStringPieceExtension, "123") .AddExtension(UnitTestProtoFile.RepeatedUint32Extension, 123u) .AddExtension(UnitTestProtoFile.RepeatedUint64Extension, 123u) ; TestAllExtensions msg = builder.Build(); ExtensionRegistry registry = ExtensionRegistry.CreateInstance(); UnitTestProtoFile.RegisterAllExtensions(registry); TestAllExtensions.Builder copyBuilder = TestAllExtensions.CreateBuilder().MergeFrom(msg.ToByteArray(), registry); TestAllExtensions copy = copyBuilder.Build(); TestUtil.AssertBytesEqual(msg.ToByteArray(), copy.ToByteArray()); Assert.AreEqual(true, copy.GetExtension(UnitTestProtoFile.DefaultBoolExtension)); Assert.AreEqual(ByteString.CopyFromUtf8("123"), copy.GetExtension(UnitTestProtoFile.DefaultBytesExtension)); Assert.AreEqual("123", copy.GetExtension(UnitTestProtoFile.DefaultCordExtension)); Assert.AreEqual(123, copy.GetExtension(UnitTestProtoFile.DefaultDoubleExtension)); Assert.AreEqual(123u, copy.GetExtension(UnitTestProtoFile.DefaultFixed32Extension)); Assert.AreEqual(123u, copy.GetExtension(UnitTestProtoFile.DefaultFixed64Extension)); Assert.AreEqual(123, copy.GetExtension(UnitTestProtoFile.DefaultFloatExtension)); Assert.AreEqual(ForeignEnum.FOREIGN_BAZ, copy.GetExtension(UnitTestProtoFile.DefaultForeignEnumExtension)); Assert.AreEqual(ImportEnum.IMPORT_BAZ, copy.GetExtension(UnitTestProtoFile.DefaultImportEnumExtension)); Assert.AreEqual(123, copy.GetExtension(UnitTestProtoFile.DefaultInt32Extension)); Assert.AreEqual(123, copy.GetExtension(UnitTestProtoFile.DefaultInt64Extension)); Assert.AreEqual(TestAllTypes.Types.NestedEnum.FOO, copy.GetExtension(UnitTestProtoFile.DefaultNestedEnumExtension)); Assert.AreEqual(123, copy.GetExtension(UnitTestProtoFile.DefaultSfixed32Extension)); Assert.AreEqual(123, copy.GetExtension(UnitTestProtoFile.DefaultSfixed64Extension)); Assert.AreEqual(123, copy.GetExtension(UnitTestProtoFile.DefaultSint32Extension)); Assert.AreEqual(123, copy.GetExtension(UnitTestProtoFile.DefaultSint64Extension)); Assert.AreEqual("123", copy.GetExtension(UnitTestProtoFile.DefaultStringExtension)); Assert.AreEqual("123", copy.GetExtension(UnitTestProtoFile.DefaultStringPieceExtension)); Assert.AreEqual(123u, copy.GetExtension(UnitTestProtoFile.DefaultUint32Extension)); Assert.AreEqual(123u, copy.GetExtension(UnitTestProtoFile.DefaultUint64Extension)); Assert.AreEqual(true, copy.GetExtension(UnitTestProtoFile.OptionalBoolExtension)); Assert.AreEqual(ByteString.CopyFromUtf8("123"), copy.GetExtension(UnitTestProtoFile.OptionalBytesExtension)); Assert.AreEqual("123", copy.GetExtension(UnitTestProtoFile.OptionalCordExtension)); Assert.AreEqual(123, copy.GetExtension(UnitTestProtoFile.OptionalDoubleExtension)); Assert.AreEqual(123u, copy.GetExtension(UnitTestProtoFile.OptionalFixed32Extension)); Assert.AreEqual(123u, copy.GetExtension(UnitTestProtoFile.OptionalFixed64Extension)); Assert.AreEqual(123, copy.GetExtension(UnitTestProtoFile.OptionalFloatExtension)); Assert.AreEqual(ForeignEnum.FOREIGN_BAZ, copy.GetExtension(UnitTestProtoFile.OptionalForeignEnumExtension)); Assert.AreEqual(ImportEnum.IMPORT_BAZ, copy.GetExtension(UnitTestProtoFile.OptionalImportEnumExtension)); Assert.AreEqual(123, copy.GetExtension(UnitTestProtoFile.OptionalInt32Extension)); Assert.AreEqual(123, copy.GetExtension(UnitTestProtoFile.OptionalInt64Extension)); Assert.AreEqual(TestAllTypes.Types.NestedEnum.FOO, copy.GetExtension(UnitTestProtoFile.OptionalNestedEnumExtension)); Assert.AreEqual(123, copy.GetExtension(UnitTestProtoFile.OptionalSfixed32Extension)); Assert.AreEqual(123, copy.GetExtension(UnitTestProtoFile.OptionalSfixed64Extension)); Assert.AreEqual(123, copy.GetExtension(UnitTestProtoFile.OptionalSint32Extension)); Assert.AreEqual(123, copy.GetExtension(UnitTestProtoFile.OptionalSint64Extension)); Assert.AreEqual("123", copy.GetExtension(UnitTestProtoFile.OptionalStringExtension)); Assert.AreEqual("123", copy.GetExtension(UnitTestProtoFile.OptionalStringPieceExtension)); Assert.AreEqual(123u, copy.GetExtension(UnitTestProtoFile.OptionalUint32Extension)); Assert.AreEqual(123u, copy.GetExtension(UnitTestProtoFile.OptionalUint64Extension)); Assert.AreEqual(true, copy.GetExtension(UnitTestProtoFile.RepeatedBoolExtension, 0)); Assert.AreEqual(ByteString.CopyFromUtf8("123"), copy.GetExtension(UnitTestProtoFile.RepeatedBytesExtension, 0)); Assert.AreEqual("123", copy.GetExtension(UnitTestProtoFile.RepeatedCordExtension, 0)); Assert.AreEqual(123, copy.GetExtension(UnitTestProtoFile.RepeatedDoubleExtension, 0)); Assert.AreEqual(123u, copy.GetExtension(UnitTestProtoFile.RepeatedFixed32Extension, 0)); Assert.AreEqual(123u, copy.GetExtension(UnitTestProtoFile.RepeatedFixed64Extension, 0)); Assert.AreEqual(123, copy.GetExtension(UnitTestProtoFile.RepeatedFloatExtension, 0)); Assert.AreEqual(ForeignEnum.FOREIGN_BAZ, copy.GetExtension(UnitTestProtoFile.RepeatedForeignEnumExtension, 0)); Assert.AreEqual(ImportEnum.IMPORT_BAZ, copy.GetExtension(UnitTestProtoFile.RepeatedImportEnumExtension, 0)); Assert.AreEqual(123, copy.GetExtension(UnitTestProtoFile.RepeatedInt32Extension, 0)); Assert.AreEqual(123, copy.GetExtension(UnitTestProtoFile.RepeatedInt64Extension, 0)); Assert.AreEqual(TestAllTypes.Types.NestedEnum.FOO, copy.GetExtension(UnitTestProtoFile.RepeatedNestedEnumExtension, 0)); Assert.AreEqual(123, copy.GetExtension(UnitTestProtoFile.RepeatedSfixed32Extension, 0)); Assert.AreEqual(123, copy.GetExtension(UnitTestProtoFile.RepeatedSfixed64Extension, 0)); Assert.AreEqual(123, copy.GetExtension(UnitTestProtoFile.RepeatedSint32Extension, 0)); Assert.AreEqual(123, copy.GetExtension(UnitTestProtoFile.RepeatedSint64Extension, 0)); Assert.AreEqual("123", copy.GetExtension(UnitTestProtoFile.RepeatedStringExtension, 0)); Assert.AreEqual("123", copy.GetExtension(UnitTestProtoFile.RepeatedStringPieceExtension, 0)); Assert.AreEqual(123u, copy.GetExtension(UnitTestProtoFile.RepeatedUint32Extension, 0)); Assert.AreEqual(123u, copy.GetExtension(UnitTestProtoFile.RepeatedUint64Extension, 0)); } } }