tests/integrated-tests/test-socket.cpp - ChronoSync - Gitiles

 /* -*- Mode: C++; c-file-style: "gnu"; indent-tabs-mode:nil -*- */
 /*
  * Copyright (c) 2012-2014 University of California, Los Angeles
  *
  * This file is part of ChronoSync, synchronization library for distributed realtime
  * applications for NDN.
  *
  * ChronoSync is free software: you can redistribute it and/or modify it under the terms
  * of the GNU General Public License as published by the Free Software Foundation, either
  * version 3 of the License, or (at your option) any later version.
  *
  * ChronoSync is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY;
  * without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
  * PURPOSE.  See the GNU General Public License for more details.
  *
  * You should have received a copy of the GNU General Public License along with
  * ChronoSync, e.g., in COPYING.md file.  If not, see <http://www.gnu.org/licenses/>.
  */

 #include "socket.hpp"

 #include "boost-test.hpp"

 namespace chronosync {
 namespace test {

 using std::string;
 using std::vector;
 using std::map;

 /**
  * @brief Emulate an app that use the Socket class
  *
  * The app has two types of data set: one is simply string while the other is integer array.
  * For each type of data set, the app has a specific fetching strategy.
  */
 class SocketTestApp : noncopyable
 {
 public:
   SocketTestApp(const Name& syncPrefix,
                 const Name& userPrefix,
                 ndn::Face& face,
                 bool isNum)
     : sum(0)
     , socket(syncPrefix,
              userPrefix,
              face,
              isNum ? bind(&SocketTestApp::fetchNumbers, this, _1) :
                      bind(&SocketTestApp::fetchAll, this, _1))
   {
   }

   void
   set(const shared_ptr<const Data>& dataPacket)
   {
     // std::cerr << "set Data" << std::endl;
     Name dataName(dataPacket->getName());
     string str2(reinterpret_cast<const char*>(dataPacket->getContent().value()),
                 dataPacket->getContent().value_size());
     data.insert(make_pair(dataName, str2));
   }

   void
   set(Name name, const char* buf, int len)
   {
     string str2(buf, len);
     data.insert(make_pair(name, str2));
   }

   void
   setNum(const shared_ptr<const Data>& dataPacket)
   {
     // std::cerr << "setNum Data" << std::endl;
     size_t n = dataPacket->getContent().value_size() / 4;
     const uint32_t* numbers = reinterpret_cast<const uint32_t*>(dataPacket->getContent().value());
     for (size_t i = 0; i < n; i++) {
       sum += numbers[i];
     }
   }

   void
   setNum(Name name, const uint8_t* buf, int len)
   {
     BOOST_ASSERT(len >= 4);

     int n = len / 4;
     const uint32_t* numbers = reinterpret_cast<const uint32_t*>(buf);
     for (int i = 0; i < n; i++) {
       sum += numbers[i];
     }
   }

   void
   fetchAll(const vector<MissingDataInfo>& v)
   {
     // std::cerr << "fetchAll" << std::endl;
     for (int i = 0; i < v.size(); i++) {
       for(SeqNo s = v[i].low; s <= v[i].high; ++s) {
         socket.fetchData(v[i].session, s, [this] (const shared_ptr<const Data>& dataPacket) {
             this->set(dataPacket);
           });
       }
     }
   }

   void
   fetchNumbers(const vector<MissingDataInfo> &v)
   {
     // std::cerr << "fetchNumbers" << std::endl;
     for (int i = 0; i < v.size(); i++) {
       for(SeqNo s = v[i].low; s <= v[i].high; ++s) {
         socket.fetchData(v[i].session, s, [this] (const shared_ptr<const Data>& dataPacket) {
             this->setNum(dataPacket);
           });
       }
     }
   }

   string
   toString()
   {
     string str = "\n";
     for (map<Name, string>::iterator it = data.begin(); it != data.end(); ++it) {
       str += "<";
       str += it->first.toUri();
       str += "|";
       str += it->second;
       str += ">";
       str += "\n";
     }

     return str;
   }

   map<ndn::Name, string> data;
   uint32_t sum;
   Socket socket;
 };

 class SocketFixture
 {
 public:
   SocketFixture()
     : syncPrefix("/ndn/broadcast/sync")
     , scheduler(io)
   {
     syncPrefix.appendVersion();
     userPrefix[0] = Name("/user0");
     userPrefix[1] = Name("/user1");
     userPrefix[2] = Name("/user2");

     faces[0] = make_shared<ndn::Face>(ref(io));
     faces[1] = make_shared<ndn::Face>(ref(io));
     faces[2] = make_shared<ndn::Face>(ref(io));
   }

   void
   createSocket(size_t idx, bool isNum)
   {
     app[idx] = make_shared<SocketTestApp>(syncPrefix, userPrefix[idx], ref(*faces[idx]), isNum);
     sessionName[idx] = app[idx]->socket.getLogic().getSessionName();
   }

   void
   publishAppData(size_t idx, const string& data)
   {
     app[idx]->socket.publishData(reinterpret_cast<const uint8_t*>(data.c_str()), data.size(),
                                  ndn::time::milliseconds(1000));
   }

   void
   setAppData(size_t idx, SeqNo seqNo, const string& data)
   {
     Name dataName = sessionName[idx];
     dataName.appendNumber(seqNo);
     app[idx]->set(dataName, data.c_str(), data.size());
   }

   void
   publishAppNum(size_t idx, const uint8_t* buf, size_t size)
   {
     app[idx]->socket.publishData(buf, size, ndn::time::milliseconds(1000));
   }

   void
   setAppNum(size_t idx, SeqNo seqNo, const uint8_t* buf, size_t size)
   {
     Name dataName = sessionName[idx];
     dataName.appendNumber(seqNo);
     app[idx]->setNum(dataName, buf, size);
   }

   void
   check(int round)
   {
     BOOST_CHECK_EQUAL(app[0]->toString(), app[1]->toString());
     BOOST_CHECK_EQUAL(app[0]->toString(), app[2]->toString());
   }

   void
   check2Num(int num)
   {
     BOOST_CHECK_EQUAL(app[0]->sum, app[1]->sum);
     BOOST_CHECK_EQUAL(app[1]->sum, num);
   }

   void
   terminate()
   {
     io.stop();
   }

   Name syncPrefix;
   Name userPrefix[3];
   Name sessionName[3];

   boost::asio::io_service io;
   shared_ptr<ndn::Face> faces[3];
   ndn::Scheduler scheduler;
   shared_ptr<SocketTestApp> app[3];
 };


 BOOST_FIXTURE_TEST_SUITE(SocketTests, SocketFixture)

 BOOST_AUTO_TEST_CASE(BasicData)
 {
   scheduler.scheduleEvent(ndn::time::milliseconds(0),
                           bind(&SocketFixture::createSocket, this, 0, false));

   scheduler.scheduleEvent(ndn::time::milliseconds(50),
                           bind(&SocketFixture::createSocket, this, 1, false));

   scheduler.scheduleEvent(ndn::time::milliseconds(100),
                           bind(&SocketFixture::createSocket, this, 2, false));

   string data0 = "Very funny Scotty, now beam down my clothes";
   scheduler.scheduleEvent(ndn::time::milliseconds(150),
                           bind(&SocketFixture::publishAppData, this, 0, data0));
   scheduler.scheduleEvent(ndn::time::milliseconds(1150),
                           bind(&SocketFixture::setAppData, this, 0, 1, data0));
   scheduler.scheduleEvent(ndn::time::milliseconds(1160),
                           bind(&SocketFixture::check, this, 1));

   string data1 = "Yes, give me that ketchup";
   string data2 = "Don't look conspicuous, it draws fire";
   scheduler.scheduleEvent(ndn::time::milliseconds(1170),
                           bind(&SocketFixture::publishAppData, this, 0, data1));
   scheduler.scheduleEvent(ndn::time::milliseconds(1180),
                           bind(&SocketFixture::publishAppData, this, 0, data2));
   scheduler.scheduleEvent(ndn::time::milliseconds(2150),
                           bind(&SocketFixture::setAppData, this, 0, 2, data1));
   scheduler.scheduleEvent(ndn::time::milliseconds(2160),
                           bind(&SocketFixture::setAppData, this, 0, 3, data2));
   scheduler.scheduleEvent(ndn::time::milliseconds(2170),
                           bind(&SocketFixture::check, this, 2));

   string data3 = "You surf the Internet, I surf the real world";
   string data4 = "I got a fortune cookie once that said 'You like Chinese food'";
   string data5 = "Real men wear pink. Why? Because their wives make them";
   scheduler.scheduleEvent(ndn::time::milliseconds(3180),
                           bind(&SocketFixture::publishAppData, this, 2, data3));
   scheduler.scheduleEvent(ndn::time::milliseconds(3200),
                           bind(&SocketFixture::publishAppData, this, 1, data4));
   scheduler.scheduleEvent(ndn::time::milliseconds(3210),
                           bind(&SocketFixture::publishAppData, this, 1, data5));
   scheduler.scheduleEvent(ndn::time::milliseconds(4710),
                           bind(&SocketFixture::setAppData, this, 2, 1, data3));
   scheduler.scheduleEvent(ndn::time::milliseconds(4720),
                           bind(&SocketFixture::setAppData, this, 1, 1, data4));
   scheduler.scheduleEvent(ndn::time::milliseconds(4730),
                           bind(&SocketFixture::setAppData, this, 1, 2, data5));
   scheduler.scheduleEvent(ndn::time::milliseconds(4800),
                           bind(&SocketFixture::check, this, 3));

   // not sure weither this is simultanous data generation from multiple sources
   string data6 = "Shakespeare says: 'Prose before hos.'";
   string data7 = "Pick good people, talent never wears out";
   scheduler.scheduleEvent(ndn::time::milliseconds(5500),
                           bind(&SocketFixture::publishAppData, this, 0, data6));
   scheduler.scheduleEvent(ndn::time::milliseconds(5500),
                           bind(&SocketFixture::publishAppData, this, 1, data7));
   scheduler.scheduleEvent(ndn::time::milliseconds(6800),
                           bind(&SocketFixture::setAppData, this, 0, 4, data6));
   scheduler.scheduleEvent(ndn::time::milliseconds(6800),
                           bind(&SocketFixture::setAppData, this, 1, 3, data7));
   scheduler.scheduleEvent(ndn::time::milliseconds(6900),
                           bind(&SocketFixture::check, this, 4));

   scheduler.scheduleEvent(ndn::time::milliseconds(7000),
                           bind(&SocketFixture::terminate, this));

   io.run();
 }

 BOOST_AUTO_TEST_CASE(BasicNumber)
 {
   scheduler.scheduleEvent(ndn::time::milliseconds(0),
                           bind(&SocketFixture::createSocket, this, 0, true));
   scheduler.scheduleEvent(ndn::time::milliseconds(50),
                           bind(&SocketFixture::createSocket, this, 1, true));

   uint32_t num1[5] = {0, 1, 2, 3, 4};
   uint8_t* buf1 = reinterpret_cast<uint8_t*>(num1);
   size_t size1 = sizeof(num1);
   scheduler.scheduleEvent(ndn::time::milliseconds(100),
                           bind(&SocketFixture::publishAppNum, this, 0, buf1, size1));
   scheduler.scheduleEvent(ndn::time::milliseconds(150),
                           bind(&SocketFixture::setAppNum, this, 0, 0, buf1, size1));
   scheduler.scheduleEvent(ndn::time::milliseconds(1000),
                           bind(&SocketFixture::check2Num, this, 10));

   uint32_t num2[5] = {9, 7, 2, 1, 1};
   uint8_t* buf2 = reinterpret_cast<uint8_t*>(num2);
   size_t size2 = sizeof(num2);
   scheduler.scheduleEvent(ndn::time::milliseconds(1100),
                           bind(&SocketFixture::publishAppNum, this, 1, buf2, size2));
   scheduler.scheduleEvent(ndn::time::milliseconds(1150),
                           bind(&SocketFixture::setAppNum, this, 1, 0, buf2, size2));
   scheduler.scheduleEvent(ndn::time::milliseconds(2000),
                           bind(&SocketFixture::check2Num, this, 30));

   scheduler.scheduleEvent(ndn::time::milliseconds(7000),
                           bind(&SocketFixture::terminate, this));

   io.run();
 }

 BOOST_AUTO_TEST_SUITE_END()

 } // namespace test
 } // namespace chronosync
	/* -- Mode: C++; c-file-style: "gnu"; indent-tabs-mode:nil -- */
	/*
	* Copyright (c) 2012-2014 University of California, Los Angeles
	*
	* This file is part of ChronoSync, synchronization library for distributed realtime
	* applications for NDN.
	*
	* ChronoSync is free software: you can redistribute it and/or modify it under the terms
	* of the GNU General Public License as published by the Free Software Foundation, either
	* version 3 of the License, or (at your option) any later version.
	*
	* ChronoSync is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY;
	* without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
	* PURPOSE. See the GNU General Public License for more details.
	*
	* You should have received a copy of the GNU General Public License along with
	* ChronoSync, e.g., in COPYING.md file. If not, see <http://www.gnu.org/licenses/>.
	*/

	#include "socket.hpp"

	#include "boost-test.hpp"

	namespace chronosync {
	namespace test {

	using std::string;
	using std::vector;
	using std::map;

	/**
	* @brief Emulate an app that use the Socket class
	*
	* The app has two types of data set: one is simply string while the other is integer array.
	* For each type of data set, the app has a specific fetching strategy.
	*/
	class SocketTestApp : noncopyable
	{
	public:
	SocketTestApp(const Name& syncPrefix,
	const Name& userPrefix,
	ndn::Face& face,
	bool isNum)
	: sum(0)
	, socket(syncPrefix,
	userPrefix,
	face,
	isNum ? bind(&SocketTestApp::fetchNumbers, this, _1) :
	bind(&SocketTestApp::fetchAll, this, _1))
	{
	}

	void
	set(const shared_ptr<const Data>& dataPacket)
	{
	// std::cerr << "set Data" << std::endl;
	Name dataName(dataPacket->getName());
	string str2(reinterpret_cast<const char*>(dataPacket->getContent().value()),
	dataPacket->getContent().value_size());
	data.insert(make_pair(dataName, str2));
	}

	void
	set(Name name, const char* buf, int len)
	{
	string str2(buf, len);
	data.insert(make_pair(name, str2));
	}

	void
	setNum(const shared_ptr<const Data>& dataPacket)
	{
	// std::cerr << "setNum Data" << std::endl;
	size_t n = dataPacket->getContent().value_size() / 4;
	const uint32_t* numbers = reinterpret_cast<const uint32_t*>(dataPacket->getContent().value());
	for (size_t i = 0; i < n; i++) {
	sum += numbers[i];
	}
	}

	void
	setNum(Name name, const uint8_t* buf, int len)
	{
	BOOST_ASSERT(len >= 4);

	int n = len / 4;
	const uint32_t* numbers = reinterpret_cast<const uint32_t*>(buf);
	for (int i = 0; i < n; i++) {
	sum += numbers[i];
	}
	}

	void
	fetchAll(const vector<MissingDataInfo>& v)
	{
	// std::cerr << "fetchAll" << std::endl;
	for (int i = 0; i < v.size(); i++) {
	for(SeqNo s = v[i].low; s <= v[i].high; ++s) {
	socket.fetchData(v[i].session, s, [this] (const shared_ptr<const Data>& dataPacket) {
	this->set(dataPacket);
	});
	}
	}
	}

	void
	fetchNumbers(const vector<MissingDataInfo> &v)
	{
	// std::cerr << "fetchNumbers" << std::endl;
	for (int i = 0; i < v.size(); i++) {
	for(SeqNo s = v[i].low; s <= v[i].high; ++s) {
	socket.fetchData(v[i].session, s, [this] (const shared_ptr<const Data>& dataPacket) {
	this->setNum(dataPacket);
	});
	}
	}
	}

	string
	toString()
	{
	string str = "\n";
	for (map<Name, string>::iterator it = data.begin(); it != data.end(); ++it) {
	str += "<";
	str += it->first.toUri();
	str += "\|";
	str += it->second;
	str += ">";
	str += "\n";
	}

	return str;
	}

	map<ndn::Name, string> data;
	uint32_t sum;
	Socket socket;
	};

	class SocketFixture
	{
	public:
	SocketFixture()
	: syncPrefix("/ndn/broadcast/sync")
	, scheduler(io)
	{
	syncPrefix.appendVersion();
	userPrefix[0] = Name("/user0");
	userPrefix[1] = Name("/user1");
	userPrefix[2] = Name("/user2");

	faces[0] = make_shared<ndn::Face>(ref(io));
	faces[1] = make_shared<ndn::Face>(ref(io));
	faces[2] = make_shared<ndn::Face>(ref(io));
	}

	void
	createSocket(size_t idx, bool isNum)
	{
	app[idx] = make_shared<SocketTestApp>(syncPrefix, userPrefix[idx], ref(*faces[idx]), isNum);
	sessionName[idx] = app[idx]->socket.getLogic().getSessionName();
	}

	void
	publishAppData(size_t idx, const string& data)
	{
	app[idx]->socket.publishData(reinterpret_cast<const uint8_t*>(data.c_str()), data.size(),
	ndn::time::milliseconds(1000));
	}

	void
	setAppData(size_t idx, SeqNo seqNo, const string& data)
	{
	Name dataName = sessionName[idx];
	dataName.appendNumber(seqNo);
	app[idx]->set(dataName, data.c_str(), data.size());
	}

	void
	publishAppNum(size_t idx, const uint8_t* buf, size_t size)
	{
	app[idx]->socket.publishData(buf, size, ndn::time::milliseconds(1000));
	}

	void
	setAppNum(size_t idx, SeqNo seqNo, const uint8_t* buf, size_t size)
	{
	Name dataName = sessionName[idx];
	dataName.appendNumber(seqNo);
	app[idx]->setNum(dataName, buf, size);
	}

	void
	check(int round)
	{
	BOOST_CHECK_EQUAL(app[0]->toString(), app[1]->toString());
	BOOST_CHECK_EQUAL(app[0]->toString(), app[2]->toString());
	}

	void
	check2Num(int num)
	{
	BOOST_CHECK_EQUAL(app[0]->sum, app[1]->sum);
	BOOST_CHECK_EQUAL(app[1]->sum, num);
	}

	void
	terminate()
	{
	io.stop();
	}

	Name syncPrefix;
	Name userPrefix[3];
	Name sessionName[3];

	boost::asio::io_service io;
	shared_ptr<ndn::Face> faces[3];
	ndn::Scheduler scheduler;
	shared_ptr<SocketTestApp> app[3];
	};



	BOOST_FIXTURE_TEST_SUITE(SocketTests, SocketFixture)

	BOOST_AUTO_TEST_CASE(BasicData)
	{
	scheduler.scheduleEvent(ndn::time::milliseconds(0),
	bind(&SocketFixture::createSocket, this, 0, false));

	scheduler.scheduleEvent(ndn::time::milliseconds(50),
	bind(&SocketFixture::createSocket, this, 1, false));

	scheduler.scheduleEvent(ndn::time::milliseconds(100),
	bind(&SocketFixture::createSocket, this, 2, false));

	string data0 = "Very funny Scotty, now beam down my clothes";
	scheduler.scheduleEvent(ndn::time::milliseconds(150),
	bind(&SocketFixture::publishAppData, this, 0, data0));
	scheduler.scheduleEvent(ndn::time::milliseconds(1150),
	bind(&SocketFixture::setAppData, this, 0, 1, data0));
	scheduler.scheduleEvent(ndn::time::milliseconds(1160),
	bind(&SocketFixture::check, this, 1));

	string data1 = "Yes, give me that ketchup";
	string data2 = "Don't look conspicuous, it draws fire";
	scheduler.scheduleEvent(ndn::time::milliseconds(1170),
	bind(&SocketFixture::publishAppData, this, 0, data1));
	scheduler.scheduleEvent(ndn::time::milliseconds(1180),
	bind(&SocketFixture::publishAppData, this, 0, data2));
	scheduler.scheduleEvent(ndn::time::milliseconds(2150),
	bind(&SocketFixture::setAppData, this, 0, 2, data1));
	scheduler.scheduleEvent(ndn::time::milliseconds(2160),
	bind(&SocketFixture::setAppData, this, 0, 3, data2));
	scheduler.scheduleEvent(ndn::time::milliseconds(2170),
	bind(&SocketFixture::check, this, 2));

	string data3 = "You surf the Internet, I surf the real world";
	string data4 = "I got a fortune cookie once that said 'You like Chinese food'";
	string data5 = "Real men wear pink. Why? Because their wives make them";
	scheduler.scheduleEvent(ndn::time::milliseconds(3180),
	bind(&SocketFixture::publishAppData, this, 2, data3));
	scheduler.scheduleEvent(ndn::time::milliseconds(3200),
	bind(&SocketFixture::publishAppData, this, 1, data4));
	scheduler.scheduleEvent(ndn::time::milliseconds(3210),
	bind(&SocketFixture::publishAppData, this, 1, data5));
	scheduler.scheduleEvent(ndn::time::milliseconds(4710),
	bind(&SocketFixture::setAppData, this, 2, 1, data3));
	scheduler.scheduleEvent(ndn::time::milliseconds(4720),
	bind(&SocketFixture::setAppData, this, 1, 1, data4));
	scheduler.scheduleEvent(ndn::time::milliseconds(4730),
	bind(&SocketFixture::setAppData, this, 1, 2, data5));
	scheduler.scheduleEvent(ndn::time::milliseconds(4800),
	bind(&SocketFixture::check, this, 3));

	// not sure weither this is simultanous data generation from multiple sources
	string data6 = "Shakespeare says: 'Prose before hos.'";
	string data7 = "Pick good people, talent never wears out";
	scheduler.scheduleEvent(ndn::time::milliseconds(5500),
	bind(&SocketFixture::publishAppData, this, 0, data6));
	scheduler.scheduleEvent(ndn::time::milliseconds(5500),
	bind(&SocketFixture::publishAppData, this, 1, data7));
	scheduler.scheduleEvent(ndn::time::milliseconds(6800),
	bind(&SocketFixture::setAppData, this, 0, 4, data6));
	scheduler.scheduleEvent(ndn::time::milliseconds(6800),
	bind(&SocketFixture::setAppData, this, 1, 3, data7));
	scheduler.scheduleEvent(ndn::time::milliseconds(6900),
	bind(&SocketFixture::check, this, 4));

	scheduler.scheduleEvent(ndn::time::milliseconds(7000),
	bind(&SocketFixture::terminate, this));

	io.run();
	}

	BOOST_AUTO_TEST_CASE(BasicNumber)
	{
	scheduler.scheduleEvent(ndn::time::milliseconds(0),
	bind(&SocketFixture::createSocket, this, 0, true));
	scheduler.scheduleEvent(ndn::time::milliseconds(50),
	bind(&SocketFixture::createSocket, this, 1, true));

	uint32_t num1[5] = {0, 1, 2, 3, 4};
	uint8_t* buf1 = reinterpret_cast<uint8_t*>(num1);
	size_t size1 = sizeof(num1);
	scheduler.scheduleEvent(ndn::time::milliseconds(100),
	bind(&SocketFixture::publishAppNum, this, 0, buf1, size1));
	scheduler.scheduleEvent(ndn::time::milliseconds(150),
	bind(&SocketFixture::setAppNum, this, 0, 0, buf1, size1));
	scheduler.scheduleEvent(ndn::time::milliseconds(1000),
	bind(&SocketFixture::check2Num, this, 10));

	uint32_t num2[5] = {9, 7, 2, 1, 1};
	uint8_t* buf2 = reinterpret_cast<uint8_t*>(num2);
	size_t size2 = sizeof(num2);
	scheduler.scheduleEvent(ndn::time::milliseconds(1100),
	bind(&SocketFixture::publishAppNum, this, 1, buf2, size2));
	scheduler.scheduleEvent(ndn::time::milliseconds(1150),
	bind(&SocketFixture::setAppNum, this, 1, 0, buf2, size2));
	scheduler.scheduleEvent(ndn::time::milliseconds(2000),
	bind(&SocketFixture::check2Num, this, 30));

	scheduler.scheduleEvent(ndn::time::milliseconds(7000),
	bind(&SocketFixture::terminate, this));

	io.run();
	}

	BOOST_AUTO_TEST_SUITE_END()

	} // namespace test
	} // namespace chronosync