/*
This file is part of cpp-ethereum.

cpp-ethereum 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.

cpp-ethereum 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 cpp-ethereum.  If not, see <http://www.gnu.org/licenses/>.
*/
/** @file whisperMessage.cpp
* @author Vladislav Gluhovsky <vlad@ethdev.com>
* @date July 2015
*/

#include <thread>
#include <boost/test/unit_test.hpp>
#include <libp2p/Host.h>
#include <libwhisper/WhisperDB.h>
#include <libwhisper/WhisperHost.h>

using namespace std;
using namespace dev;
using namespace dev::shh;
using namespace dev::p2p;

struct P2PFixture
{
	P2PFixture() { dev::p2p::NodeIPEndpoint::test_allowLocal = true; }
	~P2PFixture() { dev::p2p::NodeIPEndpoint::test_allowLocal = false; }
};

BOOST_FIXTURE_TEST_SUITE(whisperDB, P2PFixture)

BOOST_AUTO_TEST_CASE(basic)
{
	VerbosityHolder setTemporaryLevel(10);
	cnote << "Testing Whisper DB...";

	string s;
	string const text1 = "lorem";
	string const text2 = "ipsum";
	h256 h1(0xBEEF);
	h256 h2(0xC0FFEE);
	WhisperDB db(WhisperDB::Messages);

	db.kill(h1);
	db.kill(h2);

	s = db.lookup(h1);
	BOOST_REQUIRE(s.empty());

	db.insert(h1, text2);
	s = db.lookup(h2);
	BOOST_REQUIRE(s.empty());
	s = db.lookup(h1);
	BOOST_REQUIRE(!s.compare(text2));

	db.insert(h1, text1);
	s = db.lookup(h2);
	BOOST_REQUIRE(s.empty());
	s = db.lookup(h1);
	BOOST_REQUIRE(!s.compare(text1));

	db.insert(h2, text2);
	s = db.lookup(h2);
	BOOST_REQUIRE(!s.compare(text2));
	s = db.lookup(h1);
	BOOST_REQUIRE(!s.compare(text1));

	db.kill(h1);
	db.kill(h2);

	s = db.lookup(h2);
	BOOST_REQUIRE(s.empty());
	s = db.lookup(h1);
	BOOST_REQUIRE(s.empty());
}

BOOST_AUTO_TEST_CASE(persistence)
{
	VerbosityHolder setTemporaryLevel(10);
	cnote << "Testing persistence of Whisper DB...";

	string s;
	string const text1 = "sator";
	string const text2 = "arepo";
	h256 const h1(0x12345678);
	h256 const h2(0xBADD00DE);

	{
		WhisperDB db(WhisperDB::Messages);
		db.kill(h1);
		db.kill(h2);
		s = db.lookup(h1);
		BOOST_REQUIRE(s.empty());
		db.insert(h1, text2);
		s = db.lookup(h2);
		BOOST_REQUIRE(s.empty());
		s = db.lookup(h1);
		BOOST_REQUIRE(!s.compare(text2));
	}

	this_thread::sleep_for(chrono::milliseconds(20));

	{
		WhisperDB db(WhisperDB::Messages);
		db.insert(h1, text1);
		db.insert(h2, text2);
	}

	this_thread::sleep_for(chrono::milliseconds(20));

	{
		WhisperDB db(WhisperDB::Messages);
		s = db.lookup(h2);
		BOOST_REQUIRE(!s.compare(text2));
		s = db.lookup(h1);
		BOOST_REQUIRE(!s.compare(text1));
		db.kill(h1);
		db.kill(h2);
	}
}

BOOST_AUTO_TEST_CASE(messages)
{
	cnote << "Testing load/save Whisper messages...";
	VerbosityHolder setTemporaryLevel(2);
	unsigned const TestSize = 3;
	map<h256, Envelope> m1;
	map<h256, Envelope> preexisting;
	KeyPair us = KeyPair::create();

	{
		p2p::Host h("Test");
		auto wh = h.registerCapability(new WhisperHost(true));
		preexisting = wh->all();
		cnote << preexisting.size() << "preexisting messages in DB";
		wh->installWatch(BuildTopic("test"));

		for (unsigned i = 0; i < TestSize; ++i)
			wh->post(us.sec(), RLPStream().append(i).out(), BuildTopic("test"), 0xFFFFF);

		m1 = wh->all();
	}

	{
		p2p::Host h("Test");
		auto wh = h.registerCapability(new WhisperHost(true));
		map<h256, Envelope> m2 = wh->all();
		wh->installWatch(BuildTopic("test"));
		BOOST_REQUIRE_EQUAL(m1.size(), m2.size());
		BOOST_REQUIRE_EQUAL(m1.size() - preexisting.size(), TestSize);

		for (auto i: m1)
		{
			RLPStream rlp1;
			RLPStream rlp2;
			i.second.streamRLP(rlp1);
			m2[i.first].streamRLP(rlp2);
			BOOST_REQUIRE_EQUAL(rlp1.out().size(), rlp2.out().size());
			for (unsigned j = 0; j < rlp1.out().size(); ++j)
				BOOST_REQUIRE_EQUAL(rlp1.out()[j], rlp2.out()[j]);
		}
	}

	WhisperDB db(WhisperDB::Messages);
	unsigned x = 0;

	for (auto i: m1)
		if (preexisting.find(i.first) == preexisting.end())
		{
			db.kill(i.first);
			++x;
		}

	BOOST_REQUIRE_EQUAL(x, TestSize);
}

BOOST_AUTO_TEST_CASE(corruptedData)
{
	cnote << "Testing corrupted data...";
	VerbosityHolder setTemporaryLevel(2);
	map<h256, Envelope> m;
	h256 x = h256::random();

	{
		WhisperDB db(WhisperDB::Messages);
		db.insert(x, "this is a test input, representing corrupt data");
	}

	{
		p2p::Host h("Test");
		auto wh = h.registerCapability(new WhisperHost(true));
		m = wh->all();
		BOOST_REQUIRE(m.end() == m.find(x));
	}

	{
		WhisperDB db(WhisperDB::Messages);
		string s = db.lookup(x);
		BOOST_REQUIRE(s.empty());
	}
}

BOOST_AUTO_TEST_CASE(filters)
{
	cnote << "Testing filters saving...";
	VerbosityHolder setTemporaryLevel(2);
	string const app("test suite whisperDB/filters");
	string const password("some pseudorandom stuff");

	{
		p2p::Host h("Test");
		auto wh = h.registerCapability(new WhisperHost());
		wh->installWatch(BuildTopic("t1"));
		wh->installWatch(BuildTopic("t2"));
		wh->saveTopicsToDB(app, password);
	}

	short unsigned port1 = 30313;
	unsigned const step = 10;
	bool host1Ready = false;
	bool sent = false;
	unsigned result = 0;
	unsigned messageCount = 0;

	Host host1("Test", NetworkPreferences("127.0.0.1", port1, false));
	host1.setIdealPeerCount(1);
	auto whost1 = host1.registerCapability(new WhisperHost());
	host1.start();
	auto ids = whost1->restoreTopicsFromDB(app, password);

	std::thread listener([&]()
	{
		setThreadName("other");
		host1Ready = true;

		for (unsigned i = 0; i < 16000 && !sent; i += step)
			this_thread::sleep_for(chrono::milliseconds(step));

		for (unsigned j = 0; j < 200 && messageCount < 2; ++j)
		{
			for (unsigned id: ids)
				for (auto const& e: whost1->checkWatch(id))
				{
					Message msg = whost1->envelope(e).open(whost1->fullTopics(id));
					unsigned x = RLP(msg.payload()).toInt<unsigned>();
					cnote << "New message:" << x;
					result += x;
					++messageCount;
				}

			this_thread::sleep_for(chrono::milliseconds(50));
		}
	});

	Host host2("Test", NetworkPreferences("127.0.0.1", 30314, false));
	host2.setIdealPeerCount(1);
	auto whost2 = host2.registerCapability(new WhisperHost());
	host2.start();

	for (unsigned i = 0; i < 3000 && !host1.haveNetwork(); i += step)
		this_thread::sleep_for(chrono::milliseconds(step));

	host2.requirePeer(host1.id(), NodeIPEndpoint(bi::address::from_string("127.0.0.1"), port1, port1));

	for (unsigned i = 0; i < 3000 && !host1Ready; i += step)
		this_thread::sleep_for(chrono::milliseconds(step));

	for (unsigned i = 0; i < 3000 && (!host1.peerCount() || !host2.peerCount()); i += step)
		this_thread::sleep_for(chrono::milliseconds(step));

	unsigned ttl = 777000;
	whost2->post(RLPStream().append(8).out(), BuildTopic("t8"), ttl);
	whost2->post(RLPStream().append(4).out(), BuildTopic("t4"), ttl);
	whost2->post(RLPStream().append(1).out(), BuildTopic("t1"), ttl);
	whost2->post(RLPStream().append(2).out(), BuildTopic("t2"), ttl);
	whost2->post(RLPStream().append(16).out(), BuildTopic("t16"), ttl);
	sent = true;

	for (unsigned i = 0; i < 3000 && messageCount < 2; i += step)
		this_thread::sleep_for(chrono::milliseconds(step));

	listener.join();
	BOOST_REQUIRE_EQUAL(messageCount, 2);
	BOOST_REQUIRE_EQUAL(result, 3);
}

BOOST_AUTO_TEST_SUITE_END()