1 #region Copyright notice and license 2 // Protocol Buffers - Google's data interchange format 3 // Copyright 2008 Google Inc. All rights reserved. 4 // 5 // Use of this source code is governed by a BSD-style 6 // license that can be found in the LICENSE file or at 7 // https://developers.google.com/open-source/licenses/bsd 8 #endregion 9 10 using System.IO; 11 using Google.Protobuf.TestProtos; 12 using NUnit.Framework; 13 14 namespace Google.Protobuf 15 { 16 public class UnknownFieldSetTest 17 { 18 public class Data 19 { 20 public static System.Collections.IEnumerable Messages 21 { 22 get 23 { 24 yield return SampleMessages.CreateFullTestAllTypesProto2(); 25 yield return SampleMessages.CreateFullTestAllTypes(); 26 } 27 } 28 } 29 30 [Test] EmptyUnknownFieldSet()31 public void EmptyUnknownFieldSet() 32 { 33 UnknownFieldSet unknownFields = new UnknownFieldSet(); 34 Assert.AreEqual(0, unknownFields.CalculateSize()); 35 } 36 37 [Test] MergeUnknownFieldSet()38 public void MergeUnknownFieldSet() 39 { 40 UnknownFieldSet unknownFields = new UnknownFieldSet(); 41 UnknownField field = new UnknownField(); 42 field.AddFixed32(123); 43 unknownFields.AddOrReplaceField(1, field); 44 UnknownFieldSet otherUnknownFields = new UnknownFieldSet(); 45 Assert.IsFalse(otherUnknownFields.HasField(1)); 46 UnknownFieldSet.MergeFrom(otherUnknownFields, unknownFields); 47 Assert.IsTrue(otherUnknownFields.HasField(1)); 48 } 49 50 [Test] 51 [TestCaseSource(typeof(Data), "Messages")] TestMergeCodedInput(IMessage message)52 public void TestMergeCodedInput(IMessage message) 53 { 54 var emptyMessage = new TestEmptyMessage(); 55 emptyMessage.MergeFrom(message.ToByteArray()); 56 Assert.AreEqual(message.CalculateSize(), emptyMessage.CalculateSize()); 57 Assert.AreEqual(message.ToByteArray(), emptyMessage.ToByteArray()); 58 59 var newMessage = message.Descriptor.Parser.ParseFrom(emptyMessage.ToByteArray()); 60 Assert.AreEqual(message, newMessage); 61 Assert.AreEqual(message.CalculateSize(), newMessage.CalculateSize()); 62 } 63 64 [Test] 65 [TestCaseSource(typeof(Data), "Messages")] TestMergeMessage(IMessage message)66 public void TestMergeMessage(IMessage message) 67 { 68 var emptyMessage = new TestEmptyMessage(); 69 var otherEmptyMessage = new TestEmptyMessage(); 70 emptyMessage.MergeFrom(message.ToByteArray()); 71 otherEmptyMessage.MergeFrom(emptyMessage); 72 73 Assert.AreEqual(message.CalculateSize(), otherEmptyMessage.CalculateSize()); 74 Assert.AreEqual(message.ToByteArray(), otherEmptyMessage.ToByteArray()); 75 } 76 77 [Test] 78 [TestCaseSource(typeof(Data), "Messages")] TestEquals(IMessage message)79 public void TestEquals(IMessage message) 80 { 81 var emptyMessage = new TestEmptyMessage(); 82 var otherEmptyMessage = new TestEmptyMessage(); 83 Assert.AreEqual(emptyMessage, otherEmptyMessage); 84 emptyMessage.MergeFrom(message.ToByteArray()); 85 Assert.AreNotEqual(emptyMessage.CalculateSize(), 86 otherEmptyMessage.CalculateSize()); 87 Assert.AreNotEqual(emptyMessage, otherEmptyMessage); 88 } 89 90 [Test] 91 [TestCaseSource(typeof(Data), "Messages")] TestHashCode(IMessage message)92 public void TestHashCode(IMessage message) 93 { 94 var emptyMessage = new TestEmptyMessage(); 95 int hashCode = emptyMessage.GetHashCode(); 96 emptyMessage.MergeFrom(message.ToByteArray()); 97 Assert.AreNotEqual(hashCode, emptyMessage.GetHashCode()); 98 } 99 100 [Test] 101 [TestCaseSource(typeof(Data), "Messages")] TestClone(IMessage message)102 public void TestClone(IMessage message) 103 { 104 var emptyMessage = new TestEmptyMessage(); 105 TestEmptyMessage otherEmptyMessage = emptyMessage.Clone(); 106 Assert.AreEqual(emptyMessage.CalculateSize(), otherEmptyMessage.CalculateSize()); 107 Assert.AreEqual(emptyMessage.ToByteArray(), otherEmptyMessage.ToByteArray()); 108 109 emptyMessage.MergeFrom(message.ToByteArray()); 110 otherEmptyMessage = emptyMessage.Clone(); 111 Assert.AreEqual(message.CalculateSize(), otherEmptyMessage.CalculateSize()); 112 Assert.AreEqual(message.ToByteArray(), otherEmptyMessage.ToByteArray()); 113 } 114 115 [Test] TestClone_LengthDelimited()116 public void TestClone_LengthDelimited() 117 { 118 var unknownVarintField = new UnknownField(); 119 unknownVarintField.AddVarint(99); 120 121 var unknownLengthDelimitedField1 = new UnknownField(); 122 unknownLengthDelimitedField1.AddLengthDelimited(ByteString.CopyFromUtf8("some data")); 123 124 var unknownLengthDelimitedField2 = new UnknownField(); 125 unknownLengthDelimitedField2.AddLengthDelimited(ByteString.CopyFromUtf8("some more data")); 126 127 var destUnknownFieldSet = new UnknownFieldSet(); 128 destUnknownFieldSet.AddOrReplaceField(997, unknownVarintField); 129 destUnknownFieldSet.AddOrReplaceField(999, unknownLengthDelimitedField1); 130 destUnknownFieldSet.AddOrReplaceField(999, unknownLengthDelimitedField2); 131 132 var clone = UnknownFieldSet.Clone(destUnknownFieldSet); 133 134 Assert.IsTrue(clone.HasField(997)); 135 Assert.IsTrue(clone.HasField(999)); 136 } 137 138 [Test] 139 [TestCaseSource(typeof(Data), "Messages")] TestDiscardUnknownFields(IMessage message)140 public void TestDiscardUnknownFields(IMessage message) 141 { 142 var goldenEmptyMessage = new TestEmptyMessage(); 143 byte[] data = message.ToByteArray(); 144 int fullSize = message.CalculateSize(); 145 146 void AssertEmpty(IMessage msg) 147 { 148 Assert.AreEqual(0, msg.CalculateSize()); 149 Assert.AreEqual(goldenEmptyMessage, msg); 150 } 151 152 void AssertFull(IMessage msg) => Assert.AreEqual(fullSize, msg.CalculateSize()); 153 154 // Test the behavior of the parsers with and without discarding, both generic and non-generic. 155 MessageParser<TestEmptyMessage> retainingParser1 = TestEmptyMessage.Parser; 156 MessageParser retainingParser2 = retainingParser1; 157 MessageParser<TestEmptyMessage> discardingParser1 = retainingParser1.WithDiscardUnknownFields(true); 158 MessageParser discardingParser2 = retainingParser2.WithDiscardUnknownFields(true); 159 160 // Test parse from byte[] 161 MessageParsingHelpers.AssertReadingMessage(retainingParser1, data, m => AssertFull(m)); 162 MessageParsingHelpers.AssertReadingMessage(retainingParser2, data, m => AssertFull(m)); 163 MessageParsingHelpers.AssertReadingMessage(discardingParser1, data, m => AssertEmpty(m)); 164 MessageParsingHelpers.AssertReadingMessage(discardingParser2, data, m => AssertEmpty(m)); 165 166 // Test parse from byte[] with offset 167 AssertFull(retainingParser1.ParseFrom(data, 0, data.Length)); 168 AssertFull(retainingParser2.ParseFrom(data, 0, data.Length)); 169 AssertEmpty(discardingParser1.ParseFrom(data, 0, data.Length)); 170 AssertEmpty(discardingParser2.ParseFrom(data, 0, data.Length)); 171 172 // Test parse from CodedInputStream 173 AssertFull(retainingParser1.ParseFrom(new CodedInputStream(data))); 174 AssertFull(retainingParser2.ParseFrom(new CodedInputStream(data))); 175 AssertEmpty(discardingParser1.ParseFrom(new CodedInputStream(data))); 176 AssertEmpty(discardingParser2.ParseFrom(new CodedInputStream(data))); 177 178 // Test parse from Stream 179 AssertFull(retainingParser1.ParseFrom(new MemoryStream(data))); 180 AssertFull(retainingParser2.ParseFrom(new MemoryStream(data))); 181 AssertEmpty(discardingParser1.ParseFrom(new MemoryStream(data))); 182 AssertEmpty(discardingParser2.ParseFrom(new MemoryStream(data))); 183 } 184 185 [Test] TestReadInvalidWireTypeThrowsInvalidProtocolBufferException()186 public void TestReadInvalidWireTypeThrowsInvalidProtocolBufferException() 187 { 188 MemoryStream ms = new MemoryStream(); 189 CodedOutputStream output = new CodedOutputStream(ms); 190 191 uint tag = WireFormat.MakeTag(1, (WireFormat.WireType)6); 192 output.WriteRawVarint32(tag); 193 output.WriteLength(-1); 194 output.Flush(); 195 ms.Position = 0; 196 197 CodedInputStream input = new CodedInputStream(ms); 198 Assert.AreEqual(tag, input.ReadTag()); 199 200 Assert.Throws<InvalidProtocolBufferException>(() => UnknownFieldSet.MergeFieldFrom(null, input)); 201 } 202 } 203 } 204