SstCloseTest() {
plugins.load( "tcpsst" );
uint32 randport = 3000 + (uint32)(Sirikata::Task::LocalTime::now().raw() % 20000);
mPort = boost::lexical_cast<std::string>(randport);
mBytes=65536;
mChunks=3;
mOffset=1;
mSendService = new IOServicePool(4);
mSendStrand = mSendService->service()->createStrand();
mRecvService = new IOServicePool(4);
mRecvStrand = mRecvService->service()->createStrand();
mListener = StreamListenerFactory::getSingleton().getDefaultConstructor()(mRecvStrand,StreamListenerFactory::getSingleton().getDefaultOptionParser()(String()));
using std::tr1::placeholders::_1;
using std::tr1::placeholders::_2;
mListener->listen(Address("127.0.0.1",mPort),std::tr1::bind(&SstCloseTest::listenerNewStreamCallback,this,_1,_2));
mRecvService->run();
#ifdef _WIN32
Sleep(1000);
#else
sleep(1);
#endif
}
void listenerDataRecvCallback(Stream *s,int id, const Chunk&data, const Stream::PauseReceiveCallback& pauseReceive) {
if (mSendReceiveMap[id]==-1) {
mSendReceiveMap[id]=(unsigned char)data[0];
}
static bool pause=true;
if (rand()<RAND_MAX/5||pause)
pause=!pause;
if (pause) {
mRecvService->service()->post(std::tr1::bind(&Stream::readyRead,s));
pauseReceive();
return;
}
mReceiversData[mSendReceiveMap[id]]+=(int)(data.size()*2);
}
void testConnectSend (void )
{
Stream*z=NULL;
bool doSubstreams=true;
{
Stream *r=StreamFactory::getSingleton().getDefaultConstructor()(mServicePool->service(),StreamFactory::getSingleton().getDefaultOptionParser()(String()));
simpleConnect(r,Address("127.0.0.1",mPort));
runRoutine(r);
if (doSubstreams) {
{
Stream*zz=r->factory();
zz = r->clone(&SstTest::noopSubstream);
if (zz) {
runRoutine(zz);
zz->close();
}
delete zz;
}
using std::tr1::placeholders::_1;
using std::tr1::placeholders::_2;
z=r->clone(std::tr1::bind(&SstTest::testSubstream,this,_1,_2));
if (z) {
runRoutine(z);
}else {
++mDisconCount;
}
}
//wait until done
time_t last_time=time(NULL);
int retry_count=20;
while(mCount<(int)(mMessagesToSend.size()*(doSubstreams?5:2))&&!mAbortTest) {
if (rand()<RAND_MAX/10) {
r->readyRead();
z->readyRead();
{
unique_mutex_lock lck(mMutex);
for(std::vector<Stream*>::iterator i=mStreams.begin(),ie=mStreams.end();i!=ie;++i) {
(*i)->readyRead();
}
}
}
time_t this_time=time(NULL);
if (this_time>last_time+5) {
std::cerr<<"Message Receive Count == "<<mCount.read()<<'\n';
last_time=this_time;
if (--retry_count<=0) {
TS_FAIL("Timeout in receiving messages");
TS_ASSERT_EQUALS(mCount.read(),(int)(mMessagesToSend.size()*(doSubstreams?5:2)));
break;
}
}
}
TS_ASSERT(mAbortTest==false);
for (std::map<unsigned int,std::vector<Sirikata::Network::Chunk> >::iterator datamapiter=mDataMap.begin();
datamapiter!=mDataMap.end();
++datamapiter) {
validateVector(datamapiter->first,datamapiter->second,mMessagesToSend);
}
r->close();
delete r;
}
if( doSubstreams){
z->close();
time_t last_time=time(NULL);
int retry_count=6;
while(mDisconCount.read()<3){
if (rand()<RAND_MAX/10) {
z->readyRead();
{
unique_mutex_lock lck(mMutex);
for (ptrdiff_t i=((ptrdiff_t)mStreams.size())-1;i>=0;--i) {
mStreams[i]->readyRead();
}
}
}
time_t this_time=time(NULL);
if (this_time>last_time+5) {
std::cerr<<"Close Receive Count == "<<mDisconCount.read()<<'\n';
last_time=this_time;
if (--retry_count<=0) {
TS_FAIL("Timeout in receiving close signals");
TS_ASSERT_EQUALS(mDisconCount.read(),3);
break;
}
}
}
delete z;
}
time_t last_time=time(NULL);
int retry_count=3;
while(mEndCount.read()<1){//checking for that final call to newSubstream
time_t this_time=time(NULL);
if (rand()<RAND_MAX/10) {
{
unique_mutex_lock lck(mMutex);
for (ptrdiff_t i=((ptrdiff_t)mStreams.size())-1;i>=0;--i) {
mStreams[i]->readyRead();
}
}
}
if (this_time>last_time+5) {
std::cerr<<"SubStream End Receive Count == "<<mEndCount.read()<<'\n';
last_time=this_time;
if (--retry_count<=0) {
TS_FAIL("Timeout in receiving newSubstream functor deallocation request");
TS_ASSERT_EQUALS(mEndCount.read(),1);
break;
}
}
}
}
SstTest():mCount(0),mDisconCount(0),mEndCount(0),ENDSTRING("T end"),mAbortTest(false) {
Sirikata::PluginManager plugins;
plugins.load( Sirikata::DynamicLibrary::filename("tcpsst") );
mPort="9142";
mServicePool = new IOServicePool(1);
mListener = StreamListenerFactory::getSingleton().getDefaultConstructor()(mServicePool->service(),StreamListenerFactory::getSingleton().getDefaultOptionParser()(String()));
using std::tr1::placeholders::_1;
using std::tr1::placeholders::_2;
mListener->listen(Address("127.0.0.1",mPort),std::tr1::bind(&SstTest::listenerNewStreamCallback,this,0,_1,_2));
mServicePool->run();
bool doUnorderedTest=true;
bool doShortTest=false;
if (doUnorderedTest){
mMessagesToSend.push_back("U:0");
mMessagesToSend.push_back("U:1");
mMessagesToSend.push_back("U:2");
}
mMessagesToSend.push_back("T0th");
mMessagesToSend.push_back("T1st");
mMessagesToSend.push_back("T2nd");
mMessagesToSend.push_back("T3rd");
mMessagesToSend.push_back("T4th");
mMessagesToSend.push_back("T5th");
if (doUnorderedTest){
mMessagesToSend.push_back("U:3");
mMessagesToSend.push_back("U:4");
mMessagesToSend.push_back("U:5");
}
mMessagesToSend.push_back("T6th");
mMessagesToSend.push_back("T7th");
mMessagesToSend.push_back("T8th");
mMessagesToSend.push_back("T9th");
if (!doShortTest) {
std::string test("T");
for (unsigned int i=0;i<16385;++i) {
test+=(char)((i+5)%128);
}
mMessagesToSend.push_back(test);
if (doUnorderedTest){
test[0]='U';
mMessagesToSend.push_back(test);
}
for (unsigned int i=0;i<256*1024;++i) {
test+=(char)((rand())%256);
}
if (doUnorderedTest){
mMessagesToSend.push_back(test);
}
test[0]='T';
mMessagesToSend.push_back(test);
int pattern[40]={127,257,511,65537,129,16383,254,255,16384,256,16385,32767,1440,32768,32769,65535,65536,1401,1399,1280,
228,25215,5141,627,151,1,4,124240,3,296,114385,25,17,47,42,24222,655,1441,1439,1024};
for (size_t i=0;i<sizeof(pattern)/sizeof(pattern[0]);++i) {
std::string pat="T";
for (int j=0;j<pattern[i];++j) {
pat+='~'-(i%((pattern[i]%90)+3));
}
mMessagesToSend.push_back(pat);
pat[0]='U';
mMessagesToSend.push_back(pat);
}
mMessagesToSend.push_back("T_0th");
mMessagesToSend.push_back("T_1st");
mMessagesToSend.push_back("T_2nd");
mMessagesToSend.push_back("T_3rd");
mMessagesToSend.push_back("T_4th");
mMessagesToSend.push_back("T_5th");
if (doUnorderedTest){
mMessagesToSend.push_back("U:6");
mMessagesToSend.push_back("U:7");
mMessagesToSend.push_back("U:8");
mMessagesToSend.push_back("U:9");
mMessagesToSend.push_back("U:A");
}
mMessagesToSend.push_back("T_6th");
mMessagesToSend.push_back("T_7th");
if (doUnorderedTest){
mMessagesToSend.push_back("U:B");
}
mMessagesToSend.push_back("T_8th");
if (doUnorderedTest){
mMessagesToSend.push_back("U:C");
}
mMessagesToSend.push_back("T_9th");
if (doUnorderedTest){
mMessagesToSend.push_back("U:D");
}
mMessagesToSend.push_back("The green grasshopper fetched.");
if (doUnorderedTest){
mMessagesToSend.push_back("U:E");
mMessagesToSend.push_back("U:F");
mMessagesToSend.push_back("U:G");
}
mMessagesToSend.push_back("T A blade of grass.");
mMessagesToSend.push_back("T From the playground .");
mMessagesToSend.push_back("T Grounds test test test this is a test test test this is a test test test this is a test test test test and the test is proceeding until it reaches signific length with a string that long however. this is not quite long enough to trigger the high water mark--well now it is I believe to the best of my abilities");
mMessagesToSend.push_back("T Grounds test test test this is a test test test this is a test test test this is a test test test test and the test is proceeding until it reaches signific length with a string that long however. this is not quite");
if (doUnorderedTest){
mMessagesToSend.push_back("U:H");
mMessagesToSend.push_back("U:I");
mMessagesToSend.push_back("U:J");
mMessagesToSend.push_back("U:K");
mMessagesToSend.push_back("U:L");
mMessagesToSend.push_back("U:M");
mMessagesToSend.push_back("U:N");
mMessagesToSend.push_back("U:O");
mMessagesToSend.push_back("U:P");
mMessagesToSend.push_back("U:Q");
}
}
mMessagesToSend.push_back(ENDSTRING);
}