public void ShouldReturnAcknowledgedHeader()
{
//Arrange
//Create ClientSocket
Socket client = new Socket(AddressFamily.InterNetwork, SocketType.Stream, ProtocolType.Tcp);
//Create server object
TCPServer_Sync server = new TCPServer_Sync(new DummyLogger(), 8181, GetEmptyMessageHandler(), 1);
//Start server
Task.Run(() => server.Start());
//Act
//Create testdata
string teststring = "TestString";
byte[] testmsgdata = Encoding.ASCII.GetBytes(teststring);
var expected = new ApplicationProtocolHeader(testmsgdata.Length, 0);
//Connect to server
client.Connect(new IPEndPoint(IPAddress.Loopback, 8181));
//Send messageheader
client.Send(expected.WrapHeaderData());
byte[] headerbuffer = new byte[8];
//Receive ACK
client.Receive(headerbuffer);
var actual = new ApplicationProtocolHeader(headerbuffer);
Assert.Equal(expected, actual);
}
public void ShouldConnectToServer_Test()
{
//Arrange
//Create ClientSocket
Socket client = new Socket(AddressFamily.InterNetwork, SocketType.Stream, ProtocolType.Tcp);
//Create server object
TCPServer_Sync server = new TCPServer_Sync(new DummyLogger(), 8585, new Dictionary <int, IMessageManager>(), 1);
//Start server
Task.Run(() => server.Start());
//Act
//Connect to server
client.Connect(new IPEndPoint(IPAddress.Loopback, 8585));
//Assert
Assert.True(client.Connected);
}
public void ShouldHandleSentString_Test()
{
//Arrange
//Create message handlers
var stringhandler = new DummyStringHandler();
var handlers = new Dictionary <int, IMessageManager>
{
{ 0, stringhandler }
};
//Create ClientSocket
Socket client = new Socket(AddressFamily.InterNetwork, SocketType.Stream, ProtocolType.Tcp);
//Create server object (Pass in handlers)
TCPServer_Sync server = new TCPServer_Sync(new DummyLogger(), 8181, handlers, 1);
//Start server
Task.Run(() => server.Start());
//Act
//Create testdata
string teststring = "TestString";
byte[] testmsgdata = Encoding.ASCII.GetBytes(teststring);
var header = new ApplicationProtocolHeader(testmsgdata.Length, 0);
//Connect to server
client.Connect(new IPEndPoint(IPAddress.Loopback, 8181));
//Send messageheader
client.Send(header.WrapHeaderData());
byte[] headerbuffer = new byte[8];
//Receive ACK
client.Receive(headerbuffer);
if (header.Equals(new ApplicationProtocolHeader(headerbuffer)))
{
//Send actual message
client.Send(testmsgdata);
}
else
{
return;
}
//Get handled message
while (String.IsNullOrEmpty(stringhandler.HandledString))
{
//Wait for string to get handled
}
//Assert
string expected = teststring;
string actual = stringhandler.HandledString;
Assert.Equal(expected, actual);
}
public void ShouldRejectConnectionOverAllowedThresholdTheory(int maxallowed)
{
//Arrange
SocketError expected = SocketError.ConnectionRefused;
//Create ClientSockets
List <Socket> sockets = new List <Socket>();
for (int i = 0; i < maxallowed; i++)
{
sockets.Add(new Socket(AddressFamily.InterNetwork, SocketType.Stream, ProtocolType.Tcp));
}
Socket failSocket = new Socket(AddressFamily.InterNetwork, SocketType.Stream, ProtocolType.Tcp);
//Create server object
TCPServer_Sync server = new TCPServer_Sync(new DummyLogger(), 8989, new Dictionary <int, IMessageManager>(), maxallowed);
//Act
//Start server
Task.Run(() => server.Start());
//Fill available server slots
foreach (var client in sockets)
{
try
{
//Throw in a wait because the sockets need time to make the connections
client.Connect(new IPEndPoint(IPAddress.Loopback, 8989));
Thread.Sleep(50);
}
catch (Exception)
{
throw;
}
}
//Connect with the socket we expect to fail
// 0 = SocketError.Sucess
SocketError actual = 0;
try
{
failSocket.Connect(IPAddress.Loopback, 8989);
}
catch (SocketException e)
{
actual = e.SocketErrorCode;
}
//Assert
Assert.Equal(expected, actual);
//Notes:
//So it appears that the actual exception type thrown when the connection is refused is a
//''System.Net.Internals.SocketExceptionFactory + ExtendedSocketException''
//which, in a try-catch, somehow translates to a ordinary SocketException,
//But in ''Assert.Throws<SocketException>..'', it does not.
}
public void ShouldShutdownConnectionIfSentMessageSentWithNoHandler()
{
//Arrange
//Create testdata
string teststring = "StringButNoStringMessageHandler";
byte[] testmsgdata = Encoding.ASCII.GetBytes(teststring);
var clientheader = new ApplicationProtocolHeader(testmsgdata.Length, 0);
//Create ClientSocket
Socket client = new Socket(AddressFamily.InterNetwork, SocketType.Stream, ProtocolType.Tcp);
//Create server object
TCPServer_Sync server = new TCPServer_Sync(new DummyLogger(), 6868, GetEmptyMessageHandler(), 1);
////Act
//Start server
Task.Run(() => server.Start());
//Connect to server
client.Connect(new IPEndPoint(IPAddress.Loopback, 6868));
SendDataWithNoMessageHandler();
byte[] by = new byte[8];
client.ReceiveTimeout = 1000;
int receivedbytes = -1;
try
{
receivedbytes = client.Receive(by);
}
catch (Exception)
{
throw;
}
//Receive() usually completes immidietly and returns 0 if
//the remote site has close the connection
Assert.True(receivedbytes == 0);
void SendDataWithNoMessageHandler()
{
//Send messageheader
client.Send(clientheader.WrapHeaderData());
byte[] headerbuffer = new byte[8];
//Receive ACK
client.Receive(headerbuffer);
var returnedheader = new ApplicationProtocolHeader(headerbuffer);
if (clientheader.Equals(returnedheader))
{
client.Send(testmsgdata);
}
}
}
