-
Notifications
You must be signed in to change notification settings - Fork 647
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Added tests to ensure databus messages from Windows to Linux are
processed correctly via the FileShareDataBus
- Loading branch information
1 parent
c41f398
commit 9aa64fa
Showing
2 changed files
with
167 additions
and
0 deletions.
There are no files selected for viewing
153 changes: 153 additions & 0 deletions
153
...us.AcceptanceTests/DataBus/When_sending_databus_properties_from_different_environments.cs
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,153 @@ | ||
namespace NServiceBus.AcceptanceTests.DataBus; | ||
|
||
using System; | ||
using System.IO; | ||
using System.Linq; | ||
using System.Threading.Tasks; | ||
using AcceptanceTesting; | ||
using AcceptanceTesting.Customization; | ||
using EndpointTemplates; | ||
using MessageMutator; | ||
using NUnit.Framework; | ||
|
||
public class When_sending_databus_properties_from_different_environments : NServiceBusAcceptanceTest | ||
{ | ||
[Test] | ||
public async Task Should_receive_messages_with_largepayload_correctly_from_windows() | ||
{ | ||
var payloadToSend = new byte[PayloadSize]; | ||
|
||
var context = await Scenario.Define<Context>() | ||
.WithEndpoint<WindowsSender>(b => b.When(session => session.Send(new MyMessageWithLargePayload | ||
{ | ||
Payload = new DataBusProperty<byte[]>(payloadToSend) | ||
}))) | ||
.WithEndpoint<Receiver>() | ||
.Done(c => c.ReceivedPayload != null) | ||
.Run(); | ||
|
||
Assert.AreEqual(payloadToSend, context.ReceivedPayload, "The large payload should be marshalled correctly using the databus"); | ||
} | ||
|
||
[Test] | ||
public async Task Should_receive_messages_with_largepayload_correctly_from_linux() | ||
{ | ||
var payloadToSend = new byte[PayloadSize]; | ||
|
||
var context = await Scenario.Define<Context>() | ||
.WithEndpoint<LinuxSender>(b => b.When(session => session.Send(new MyMessageWithLargePayload | ||
{ | ||
Payload = new DataBusProperty<byte[]>(payloadToSend) | ||
}))) | ||
.WithEndpoint<Receiver>() | ||
.Done(c => c.ReceivedPayload != null) | ||
.Run(); | ||
|
||
Assert.AreEqual(payloadToSend, context.ReceivedPayload, "The large payload should be marshalled correctly using the databus"); | ||
} | ||
|
||
const int PayloadSize = 500; | ||
|
||
public class Context : ScenarioContext | ||
{ | ||
public byte[] ReceivedPayload { get; set; } | ||
} | ||
|
||
public class WindowsSender : EndpointConfigurationBuilder | ||
{ | ||
public WindowsSender() | ||
{ | ||
EndpointSetup<DefaultServer>(builder => | ||
{ | ||
var basePath = Path.Combine(TestContext.CurrentContext.TestDirectory, "databus", "sender"); | ||
builder.UseDataBus<FileShareDataBus, SystemJsonDataBusSerializer>().BasePath(basePath); | ||
builder.UseSerialization<SystemJsonSerializer>(); | ||
builder.ConfigureRouting().RouteToEndpoint(typeof(MyMessageWithLargePayload), typeof(Receiver)); | ||
builder.RegisterMessageMutator(new MutateOutgoingForWindows()); | ||
}); | ||
} | ||
|
||
public class MutateOutgoingForWindows : IMutateOutgoingTransportMessages | ||
{ | ||
public Task MutateOutgoing(MutateOutgoingTransportMessageContext context) | ||
{ | ||
var databusHeaderKey = context.OutgoingHeaders.FirstOrDefault(f => f.Key.StartsWith("NServiceBus.DataBus.", StringComparison.OrdinalIgnoreCase)).Key; | ||
context.OutgoingHeaders[databusHeaderKey] = context.OutgoingHeaders[databusHeaderKey].Replace("/", "\\"); | ||
return Task.CompletedTask; | ||
} | ||
} | ||
} | ||
|
||
|
||
public class LinuxSender : EndpointConfigurationBuilder | ||
{ | ||
public LinuxSender() | ||
{ | ||
EndpointSetup<DefaultServer>(builder => | ||
{ | ||
var basePath = Path.Combine(TestContext.CurrentContext.TestDirectory, "databus", "sender"); | ||
builder.UseDataBus<FileShareDataBus, SystemJsonDataBusSerializer>().BasePath(basePath); | ||
builder.UseSerialization<SystemJsonSerializer>(); | ||
builder.ConfigureRouting().RouteToEndpoint(typeof(MyMessageWithLargePayload), typeof(Receiver)); | ||
builder.RegisterMessageMutator(new MutateOutgoingForLinux()); | ||
}); | ||
} | ||
|
||
public class MutateOutgoingForLinux : IMutateOutgoingTransportMessages | ||
{ | ||
public Task MutateOutgoing(MutateOutgoingTransportMessageContext context) | ||
{ | ||
var databusHeaderKey = context.OutgoingHeaders.FirstOrDefault(f => f.Key.StartsWith("NServiceBus.DataBus.", StringComparison.OrdinalIgnoreCase)).Key; | ||
context.OutgoingHeaders[databusHeaderKey] = context.OutgoingHeaders[databusHeaderKey].Replace("\\", "/"); | ||
return Task.CompletedTask; | ||
} | ||
} | ||
} | ||
|
||
public class Receiver : EndpointConfigurationBuilder | ||
{ | ||
public Receiver() | ||
{ | ||
EndpointSetup<DefaultServer>(builder => | ||
{ | ||
var basePath = Path.Combine(TestContext.CurrentContext.TestDirectory, "databus", "sender"); | ||
builder.UseDataBus<FileShareDataBus, SystemJsonDataBusSerializer>().BasePath(basePath); | ||
builder.UseSerialization<SystemJsonSerializer>(); | ||
builder.RegisterMessageMutator(new Mutator()); | ||
}); | ||
} | ||
|
||
public class MyMessageHandler : IHandleMessages<MyMessageWithLargePayload> | ||
{ | ||
public MyMessageHandler(Context context) | ||
{ | ||
testContext = context; | ||
} | ||
|
||
public Task Handle(MyMessageWithLargePayload messageWithLargePayload, IMessageHandlerContext context) | ||
{ | ||
testContext.ReceivedPayload = messageWithLargePayload.Payload.Value; | ||
|
||
return Task.CompletedTask; | ||
} | ||
|
||
Context testContext; | ||
} | ||
public class Mutator : IMutateIncomingTransportMessages | ||
{ | ||
public Task MutateIncoming(MutateIncomingTransportMessageContext context) | ||
{ | ||
if (context.Body.Length > PayloadSize) | ||
{ | ||
throw new Exception("The message body is too large, which means the DataBus was not used to transfer the payload."); | ||
} | ||
return Task.CompletedTask; | ||
} | ||
} | ||
} | ||
|
||
public class MyMessageWithLargePayload : ICommand | ||
{ | ||
public DataBusProperty<byte[]> Payload { get; set; } | ||
} | ||
} |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters