void TCPConnectionManagerTest::test_send()
{
// Use Case:
// client requests a connection
// expect client to be registered for any data
StreamDataRequest req;
DataSpec require;
require.addStreamData("testData");
req.addDataOption(require);
QTcpSocket* client = _createClient();
_sendRequest( client, req );
QCoreApplication::processEvents();
CPPUNIT_ASSERT_EQUAL( 1, _server->clientsForStream("testData") );
TestDataBlob blob;
blob.setData("sometestData");
_server->send("testData",&blob);
sleep(1);
CPPUNIT_ASSERT( client->state() == QAbstractSocket::ConnectedState );
boost::shared_ptr<ServerResponse> r = _clientProtocol->receive(*client);
CPPUNIT_ASSERT( r->type() == ServerResponse::Blob );
DataBlobResponse* res = static_cast<DataBlobResponse*>(r.get());
CPPUNIT_ASSERT( res->dataName() == "testData" );
CPPUNIT_ASSERT( res->blobClass() == "TestDataBlob" );
TestDataBlob recvBlob;
recvBlob.deserialise(*client, res->byteOrder());
CPPUNIT_ASSERT(recvBlob == blob);
}
void TCPConnectionManagerTest::test_dataSupportedRequest()
{
// Use Case:
// Send a dataSupport request with no data
// Expect: stream to return a DataSupportResponse
// and add us to the datasupport stream
QString streamInfo("__streamInfo__");
DataSupportRequest req;
QTcpSocket* client = _createClient();
CPPUNIT_ASSERT_EQUAL( 0, _server->clientsForStream(streamInfo) );
_sendRequest( client, req );
QCoreApplication::processEvents();
CPPUNIT_ASSERT_EQUAL( 1, _server->clientsForStream(streamInfo) );
sleep(1);
QCoreApplication::processEvents();
CPPUNIT_ASSERT( client->state() == QAbstractSocket::ConnectedState );
boost::shared_ptr<ServerResponse> r = _clientProtocol->receive( *client );
CPPUNIT_ASSERT( r->type() == ServerResponse::DataSupport );
// Use case:
// New stream type arrives
// Expect:
// to receive a new DataRequest with the new data
QString stream1("stream1");
TestDataBlob blob;
blob.setData("stream1Data");
_server->send(stream1,&blob);
sleep(1);
QCoreApplication::processEvents();
r = _clientProtocol->receive(*client);
CPPUNIT_ASSERT( r->type() == ServerResponse::DataSupport );
DataSupportResponse* res = static_cast<DataSupportResponse*>(r.get());
CPPUNIT_ASSERT_EQUAL( 1, res->streamData().size() );
CPPUNIT_ASSERT( res->streamData().contains(stream1) );
// Use case:
// Existing stream type arrives
// Expect:
// not to receive a new DataRequest
_server->send(stream1,&blob);
sleep(1);
QCoreApplication::processEvents();
r = _clientProtocol->receive(*client);
CPPUNIT_ASSERT( r->type() != ServerResponse::DataSupport );
}
void DataViewerTest::test_integrationWithDataClient()
{
// Use Case:
// full integration test with a rela DataBlobClient
// Setup a test server
QString xml = "<PelicanTCPBlobServer>"
" <connection port=\"0\"/>" // 0 = find unused system port
"</PelicanTCPBlobServer>";
ConfigNode config(xml);
PelicanTCPBlobServer server(config);
usleep(100);
DataBlobClient* client = _client( &server );
// DataViewer viewer(config, address);
// viewer.setClient(client);
TestDataBlob blob;
blob.setData("integration test");
server.send("testData", &blob);
delete client;
}
/**
* @details
*/
void PelicanTCPBlobServerTest::test_connection()
{
// Use Case:
// ?????? TODO
// Expect: TODO
//
// Create and configure TCP server
QString xml = "<PelicanTCPBlobServer>"
" <connection port=\"0\"/>" // 0 = find unused system port
"</PelicanTCPBlobServer>";
ConfigNode config(xml);
PelicanTCPBlobServer server(config);
sleep(1);
// Create a client and connect it to the server
QTcpSocket tcpSocket;
tcpSocket.connectToHost( QHostAddress::LocalHost, server.serverPort() );
if (!tcpSocket.waitForConnected(5000) || tcpSocket.state() == QAbstractSocket::UnconnectedState)
CPPUNIT_FAIL("Client could not connect to server");
// Define the data type which the client will except and send request
StreamDataRequest req;
DataRequirements require;
require.setStreamData("testData");
req.addDataOption(require);
PelicanClientProtocol clientProtocol;
QByteArray data = clientProtocol.serialise(req);
tcpSocket.write(data);
tcpSocket.waitForBytesWritten(data.size());
tcpSocket.flush();
/// ensure we are registered before continuing
while( server.clientsForStream("testData") == 0 )
{
sleep(1);
}
{
// Test Server send
TestDataBlob blob;
blob.setData("Testing TCPServer");
server.send("testData", &blob);
// Evaluate the response from the server
tcpSocket.waitForReadyRead();
boost::shared_ptr<ServerResponse> r = clientProtocol.receive(tcpSocket);
CPPUNIT_ASSERT( r->type() == ServerResponse::Blob );
TestDataBlob blobResult;
blobResult.deserialise(tcpSocket, ((DataBlobResponse*)r.get())->byteOrder());
CPPUNIT_ASSERT(blobResult == blob);
}
{
// Test Server send
TestDataBlob blob;
blob.setData("Testing TCPServer again");
server.send("testData", &blob);
// Evaluate the response from the server
tcpSocket.waitForReadyRead();
boost::shared_ptr<ServerResponse> r = clientProtocol.receive(tcpSocket);
CPPUNIT_ASSERT( r->type() == ServerResponse::Blob );
TestDataBlob blobResult;
blobResult.deserialise(tcpSocket, ((DataBlobResponse*)r.get())->byteOrder());
CPPUNIT_ASSERT(blobResult == blob);
}
}
