diff --git a/source/Halibut.Tests/Halibut.Tests.csproj b/source/Halibut.Tests/Halibut.Tests.csproj index ffc36306b..58133f89e 100644 --- a/source/Halibut.Tests/Halibut.Tests.csproj +++ b/source/Halibut.Tests/Halibut.Tests.csproj @@ -10,7 +10,7 @@ enable - net48;net6.0 + net48;net6.0 net6.0 diff --git a/source/Halibut.Tests/UsageFixture.cs b/source/Halibut.Tests/UsageFixture.cs index 8a2417d74..384000636 100644 --- a/source/Halibut.Tests/UsageFixture.cs +++ b/source/Halibut.Tests/UsageFixture.cs @@ -81,7 +81,8 @@ public async Task OctopusCanSendMessagesToTentacle_WithSupportedServices(ClientA } [Test] - [LatestAndPreviousClientAndServiceVersionsTestCases] + //[LatestAndPreviousClientAndServiceVersionsTestCases()] + [LatestClientAndLatestServiceTestCases()] [FailedWebSocketTestsBecomeInconclusive] public async Task StreamsCanBeSent(ClientAndServiceTestCase clientAndServiceTestCase) { diff --git a/source/Halibut/Transport/Protocol/MessageExchangeStream.cs b/source/Halibut/Transport/Protocol/MessageExchangeStream.cs index 6adeed809..03156d8a2 100644 --- a/source/Halibut/Transport/Protocol/MessageExchangeStream.cs +++ b/source/Halibut/Transport/Protocol/MessageExchangeStream.cs @@ -182,10 +182,13 @@ public RemoteIdentity ReadRemoteIdentity() public void Send(T message) { + Stream stream = this.stream; + stream = new BufferedStream(stream, 10000000); using (var capture = StreamCapture.New()) { serializer.WriteMessage(stream, message); - WriteEachStream(capture.SerializedStreams); + WriteEachStream(capture.SerializedStreams, stream); + stream.Flush(); } log.Write(EventType.Diagnostic, "Sent: {0}", message); @@ -275,7 +278,7 @@ static DataStream FindStreamById(StreamCapture capture, Guid id) return dataStream; } - void WriteEachStream(IEnumerable streams) + void WriteEachStream(IEnumerable streams, Stream stream) { foreach (var dataStream in streams) {