public void FahClientReader_ReadRethrowsExceptionFromStreamReadAndClosesTheConnection()
{
// Arrange
Func <TcpConnection> factory = () => new MockTcpConnection(() => new MockStreamThrowsOnRead());
using (var connection = new FahClientConnection("foo", 2000, new MockTcpConnectionFactory(factory)))
{
connection.Open();
var reader = new FahClientReader(connection);
// Act & Assert
Assert.Throws <IOException>(() => reader.Read());
Assert.IsFalse(connection.Connected);
}
}
public void FahClientCommand_ExecuteAsyncRethrowsExceptionFromStreamWriteAsyncAndClosesTheConnection()
{
// Arrange
Func <TcpConnection> factory = () => new MockTcpConnection(() => new MockStreamThrowsOnWrite());
using (var connection = new FahClientConnection("foo", 2000, new MockTcpConnectionFactory(factory)))
{
connection.Open();
var command = new FahClientCommand(connection);
// Act & Assert
Assert.ThrowsAsync <IOException>(() => command.ExecuteAsync());
Assert.IsFalse(connection.Connected);
}
}
public void FahClientConnection_OpenSuccessfullyAndCloseMultipleTimes()
{
// Arrange
using (var connection = new FahClientConnection("foo", 2000, new MockTcpConnectionFactory()))
{
// Act (Open)
connection.Open();
// Act (Close)
foreach (var _ in Enumerable.Range(0, 3))
{
connection.Close();
}
}
}
public void FahClientConnection_OpenSuccessfullyAndClose()
{
// Arrange
using (var connection = new FahClientConnection("foo", 2000, new MockTcpConnectionFactory()))
{
// Act (Open)
connection.Open();
// Assert
Assert.IsTrue(connection.Connected);
// Act (Close)
connection.Close();
// Assert
Assert.IsFalse(connection.Connected);
}
}
public void FahClientConnection_DisposeClosesInnerTcpConnection()
{
// Arrange
var tcpConnectionFactory = new MockTcpConnectionFactory();
using (var connection = new FahClientConnection("foo", 2000, tcpConnectionFactory))
{
// Act (Open)
connection.Open();
// Assert
Assert.IsTrue(tcpConnectionFactory.TcpConnection.Connected);
}
// Assert
Assert.IsFalse(tcpConnectionFactory.TcpConnection.Connected);
}
public void FahClientReader_OnReadStream_ReturnsZeroWhenBufferIsNull()
{
// Arrange
Func <TcpConnection> factory = () => new MockTcpConnection();
using (var connection = new FahClientConnection("foo", 2000, new MockTcpConnectionFactory(factory)))
{
connection.Open();
var reader = new FahClientReaderReturnsZeroWhenBufferIsNull(connection);
// Act
bool result = reader.Read();
// Assert
Assert.IsFalse(result);
Assert.IsNull(reader.Message);
}
}
public void FahClientReader_ReadReadsNoMessageFromConnection()
{
// Arrange
Func <TcpConnection> factory = () => new MockTcpConnection();
using (var connection = new FahClientConnection("foo", 2000, new MockTcpConnectionFactory(factory)))
{
connection.Open();
var reader = new FahClientReader(connection);
// Act
bool result = reader.Read();
// Assert
Assert.IsFalse(result);
Assert.IsNull(reader.Message);
}
}
public void FahClientCommand_ExecuteWritesNullCommandTextToConnection()
{
// Arrange
var tcpConnectionFactory = new MockTcpConnectionFactory();
using (var connection = new FahClientConnection("foo", 2000, tcpConnectionFactory))
{
connection.Open();
var command = new FahClientCommand(connection, null);
// Act
int bytesWritten = command.Execute();
// Assert
Assert.AreEqual(0, bytesWritten);
var memoryStream = (MemoryStream)tcpConnectionFactory.TcpConnection.GetStream();
Assert.AreEqual("", Encoding.ASCII.GetString(memoryStream.ToArray()));
}
}
public async Task FahClientCommand_ExecuteAsyncWritesCommandTextToConnection()
{
// Arrange
var tcpConnectionFactory = new MockTcpConnectionFactory();
using (var connection = new FahClientConnection("foo", 2000, tcpConnectionFactory))
{
connection.Open();
var command = new FahClientCommand(connection, "command text");
// Act
int bytesWritten = await command.ExecuteAsync();
// Assert
Assert.AreEqual(13, bytesWritten);
var memoryStream = (MemoryStream)tcpConnectionFactory.TcpConnection.GetStream();
Assert.AreEqual("command text\n", Encoding.ASCII.GetString(memoryStream.ToArray()));
}
}
public void FahClientReader_ReadExtractsExistingMessage()
{
// Arrange
Func <TcpConnection> factory = () => new MockTcpConnection();
using (var connection = new FahClientConnection("foo", 2000, new MockTcpConnectionFactory(factory)))
{
connection.Open();
var reader = new FahClientReader(connection, new FahClientMessageExtractorWithMessage(MessageFromStream));
reader.BufferSize = 8;
// Act
bool result = reader.Read();
// Assert
Assert.IsTrue(result);
var message = reader.Message;
Assert.AreEqual(FahClientMessageType.Info, message.Identifier.MessageType);
Assert.AreEqual(MessageFromReader, message.MessageText.ToString());
}
}
public void FahClientConnection_SimulateHFMUpdateCommandsSynchronously()
{
using (var connection = new FahClientConnection(Host, Port))
{
connection.Open();
connection.CreateCommand("log-updates restart").Execute();
connection.CreateCommand("updates add 0 60 $heartbeat").Execute();
connection.CreateCommand("updates add 1 1 $info").Execute();
connection.CreateCommand("updates add 2 1 $(options -a)").Execute();
connection.CreateCommand("updates add 3 1 $slot-info").Execute();
var reader = connection.CreateReader();
for (int i = 0; i < 10 && reader.Read(); i++)
{
Console.WriteLine(reader.Message);
}
}
}
public async Task FahClientReader_ReadAsyncReadsMessageFromConnection()
{
// Arrange
using (var stream = CreateStreamWithMessage())
{
Func <TcpConnection> factory = () => new MockTcpConnection(() => stream);
using (var connection = new FahClientConnection("foo", 2000, new MockTcpConnectionFactory(factory)))
{
connection.Open();
var reader = new FahClientReader(connection);
reader.BufferSize = 8;
// Act
bool result = await reader.ReadAsync();
// Assert
Assert.IsTrue(result);
var message = reader.Message;
Assert.AreEqual(FahClientMessageType.Info, message.Identifier.MessageType);
Assert.AreEqual(MessageFromReader, message.MessageText.ToString());
}
}
}
public void FahClientConnection_WritesCommandsAndReadsMessageSynchronously()
{
using (var connection = new FahClientConnection(Host, Port))
{
connection.Open();
var command = connection.CreateCommand();
command.CommandText = "info";
command.Execute();
var reader = connection.CreateReader();
if (reader.Read())
{
Console.WriteLine(reader.Message);
}
command.CommandText = "log-updates restart";
command.Execute();
if (reader.Read())
{
Console.WriteLine(reader.Message);
}
}
}
public void FahClientConnection_CloseConnectionWhileExecutingCommandSynchronously()
{
using (var connection = new FahClientConnection(Host, Port))
{
connection.Open();
CloseConnectionAfter(1000, connection);
var command = connection.CreateCommand();
command.CommandText = "info";
try
{
// continually execute the command
while (command.Execute() > 0)
{
}
}
catch (Exception ex)
{
Assert.IsFalse(connection.Connected);
Console.WriteLine(ex);
}
}
}
