Skip to content

Commit e6e6741

Browse files
committed
test: 增加单元测试
1 parent de60e63 commit e6e6741

File tree

1 file changed

+15
-0
lines changed

1 file changed

+15
-0
lines changed

test/UnitTest/Services/TcpSocketFactoryTest.cs

Lines changed: 15 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -143,12 +143,27 @@ public async Task ReceiveAsync_Error()
143143
Assert.Equal(1024 * 20, client.ReceiveBufferSize);
144144

145145
client.SetDataHandler(new MockReceiveErrorHandler());
146+
147+
ReadOnlyMemory<byte> buffer = ReadOnlyMemory<byte>.Empty;
148+
var tcs = new TaskCompletionSource();
149+
150+
// 增加接收回调方法
151+
client.ReceivedCallBack = b =>
152+
{
153+
buffer = b;
154+
tcs.SetResult();
155+
return ValueTask.CompletedTask;
156+
};
157+
146158
await client.ConnectAsync("localhost", port);
147159

148160
// 发送数据导致接收数据异常
149161
var data = new ReadOnlyMemory<byte>([1, 2, 3, 4, 5]);
150162
await client.SendAsync(data);
151163

164+
await tcs.Task;
165+
Assert.Equal(buffer.ToArray(), [1, 2, 3, 4, 5]);
166+
152167
// 关闭连接
153168
StopTcpServer(server);
154169
}

0 commit comments

Comments
 (0)