1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
|
using System;
using System.Collections.Generic;
using Hazel.Udp;
using Microsoft.VisualStudio.TestTools.UnitTesting;
namespace Hazel.UnitTests
{
[TestClass]
public class UdpReliabilityTests
{
[TestMethod]
public void TestReliableWrapOffByOne()
{
List<MessageReader> messagesReceived = new List<MessageReader>();
UdpConnectionTestHarness dut = new UdpConnectionTestHarness();
dut.DataReceived += evt =>
{
messagesReceived.Add(evt.Message);
};
MessageWriter data = MessageWriter.Get(SendOption.Reliable);
Assert.AreEqual(ushort.MaxValue, dut.ReliableReceiveLast);
SetReliableId(data, 10);
dut.Test_Receive(data);
// This message may not be received if there is an off-by-one error when marking missed pkts up to 10.
SetReliableId(data, 9);
dut.Test_Receive(data);
// Both messages should be received.
Assert.AreEqual(2, messagesReceived.Count);
messagesReceived.Clear();
Assert.AreEqual(2, dut.BytesSent.Count);
dut.BytesSent.Clear();
}
[TestMethod]
public void TestThatAllMessagesAreReceived()
{
List<MessageReader> messagesReceived = new List<MessageReader>();
UdpConnectionTestHarness dut = new UdpConnectionTestHarness();
dut.DataReceived += evt =>
{
messagesReceived.Add(evt.Message);
};
MessageWriter data = MessageWriter.Get(SendOption.Reliable);
for (int i = 0; i < ushort.MaxValue * 2; ++i)
{
// Send a new message, it should be received and ack'd
SetReliableId(data, i);
dut.Test_Receive(data);
// Resend an old message, it should be ignored
if (i > 2)
{
SetReliableId(data, i - 1);
dut.Test_Receive(data);
// It should still be ack'd
Assert.AreEqual(2, dut.BytesSent.Count);
dut.BytesSent.RemoveAt(1);
}
Assert.AreEqual(1, messagesReceived.Count);
messagesReceived.Clear();
Assert.AreEqual(1, dut.BytesSent.Count);
dut.BytesSent.Clear();
}
}
[TestMethod]
public void TestAcksForNotReceivedMessages()
{
List<MessageReader> messagesReceived = new List<MessageReader>();
UdpConnectionTestHarness dut = new UdpConnectionTestHarness();
dut.DataReceived += evt =>
{
messagesReceived.Add(evt.Message);
};
MessageWriter data = MessageWriter.Get(SendOption.Reliable);
SetReliableId(data, 1);
dut.Test_Receive(data);
SetReliableId(data, 3);
dut.Test_Receive(data);
MessageReader ackPacket = dut.BytesSent[1];
// Must be ack
Assert.AreEqual(4, ackPacket.Length);
byte recentPackets = ackPacket.Buffer[3];
// Last packet was not received
Assert.AreEqual(0, recentPackets & 1);
// The packet before that was.
Assert.AreEqual(1, (recentPackets >> 1) & 1);
}
private static void SetReliableId(MessageWriter data, int i)
{
ushort id = (ushort)i;
data.Buffer[1] = (byte)(id >> 8);
data.Buffer[2] = (byte)id;
}
}
}
|
