Free cookie consent management tool by TermsFeed Policy Generator

source: trunk/sources/HeuristicLab.ExtLibs/HeuristicLab.ProtobufCS/0.9.1/ProtobufCS/src/ProtocolBuffers.Test/MessageStreamIteratorTest.cs @ 5757

Last change on this file since 5757 was 3857, checked in by abeham, 15 years ago

#866

  • Added protobuf-csharp-port project source to ExtLibs
File size: 3.7 KB
Line 
1#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.Collections.Generic;
36using System.IO;
37using Google.ProtocolBuffers.TestProtos;
38using NUnit.Framework;
39using NestedMessage = Google.ProtocolBuffers.TestProtos.TestAllTypes.Types.NestedMessage;
40
41namespace Google.ProtocolBuffers {
42  [TestFixture]
43  public class MessageStreamIteratorTest {
44
45    [Test]
46    public void ThreeMessagesInMemory() {
47      MemoryStream stream = new MemoryStream(MessageStreamWriterTest.ThreeMessageData);     
48      IEnumerable<NestedMessage> iterator = MessageStreamIterator<NestedMessage>.FromStreamProvider(() => stream);
49      List<NestedMessage> messages = new List<NestedMessage>(iterator);
50
51      Assert.AreEqual(3, messages.Count);
52      Assert.AreEqual(5, messages[0].Bb);
53      Assert.AreEqual(1500, messages[1].Bb);
54      Assert.IsFalse(messages[2].HasBb);
55    }
56
57    [Test]
58    public void ManyMessagesShouldNotTriggerSizeAlert() {
59      int messageSize = TestUtil.GetAllSet().SerializedSize;
60      // Enough messages to trigger the alert unless we've reset the size
61      // Note that currently we need to make this big enough to copy two whole buffers,
62      // as otherwise when we refill the buffer the second type, the alert triggers instantly.
63      int correctCount = (CodedInputStream.BufferSize * 2) / messageSize + 1;
64      using (MemoryStream stream = new MemoryStream()) {
65        MessageStreamWriter<TestAllTypes> writer = new MessageStreamWriter<TestAllTypes>(stream);
66        for (int i = 0; i < correctCount; i++) {
67          writer.Write(TestUtil.GetAllSet());
68        }
69        writer.Flush();
70
71        stream.Position = 0;
72
73        int count = 0;
74        foreach (var message in MessageStreamIterator<TestAllTypes>.FromStreamProvider(() => stream)
75          .WithSizeLimit(CodedInputStream.BufferSize * 2)) {
76          count++;
77          TestUtil.AssertAllFieldsSet(message);
78        }
79        Assert.AreEqual(correctCount, count);
80      }
81    }
82  }
83}
Note: See TracBrowser for help on using the repository browser.