Add unit tests for Router configuration and transport layers
- Implemented tests for RouterConfig, RoutingOptions, StaticInstanceConfig, and RouterConfigOptions to ensure default values are set correctly. - Added tests for RouterConfigProvider to validate configurations and ensure defaults are returned when no file is specified. - Created tests for ConfigValidationResult to check success and error scenarios. - Developed tests for ServiceCollectionExtensions to verify service registration for RouterConfig. - Introduced UdpTransportTests to validate serialization, connection, request-response, and error handling in UDP transport. - Added scripts for signing authority gaps and hashing DevPortal SDK snippets.
This commit is contained in:
StellaOps Bot
@@ -0,0 +1,26 @@
|
||||
<Project Sdk="Microsoft.NET.Sdk">
|
||||
|
||||
<PropertyGroup>
|
||||
<TargetFramework>net10.0</TargetFramework>
|
||||
<ImplicitUsings>enable</ImplicitUsings>
|
||||
<Nullable>enable</Nullable>
|
||||
<LangVersion>preview</LangVersion>
|
||||
<IsPackable>false</IsPackable>
|
||||
</PropertyGroup>
|
||||
|
||||
<ItemGroup>
|
||||
<PackageReference Include="Microsoft.NET.Test.Sdk" Version="17.12.0" />
|
||||
<PackageReference Include="xunit" Version="2.9.3" />
|
||||
<PackageReference Include="xunit.runner.visualstudio" Version="3.0.1">
|
||||
<PrivateAssets>all</PrivateAssets>
|
||||
<IncludeAssets>runtime; build; native; contentfiles; analyzers; buildtransitive</IncludeAssets>
|
||||
</PackageReference>
|
||||
<PackageReference Include="Microsoft.Extensions.DependencyInjection" Version="10.0.0-rc.2.25502.107" />
|
||||
<PackageReference Include="Microsoft.Extensions.Logging" Version="10.0.0-rc.2.25502.107" />
|
||||
</ItemGroup>
|
||||
|
||||
<ItemGroup>
|
||||
<ProjectReference Include="..\..\src\__Libraries\StellaOps.Router.Transport.Udp\StellaOps.Router.Transport.Udp.csproj" />
|
||||
</ItemGroup>
|
||||
|
||||
</Project>
|
||||
523
tests/StellaOps.Router.Transport.Udp.Tests/UdpTransportTests.cs
Normal file
523
tests/StellaOps.Router.Transport.Udp.Tests/UdpTransportTests.cs
Normal file
@@ -0,0 +1,523 @@
|
||||
using System.Net;
|
||||
using System.Text;
|
||||
using Microsoft.Extensions.DependencyInjection;
|
||||
using Microsoft.Extensions.Logging;
|
||||
using Microsoft.Extensions.Options;
|
||||
using StellaOps.Router.Common.Abstractions;
|
||||
using StellaOps.Router.Common.Enums;
|
||||
using StellaOps.Router.Common.Models;
|
||||
using Xunit;
|
||||
|
||||
namespace StellaOps.Router.Transport.Udp.Tests;
|
||||
|
||||
public class UdpTransportTests
|
||||
{
|
||||
private static readonly int BasePort = 15100;
|
||||
private static int _portOffset;
|
||||
|
||||
private static int GetNextPort() => BasePort + Interlocked.Increment(ref _portOffset);
|
||||
|
||||
[Fact]
|
||||
public void UdpFrameProtocol_SerializeAndParse_RoundTrip()
|
||||
{
|
||||
// Arrange
|
||||
var originalFrame = new Frame
|
||||
{
|
||||
Type = FrameType.Request,
|
||||
CorrelationId = Guid.NewGuid().ToString("N"),
|
||||
Payload = Encoding.UTF8.GetBytes("Hello, UDP!")
|
||||
};
|
||||
|
||||
// Act
|
||||
var serialized = UdpFrameProtocol.SerializeFrame(originalFrame);
|
||||
var parsed = UdpFrameProtocol.ParseFrame(serialized);
|
||||
|
||||
// Assert
|
||||
Assert.Equal(originalFrame.Type, parsed.Type);
|
||||
Assert.Equal(originalFrame.CorrelationId, parsed.CorrelationId);
|
||||
Assert.Equal(originalFrame.Payload.ToArray(), parsed.Payload.ToArray());
|
||||
}
|
||||
|
||||
[Fact]
|
||||
public void UdpFrameProtocol_ParseFrame_WithEmptyPayload()
|
||||
{
|
||||
// Arrange
|
||||
var originalFrame = new Frame
|
||||
{
|
||||
Type = FrameType.Hello,
|
||||
CorrelationId = Guid.NewGuid().ToString("N"),
|
||||
Payload = ReadOnlyMemory<byte>.Empty
|
||||
};
|
||||
|
||||
// Act
|
||||
var serialized = UdpFrameProtocol.SerializeFrame(originalFrame);
|
||||
var parsed = UdpFrameProtocol.ParseFrame(serialized);
|
||||
|
||||
// Assert
|
||||
Assert.Equal(originalFrame.Type, parsed.Type);
|
||||
Assert.Empty(parsed.Payload.ToArray());
|
||||
}
|
||||
|
||||
[Fact]
|
||||
public void UdpFrameProtocol_ParseFrame_ThrowsOnTooSmallDatagram()
|
||||
{
|
||||
// Arrange
|
||||
var tooSmall = new byte[5]; // Less than 17 bytes (1 + 16)
|
||||
|
||||
// Act & Assert
|
||||
Assert.Throws<InvalidOperationException>(() => UdpFrameProtocol.ParseFrame(tooSmall));
|
||||
}
|
||||
|
||||
[Fact]
|
||||
public void PayloadTooLargeException_HasCorrectProperties()
|
||||
{
|
||||
// Arrange & Act
|
||||
var exception = new PayloadTooLargeException(10000, 8192);
|
||||
|
||||
// Assert
|
||||
Assert.Equal(10000, exception.ActualSize);
|
||||
Assert.Equal(8192, exception.MaxSize);
|
||||
Assert.Contains("10000", exception.Message);
|
||||
Assert.Contains("8192", exception.Message);
|
||||
}
|
||||
|
||||
[Fact]
|
||||
public async Task UdpTransportServer_StartsAndStops()
|
||||
{
|
||||
// Arrange
|
||||
var port = GetNextPort();
|
||||
var services = new ServiceCollection();
|
||||
services.AddLogging();
|
||||
services.AddUdpTransportServer(opts =>
|
||||
{
|
||||
opts.Port = port;
|
||||
opts.BindAddress = IPAddress.Loopback;
|
||||
});
|
||||
|
||||
await using var provider = services.BuildServiceProvider();
|
||||
var server = provider.GetRequiredService<UdpTransportServer>();
|
||||
|
||||
// Act
|
||||
await server.StartAsync(CancellationToken.None);
|
||||
await Task.Delay(50);
|
||||
|
||||
// Assert
|
||||
Assert.Equal(0, server.ConnectionCount);
|
||||
|
||||
// Cleanup
|
||||
await server.StopAsync(CancellationToken.None);
|
||||
}
|
||||
|
||||
[Fact]
|
||||
public async Task UdpTransportClient_ConnectsAndDisconnects()
|
||||
{
|
||||
// Arrange
|
||||
var port = GetNextPort();
|
||||
var services = new ServiceCollection();
|
||||
services.AddLogging();
|
||||
services.AddUdpTransportServer(opts =>
|
||||
{
|
||||
opts.Port = port;
|
||||
opts.BindAddress = IPAddress.Loopback;
|
||||
});
|
||||
services.AddUdpTransportClient(opts =>
|
||||
{
|
||||
opts.Host = "127.0.0.1";
|
||||
opts.Port = port;
|
||||
});
|
||||
|
||||
await using var provider = services.BuildServiceProvider();
|
||||
var server = provider.GetRequiredService<UdpTransportServer>();
|
||||
var client = provider.GetRequiredService<UdpTransportClient>();
|
||||
|
||||
await server.StartAsync(CancellationToken.None);
|
||||
await Task.Delay(50);
|
||||
|
||||
// Act
|
||||
var instance = new InstanceDescriptor
|
||||
{
|
||||
InstanceId = "test-instance",
|
||||
ServiceName = "TestService",
|
||||
Version = "1.0.0",
|
||||
Region = "local"
|
||||
};
|
||||
|
||||
await client.ConnectAsync(instance, [], CancellationToken.None);
|
||||
await Task.Delay(100);
|
||||
|
||||
// Assert
|
||||
Assert.Equal(1, server.ConnectionCount);
|
||||
|
||||
// Cleanup
|
||||
await client.DisconnectAsync();
|
||||
await server.StopAsync(CancellationToken.None);
|
||||
}
|
||||
|
||||
[Fact]
|
||||
public async Task UdpTransport_RequestResponse_Works()
|
||||
{
|
||||
// Arrange
|
||||
var port = GetNextPort();
|
||||
var services = new ServiceCollection();
|
||||
services.AddLogging();
|
||||
services.AddUdpTransportServer(opts =>
|
||||
{
|
||||
opts.Port = port;
|
||||
opts.BindAddress = IPAddress.Loopback;
|
||||
});
|
||||
services.AddUdpTransportClient(opts =>
|
||||
{
|
||||
opts.Host = "127.0.0.1";
|
||||
opts.Port = port;
|
||||
});
|
||||
|
||||
await using var provider = services.BuildServiceProvider();
|
||||
var server = provider.GetRequiredService<UdpTransportServer>();
|
||||
var client = provider.GetRequiredService<UdpTransportClient>();
|
||||
|
||||
// Set up server to respond to requests
|
||||
server.OnFrame += (connectionId, frame) =>
|
||||
{
|
||||
if (frame.Type == FrameType.Request)
|
||||
{
|
||||
var responseFrame = new Frame
|
||||
{
|
||||
Type = FrameType.Response,
|
||||
CorrelationId = frame.CorrelationId,
|
||||
Payload = Encoding.UTF8.GetBytes("Response data")
|
||||
};
|
||||
_ = server.SendFrameAsync(connectionId, responseFrame);
|
||||
}
|
||||
};
|
||||
|
||||
await server.StartAsync(CancellationToken.None);
|
||||
await Task.Delay(50);
|
||||
|
||||
var instance = new InstanceDescriptor
|
||||
{
|
||||
InstanceId = "test-instance",
|
||||
ServiceName = "TestService",
|
||||
Version = "1.0.0",
|
||||
Region = "local"
|
||||
};
|
||||
|
||||
await client.ConnectAsync(instance, [], CancellationToken.None);
|
||||
await Task.Delay(100);
|
||||
|
||||
// Act
|
||||
var connectionState = new ConnectionState
|
||||
{
|
||||
ConnectionId = "test",
|
||||
Instance = instance,
|
||||
TransportType = TransportType.Udp
|
||||
};
|
||||
|
||||
var requestFrame = new Frame
|
||||
{
|
||||
Type = FrameType.Request,
|
||||
CorrelationId = Guid.NewGuid().ToString("N"),
|
||||
Payload = Encoding.UTF8.GetBytes("Request data")
|
||||
};
|
||||
|
||||
var response = await client.SendRequestAsync(
|
||||
connectionState,
|
||||
requestFrame,
|
||||
TimeSpan.FromSeconds(5),
|
||||
CancellationToken.None);
|
||||
|
||||
// Assert
|
||||
Assert.Equal(FrameType.Response, response.Type);
|
||||
Assert.Equal("Response data", Encoding.UTF8.GetString(response.Payload.Span));
|
||||
|
||||
// Cleanup
|
||||
await client.DisconnectAsync();
|
||||
await server.StopAsync(CancellationToken.None);
|
||||
}
|
||||
|
||||
[Fact]
|
||||
public async Task UdpTransport_PayloadTooLarge_ThrowsException()
|
||||
{
|
||||
// Arrange
|
||||
var port = GetNextPort();
|
||||
var services = new ServiceCollection();
|
||||
services.AddLogging();
|
||||
services.AddUdpTransportServer(opts =>
|
||||
{
|
||||
opts.Port = port;
|
||||
opts.BindAddress = IPAddress.Loopback;
|
||||
opts.MaxDatagramSize = 100; // Small limit for testing
|
||||
});
|
||||
services.AddUdpTransportClient(opts =>
|
||||
{
|
||||
opts.Host = "127.0.0.1";
|
||||
opts.Port = port;
|
||||
opts.MaxDatagramSize = 100; // Small limit for testing
|
||||
});
|
||||
|
||||
await using var provider = services.BuildServiceProvider();
|
||||
var server = provider.GetRequiredService<UdpTransportServer>();
|
||||
var client = provider.GetRequiredService<UdpTransportClient>();
|
||||
|
||||
await server.StartAsync(CancellationToken.None);
|
||||
await Task.Delay(50);
|
||||
|
||||
var instance = new InstanceDescriptor
|
||||
{
|
||||
InstanceId = "test-instance",
|
||||
ServiceName = "TestService",
|
||||
Version = "1.0.0",
|
||||
Region = "local"
|
||||
};
|
||||
|
||||
await client.ConnectAsync(instance, [], CancellationToken.None);
|
||||
await Task.Delay(100);
|
||||
|
||||
// Act & Assert
|
||||
var connectionState = new ConnectionState
|
||||
{
|
||||
ConnectionId = "test",
|
||||
Instance = instance,
|
||||
TransportType = TransportType.Udp
|
||||
};
|
||||
|
||||
var largePayload = new byte[200]; // Exceeds 100 byte limit
|
||||
var requestFrame = new Frame
|
||||
{
|
||||
Type = FrameType.Request,
|
||||
CorrelationId = Guid.NewGuid().ToString("N"),
|
||||
Payload = largePayload
|
||||
};
|
||||
|
||||
await Assert.ThrowsAsync<PayloadTooLargeException>(() =>
|
||||
client.SendRequestAsync(
|
||||
connectionState,
|
||||
requestFrame,
|
||||
TimeSpan.FromSeconds(5),
|
||||
CancellationToken.None));
|
||||
|
||||
// Cleanup
|
||||
await client.DisconnectAsync();
|
||||
await server.StopAsync(CancellationToken.None);
|
||||
}
|
||||
|
||||
[Fact]
|
||||
public async Task UdpTransport_StreamingNotSupported_ThrowsNotSupportedException()
|
||||
{
|
||||
// Arrange
|
||||
var port = GetNextPort();
|
||||
var services = new ServiceCollection();
|
||||
services.AddLogging();
|
||||
services.AddUdpTransportClient(opts =>
|
||||
{
|
||||
opts.Host = "127.0.0.1";
|
||||
opts.Port = port;
|
||||
});
|
||||
|
||||
await using var provider = services.BuildServiceProvider();
|
||||
var client = provider.GetRequiredService<UdpTransportClient>();
|
||||
|
||||
var connectionState = new ConnectionState
|
||||
{
|
||||
ConnectionId = "test",
|
||||
Instance = new InstanceDescriptor
|
||||
{
|
||||
InstanceId = "test",
|
||||
ServiceName = "TestService",
|
||||
Version = "1.0.0",
|
||||
Region = "local"
|
||||
},
|
||||
TransportType = TransportType.Udp
|
||||
};
|
||||
|
||||
var requestFrame = new Frame
|
||||
{
|
||||
Type = FrameType.Request,
|
||||
CorrelationId = Guid.NewGuid().ToString("N"),
|
||||
Payload = ReadOnlyMemory<byte>.Empty
|
||||
};
|
||||
|
||||
// Act & Assert
|
||||
await Assert.ThrowsAsync<NotSupportedException>(() =>
|
||||
client.SendStreamingAsync(
|
||||
connectionState,
|
||||
requestFrame,
|
||||
Stream.Null,
|
||||
_ => Task.CompletedTask,
|
||||
new PayloadLimits(),
|
||||
CancellationToken.None));
|
||||
}
|
||||
|
||||
[Fact]
|
||||
public async Task UdpTransport_Timeout_ThrowsTimeoutException()
|
||||
{
|
||||
// Arrange
|
||||
var port = GetNextPort();
|
||||
var services = new ServiceCollection();
|
||||
services.AddLogging();
|
||||
services.AddUdpTransportServer(opts =>
|
||||
{
|
||||
opts.Port = port;
|
||||
opts.BindAddress = IPAddress.Loopback;
|
||||
});
|
||||
services.AddUdpTransportClient(opts =>
|
||||
{
|
||||
opts.Host = "127.0.0.1";
|
||||
opts.Port = port;
|
||||
});
|
||||
|
||||
await using var provider = services.BuildServiceProvider();
|
||||
var server = provider.GetRequiredService<UdpTransportServer>();
|
||||
var client = provider.GetRequiredService<UdpTransportClient>();
|
||||
|
||||
// Server doesn't respond to requests (no OnFrame handler)
|
||||
await server.StartAsync(CancellationToken.None);
|
||||
await Task.Delay(50);
|
||||
|
||||
var instance = new InstanceDescriptor
|
||||
{
|
||||
InstanceId = "test-instance",
|
||||
ServiceName = "TestService",
|
||||
Version = "1.0.0",
|
||||
Region = "local"
|
||||
};
|
||||
|
||||
await client.ConnectAsync(instance, [], CancellationToken.None);
|
||||
await Task.Delay(100);
|
||||
|
||||
// Act & Assert
|
||||
var connectionState = new ConnectionState
|
||||
{
|
||||
ConnectionId = "test",
|
||||
Instance = instance,
|
||||
TransportType = TransportType.Udp
|
||||
};
|
||||
|
||||
var requestFrame = new Frame
|
||||
{
|
||||
Type = FrameType.Request,
|
||||
CorrelationId = Guid.NewGuid().ToString("N"),
|
||||
Payload = Encoding.UTF8.GetBytes("Test")
|
||||
};
|
||||
|
||||
await Assert.ThrowsAsync<TimeoutException>(() =>
|
||||
client.SendRequestAsync(
|
||||
connectionState,
|
||||
requestFrame,
|
||||
TimeSpan.FromMilliseconds(100), // Short timeout
|
||||
CancellationToken.None));
|
||||
|
||||
// Cleanup
|
||||
await client.DisconnectAsync();
|
||||
await server.StopAsync(CancellationToken.None);
|
||||
}
|
||||
|
||||
[Fact]
|
||||
public void ServiceCollectionExtensions_RegistersServerCorrectly()
|
||||
{
|
||||
// Arrange
|
||||
var services = new ServiceCollection();
|
||||
services.AddLogging();
|
||||
services.AddUdpTransportServer(opts =>
|
||||
{
|
||||
opts.Port = 5102;
|
||||
});
|
||||
|
||||
// Act
|
||||
var provider = services.BuildServiceProvider();
|
||||
var server = provider.GetService<ITransportServer>();
|
||||
var udpServer = provider.GetService<UdpTransportServer>();
|
||||
|
||||
// Assert
|
||||
Assert.NotNull(server);
|
||||
Assert.NotNull(udpServer);
|
||||
Assert.Same(server, udpServer);
|
||||
}
|
||||
|
||||
[Fact]
|
||||
public void ServiceCollectionExtensions_RegistersClientCorrectly()
|
||||
{
|
||||
// Arrange
|
||||
var services = new ServiceCollection();
|
||||
services.AddLogging();
|
||||
services.AddUdpTransportClient(opts =>
|
||||
{
|
||||
opts.Host = "127.0.0.1";
|
||||
opts.Port = 5102;
|
||||
});
|
||||
|
||||
// Act
|
||||
var provider = services.BuildServiceProvider();
|
||||
var client = provider.GetService<ITransportClient>();
|
||||
var udpClient = provider.GetService<UdpTransportClient>();
|
||||
var microserviceTransport = provider.GetService<IMicroserviceTransport>();
|
||||
|
||||
// Assert
|
||||
Assert.NotNull(client);
|
||||
Assert.NotNull(udpClient);
|
||||
Assert.NotNull(microserviceTransport);
|
||||
Assert.Same(client, udpClient);
|
||||
Assert.Same(microserviceTransport, udpClient);
|
||||
}
|
||||
|
||||
[Fact]
|
||||
public async Task UdpTransport_HeartbeatSent()
|
||||
{
|
||||
// Arrange
|
||||
var port = GetNextPort();
|
||||
var heartbeatReceived = new TaskCompletionSource<bool>();
|
||||
|
||||
var services = new ServiceCollection();
|
||||
services.AddLogging();
|
||||
services.AddUdpTransportServer(opts =>
|
||||
{
|
||||
opts.Port = port;
|
||||
opts.BindAddress = IPAddress.Loopback;
|
||||
});
|
||||
services.AddUdpTransportClient(opts =>
|
||||
{
|
||||
opts.Host = "127.0.0.1";
|
||||
opts.Port = port;
|
||||
});
|
||||
|
||||
await using var provider = services.BuildServiceProvider();
|
||||
var server = provider.GetRequiredService<UdpTransportServer>();
|
||||
var client = provider.GetRequiredService<UdpTransportClient>();
|
||||
|
||||
server.OnFrame += (connectionId, frame) =>
|
||||
{
|
||||
if (frame.Type == FrameType.Heartbeat)
|
||||
{
|
||||
heartbeatReceived.TrySetResult(true);
|
||||
}
|
||||
};
|
||||
|
||||
await server.StartAsync(CancellationToken.None);
|
||||
await Task.Delay(50);
|
||||
|
||||
var instance = new InstanceDescriptor
|
||||
{
|
||||
InstanceId = "test-instance",
|
||||
ServiceName = "TestService",
|
||||
Version = "1.0.0",
|
||||
Region = "local"
|
||||
};
|
||||
|
||||
await client.ConnectAsync(instance, [], CancellationToken.None);
|
||||
await Task.Delay(100);
|
||||
|
||||
// Act
|
||||
await client.SendHeartbeatAsync(new HeartbeatPayload
|
||||
{
|
||||
InstanceId = "test-instance",
|
||||
Status = InstanceHealthStatus.Healthy
|
||||
}, CancellationToken.None);
|
||||
|
||||
// Assert
|
||||
var received = await Task.WhenAny(heartbeatReceived.Task, Task.Delay(1000));
|
||||
Assert.True(heartbeatReceived.Task.IsCompleted);
|
||||
|
||||
// Cleanup
|
||||
await client.DisconnectAsync();
|
||||
await server.StopAsync(CancellationToken.None);
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user