Jon Skeet | 60c059b | 2008-10-23 21:17:56 +0100 | [diff] [blame] | 1 | using System.Collections; |
Jon Skeet | 6803686 | 2008-10-22 13:30:34 +0100 | [diff] [blame] | 2 | using System.Collections.Generic; |
| 3 | using System.IO; |
| 4 | using NUnit.Framework; |
| 5 | using NestedMessage = Google.ProtocolBuffers.TestProtos.TestAllTypes.Types.NestedMessage; |
| 6 | |
| 7 | namespace Google.ProtocolBuffers { |
| 8 | [TestFixture] |
| 9 | public class MessageStreamIteratorTest { |
| 10 | |
| 11 | [Test] |
| 12 | public void ThreeMessagesInMemory() { |
| 13 | MemoryStream stream = new MemoryStream(MessageStreamWriterTest.ThreeMessageData); |
| 14 | IEnumerable<NestedMessage> iterator = MessageStreamIterator<NestedMessage>.FromStreamProvider(() => stream); |
| 15 | List<NestedMessage> messages = new List<NestedMessage>(iterator); |
| 16 | |
| 17 | Assert.AreEqual(3, messages.Count); |
| 18 | Assert.AreEqual(5, messages[0].Bb); |
| 19 | Assert.AreEqual(1500, messages[1].Bb); |
| 20 | Assert.IsFalse(messages[2].HasBb); |
| 21 | } |
| 22 | } |
| 23 | } |